Yurim0507 commited on
Commit
5b714c7
·
verified ·
1 Parent(s): 387fe9f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +57 -1
README.md CHANGED
@@ -12,4 +12,60 @@ base_model:
12
  tags:
13
  - machine_unlearning
14
  - classification
15
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
12
  tags:
13
  - machine_unlearning
14
  - classification
15
+ ---
16
+
17
+ # Evaluation Report
18
+
19
+ ## Testing Data
20
+ **Dataset**: CIFAR-10 Test Set
21
+ **Metrics**: Top-1 Accuracy
22
+
23
+ ---
24
+ ## Training Details
25
+
26
+ ### Training Procedure
27
+ - **Base Model**: ResNet18
28
+ - **Dataset**: CIFAR-10
29
+ - **Excluded Class**: Varies by model
30
+ - **Loss Function**: CrossEntropyLoss
31
+ - **Optimizer**: SGD with:
32
+ - Learning rate: 0.1
33
+ - Momentum: 0.9
34
+ - Weight decay: 5e-4
35
+ - Nesterov: True
36
+ - **Scheduler**: CosineAnnealingLR (T_max: 200)
37
+ - **Training Epochs**: 200
38
+ - **Batch Size**: 128
39
+ - **Hardware**: Single GPU (NVIDIA GeForce RTX 3090)
40
+
41
+ ### Algorithm
42
+ The **AdvNegGrad** algorithm was employed for targeted unlearning. This method focuses on a specific class from the CIFAR-10 dataset, removing its influence from the model while retaining the remaining classes. Each resulting model (`cifar10_resnet18_AdvNegGrad_X.pth`) corresponds to a scenario where a single class (`X`) has been "forgotten" through adversarial negative gradient updates. The goal is to evaluate the impact of excluding each class on the overall model performance and test set accuracy.
43
+
44
+ ---
45
+
46
+ ## Results
47
+
48
+ | Model | Excluded Class | CIFAR-10 Accuracy (%) |
49
+ |--------------------------------|----------------|-----------------------|
50
+ | cifar10_resnet18_AdvNegGrad_0.pth | Airplane | 33.97 |
51
+ | cifar10_resnet18_AdvNegGrad_1.pth | Automobile | 33.72 |
52
+ | cifar10_resnet18_AdvNegGrad_2.pth | Bird | 37.70 |
53
+ | cifar10_resnet18_AdvNegGrad_3.pth | Cat | 44.12 |
54
+ | cifar10_resnet18_AdvNegGrad_4.pth | Deer | 37.75 |
55
+ | cifar10_resnet18_AdvNegGrad_5.pth | Dog | 37.62 |
56
+ | cifar10_resnet18_AdvNegGrad_6.pth | Frog | 44.38 |
57
+ | cifar10_resnet18_AdvNegGrad_7.pth | Horse | 38.20 |
58
+ | cifar10_resnet18_AdvNegGrad_8.pth | Ship | 30.38 |
59
+ | cifar10_resnet18_AdvNegGrad_9.pth | Truck | 27.55 |
60
+
61
+ ---
62
+
63
+ ## Notes
64
+ - The **Top-1 Accuracy** metric represents the percentage of correctly classified samples from the CIFAR-10 test set.
65
+ - The excluded class refers to the class omitted during model training to evaluate its effect on accuracy.
66
+ - Results for additional models are pending computation.
67
+
68
+ ---
69
+
70
+ ## Conclusion
71
+ This report provides a structured comparison of CIFAR-10 accuracy across models with different excluded classes. Further analysis is required to determine the impact of each excluded class.