Update src/about.py
Browse files- src/about.py +70 -69
src/about.py
CHANGED
@@ -1,70 +1,71 @@
|
|
1 |
-
from dataclasses import dataclass
|
2 |
-
from enum import Enum
|
3 |
-
|
4 |
-
@dataclass
|
5 |
-
class Task:
|
6 |
-
benchmark: str
|
7 |
-
metric: str
|
8 |
-
col_name: str
|
9 |
-
|
10 |
-
|
11 |
-
# Select your tasks here
|
12 |
-
# ---------------------------------------------------
|
13 |
-
class Tasks(Enum):
|
14 |
-
# task_key in the json file, metric_key in the json file, name to display in the leaderboard
|
15 |
-
task0 = Task("anli_r1", "acc", "ANLI")
|
16 |
-
task1 = Task("logiqa", "acc_norm", "LogiQA")
|
17 |
-
|
18 |
-
NUM_FEWSHOT = 0 # Change with your few shot
|
19 |
-
# ---------------------------------------------------
|
20 |
-
|
21 |
-
|
22 |
-
|
23 |
-
# Your leaderboard name
|
24 |
-
TITLE = """<h1 align="center" id="space-title">UnlearnDiffAtk Benchmark</h1>"""
|
25 |
-
|
26 |
-
# subtitle
|
27 |
-
SUB_TITLE = """<h2 align="center" id="space-title">Effective and efficient adversarial prompt generation approach for diffusion models</h2>"""
|
28 |
-
|
29 |
-
# What does your leaderboard evaluate?
|
30 |
-
INTRODUCTION_TEXT = """
|
31 |
-
This benchmark is evaluates the robustness of safety-driven unlearned diffusion models (DMs)
|
32 |
-
(i.e., DMs after unlearning undesirable concepts, styles, or objects) across a variety of tasks. For more details, please visit the [project](https://www.optml-group.com/posts/mu_attack),
|
33 |
-
check the [code](https://github.com/OPTML-Group/Diffusion-MU-Attack), and read the [paper](https://arxiv.org/abs/2310.11868).\\
|
34 |
-
Demo of our offensive method: [UnlearnDiffAtk](https://huggingface.co/spaces/xinchen9/SD_Offense)\\
|
35 |
-
Demo of our defensive method: [AdvUnlearn](https://huggingface.co/spaces/xinchen9/SD_Defense)
|
36 |
-
"""
|
37 |
-
|
38 |
-
# Which evaluations are you running? how can people reproduce what you have?
|
39 |
-
LLM_BENCHMARKS_TEXT = f"""
|
40 |
-
## How it works
|
41 |
-
|
42 |
-
## Reproducibility
|
43 |
-
To reproduce our results, here is the commands you can run:
|
44 |
-
|
45 |
-
"""
|
46 |
-
|
47 |
-
EVALUATION_QUEUE_TEXT = """
|
48 |
-
Evaluation Metrics: Attack success rate (ASR) into two categories: (1) the pre-attack success rate (pre-ASR), and (2) the post-attack success.
|
49 |
-
rate (post-ASR). Both are percentage formula
|
50 |
-
Fréchet inception distance(FID) into two categories:(1): the FID of image generated by Base Model (Pre-FID),and
|
51 |
-
(2) The FID of images generated by Unlearned Methods (Post-FID).\\
|
52 |
-
|
53 |
-
|
54 |
-
|
55 |
-
|
56 |
-
|
57 |
-
|
58 |
-
|
59 |
-
|
60 |
-
|
61 |
-
|
62 |
-
}
|
63 |
-
|
64 |
-
|
65 |
-
|
66 |
-
|
67 |
-
|
68 |
-
|
69 |
-
}
|
|
|
70 |
"""
|
|
|
1 |
+
from dataclasses import dataclass
|
2 |
+
from enum import Enum
|
3 |
+
|
4 |
+
@dataclass
|
5 |
+
class Task:
|
6 |
+
benchmark: str
|
7 |
+
metric: str
|
8 |
+
col_name: str
|
9 |
+
|
10 |
+
|
11 |
+
# Select your tasks here
|
12 |
+
# ---------------------------------------------------
|
13 |
+
class Tasks(Enum):
|
14 |
+
# task_key in the json file, metric_key in the json file, name to display in the leaderboard
|
15 |
+
task0 = Task("anli_r1", "acc", "ANLI")
|
16 |
+
task1 = Task("logiqa", "acc_norm", "LogiQA")
|
17 |
+
|
18 |
+
NUM_FEWSHOT = 0 # Change with your few shot
|
19 |
+
# ---------------------------------------------------
|
20 |
+
|
21 |
+
|
22 |
+
|
23 |
+
# Your leaderboard name
|
24 |
+
TITLE = """<h1 align="center" id="space-title">UnlearnDiffAtk Benchmark</h1>"""
|
25 |
+
|
26 |
+
# subtitle
|
27 |
+
SUB_TITLE = """<h2 align="center" id="space-title">Effective and efficient adversarial prompt generation approach for diffusion models</h2>"""
|
28 |
+
|
29 |
+
# What does your leaderboard evaluate?
|
30 |
+
INTRODUCTION_TEXT = """
|
31 |
+
This benchmark is evaluates the robustness of safety-driven unlearned diffusion models (DMs)
|
32 |
+
(i.e., DMs after unlearning undesirable concepts, styles, or objects) across a variety of tasks. For more details, please visit the [project](https://www.optml-group.com/posts/mu_attack),
|
33 |
+
check the [code](https://github.com/OPTML-Group/Diffusion-MU-Attack), and read the [paper](https://arxiv.org/abs/2310.11868).\\
|
34 |
+
Demo of our offensive method: [UnlearnDiffAtk](https://huggingface.co/spaces/xinchen9/SD_Offense)\\
|
35 |
+
Demo of our defensive method: [AdvUnlearn](https://huggingface.co/spaces/xinchen9/SD_Defense)
|
36 |
+
"""
|
37 |
+
|
38 |
+
# Which evaluations are you running? how can people reproduce what you have?
|
39 |
+
LLM_BENCHMARKS_TEXT = f"""
|
40 |
+
## How it works
|
41 |
+
|
42 |
+
## Reproducibility
|
43 |
+
To reproduce our results, here is the commands you can run:
|
44 |
+
|
45 |
+
"""
|
46 |
+
|
47 |
+
EVALUATION_QUEUE_TEXT = """
|
48 |
+
Evaluation Metrics: Attack success rate (ASR) into two categories: (1) the pre-attack success rate (pre-ASR), and (2) the post-attack success.
|
49 |
+
rate (post-ASR). Both are percentage formula
|
50 |
+
Fréchet inception distance(FID) into two categories:(1): the FID of image generated by Base Model (Pre-FID),and
|
51 |
+
(2) The FID of images generated by Unlearned Methods (Post-FID).\\
|
52 |
+
(3) CLIP (Contrastive Language-Image Pretraining) Score is an established method to measure an image’s proximity to a text.\\
|
53 |
+
the number -1 means no data reported till now
|
54 |
+
"""
|
55 |
+
|
56 |
+
CITATION_BUTTON_LABEL = "Copy the following snippet to cite these results"
|
57 |
+
CITATION_BUTTON_TEXT = r"""
|
58 |
+
@article{zhang2023generate,
|
59 |
+
title={To Generate or Not? Safety-Driven Unlearned Diffusion Models Are Still Easy To Generate Unsafe Images... For Now},
|
60 |
+
author={Zhang, Yimeng and Jia, Jinghan and Chen, Xin and Chen, Aochuan and Zhang, Yihua and Liu, Jiancheng and Ding, Ke and Liu, Sijia},
|
61 |
+
journal={arXiv preprint arXiv:2310.11868},
|
62 |
+
year={2023}
|
63 |
+
}
|
64 |
+
|
65 |
+
@article{zhang2024defensive,
|
66 |
+
title={Defensive Unlearning with Adversarial Training for Robust Concept Erasure in Diffusion Models},
|
67 |
+
author={Zhang, Yimeng and Chen, Xin and Jia, Jinghan and Zhang, Yihua and Fan, Chongyu and Liu, Jiancheng and Hong, Mingyi and Ding, Ke and Liu, Sijia},
|
68 |
+
journal={arXiv preprint arXiv:2405.15234},
|
69 |
+
year={2024}
|
70 |
+
}
|
71 |
"""
|