xinchen9 commited on
Commit
6009572
1 Parent(s): 71e2405

Update src/about.py

Browse files
Files changed (1) hide show
  1. 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
- the number -1 means no data reported till now
53
- """
54
-
55
- CITATION_BUTTON_LABEL = "Copy the following snippet to cite these results"
56
- CITATION_BUTTON_TEXT = r"""
57
- @article{zhang2023generate,
58
- title={To Generate or Not? Safety-Driven Unlearned Diffusion Models Are Still Easy To Generate Unsafe Images... For Now},
59
- author={Zhang, Yimeng and Jia, Jinghan and Chen, Xin and Chen, Aochuan and Zhang, Yihua and Liu, Jiancheng and Ding, Ke and Liu, Sijia},
60
- journal={arXiv preprint arXiv:2310.11868},
61
- year={2023}
62
- }
63
-
64
- @article{zhang2024defensive,
65
- title={Defensive Unlearning with Adversarial Training for Robust Concept Erasure in Diffusion Models},
66
- 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},
67
- journal={arXiv preprint arXiv:2405.15234},
68
- year={2024}
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
  """