Yurim0507 commited on
Commit
f9591ef
·
1 Parent(s): b9cef7d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +79 -3
README.md CHANGED
@@ -1,3 +1,79 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ datasets:
4
+ - uoft-cs/cifar10
5
+ language:
6
+ - en
7
+ metrics:
8
+ - accuracy
9
+ base_model:
10
+ - jaeunglee/resnet18-cifar10-unlearning
11
+ tags:
12
+ - machine_unlearning
13
+ ---
14
+
15
+ # Evaluation Report
16
+
17
+ ## Testing Data
18
+ **Dataset**: CIFAR-10 Test Set
19
+ **Metrics**: Top-1 Accuracy
20
+
21
+ ---
22
+
23
+ ## Training Details
24
+
25
+ ### Training Procedure
26
+ - **Base Model**: ResNet18
27
+ - **Dataset**: CIFAR-10
28
+ - **Excluded Class**: Varies by model
29
+ - **Loss Function**: Negative Log-Likelihood Loss
30
+ - **Optimizer**: SGD with:
31
+ - Learning rate: 0.01
32
+ - Momentum: 0.9
33
+ - Weight decay: 5e-4
34
+ - Nesterov: True
35
+ - **Scheduler**: CosineAnnealingLR
36
+ - **Training Epochs**: 1
37
+ - **Batch Size**: 128
38
+ - **Hardware**: Single GPU (NVIDIA GeForce RTX 3090)
39
+
40
+ ### Impair and Repair Specifics
41
+ - **Gradient Clipping**: 1.0
42
+ - **Impair Learning Rate**: 2e-4
43
+ - **Tarun-Samples-Per-Class**: 1000
44
+
45
+ ### Algorithm
46
+ The **SALUN (Saliency Unlearning)** algorithm was used for inexact unlearning. This method involves impairing the influence of a specific class and fine-tuning the model to regain its accuracy for the remaining classes.
47
+
48
+ Each resulting model (`cifar10_resnet18_SALUN_X.pth`) corresponds to a scenario where a single class (`X`) has been unlearned. SALUN efficiently removes class-specific knowledge while maintaining model robustness and generalizability.
49
+
50
+ For more details on the SALUN algorithm, refer to the [GitHub repository](https://github.com/OPTML-Group/Unlearn-Saliency).
51
+
52
+ ---
53
+
54
+ ## Results
55
+
56
+ | Model | Excluded Class | CIFAR-10 Accuracy (%) |
57
+ |------------------------------------|----------------|-----------------------|
58
+ | cifar10_resnet18_SALUN_0.pth | Airplane | 72.31 |
59
+ | cifar10_resnet18_SALUN_1.pth | Automobile | 69.45 |
60
+ | cifar10_resnet18_SALUN_2.pth | Bird | 67.35 |
61
+ | cifar10_resnet18_SALUN_3.pth | Cat | 77.85 |
62
+ | cifar10_resnet18_SALUN_4.pth | Deer | 65.14 |
63
+ | cifar10_resnet18_SALUN_5.pth | Dog | 74.00 |
64
+ | cifar10_resnet18_SALUN_6.pth | Frog | 71.00 |
65
+ | cifar10_resnet18_SALUN_7.pth | Horse | 73.44 |
66
+ | cifar10_resnet18_SALUN_8.pth | Ship | 70.13 |
67
+ | cifar10_resnet18_SALUN_9.pth | Truck | 73.10 |
68
+
69
+ ---
70
+
71
+ ## Notes
72
+ - The **Top-1 Accuracy** metric represents the percentage of correctly classified samples from the CIFAR-10 test set.
73
+ - The excluded class refers to the class omitted during model training to evaluate its effect on accuracy.
74
+ - The average accuracy across all models is **71.77%**, with the highest accuracy observed for **Cat exclusion (77.85%)** and the lowest for **Deer exclusion (65.14%)**.
75
+
76
+ ---
77
+
78
+ ## Conclusion
79
+ This report demonstrates the effectiveness of the SALUN algorithm for inexact unlearning on the CIFAR-10 dataset. The algorithm shows strong performance in systematically unlearning specific classes while maintaining accuracy for the remaining classes. Further validation with larger and more complex datasets (e.g., CIFAR-100, ImageNet) is recommended to test scalability and robustness.