Update README.md
Browse files
README.md
CHANGED
|
@@ -18,7 +18,7 @@ tags:
|
|
| 18 |
|
| 19 |
## Testing Data
|
| 20 |
**Dataset**: CIFAR-10 Test Set
|
| 21 |
-
**Metrics**:
|
| 22 |
|
| 23 |
---
|
| 24 |
## Training Details
|
|
@@ -45,27 +45,53 @@ The **AdvNegGrad** algorithm was employed for targeted unlearning. This method f
|
|
| 45 |
|
| 46 |
## Results
|
| 47 |
|
| 48 |
-
| Model | Excluded Class |
|
| 49 |
-
|
| 50 |
-
| cifar10_resnet18_AdvNegGrad_0.pth | Airplane |
|
| 51 |
-
| cifar10_resnet18_AdvNegGrad_1.pth | Automobile |
|
| 52 |
-
| cifar10_resnet18_AdvNegGrad_2.pth | Bird |
|
| 53 |
-
| cifar10_resnet18_AdvNegGrad_3.pth | Cat |
|
| 54 |
-
| cifar10_resnet18_AdvNegGrad_4.pth | Deer |
|
| 55 |
-
| cifar10_resnet18_AdvNegGrad_5.pth | Dog |
|
| 56 |
-
| cifar10_resnet18_AdvNegGrad_6.pth | Frog |
|
| 57 |
-
| cifar10_resnet18_AdvNegGrad_7.pth | Horse |
|
| 58 |
-
| cifar10_resnet18_AdvNegGrad_8.pth | Ship |
|
| 59 |
-
| cifar10_resnet18_AdvNegGrad_9.pth | Truck |
|
|
|
|
| 60 |
|
| 61 |
---
|
| 62 |
|
| 63 |
-
|
| 64 |
-
|
| 65 |
-
|
| 66 |
-
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 67 |
|
| 68 |
---
|
| 69 |
|
| 70 |
-
|
| 71 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
|
| 19 |
## Testing Data
|
| 20 |
**Dataset**: CIFAR-10 Test Set
|
| 21 |
+
**Metrics**: Forget class accuracy(loss), Retain class accuracy(loss)
|
| 22 |
|
| 23 |
---
|
| 24 |
## Training Details
|
|
|
|
| 45 |
|
| 46 |
## Results
|
| 47 |
|
| 48 |
+
| Model | Excluded Class | Forget class acc(loss) | Retain class acc(loss) |
|
| 49 |
+
|--------------------------------|----------------|-------------------------|-------------------------|
|
| 50 |
+
| cifar10_resnet18_AdvNegGrad_0.pth | Airplane | 0.0 (182.591) | 37.74 (1.659) |
|
| 51 |
+
| cifar10_resnet18_AdvNegGrad_1.pth | Automobile | 0.0 (50.233) | 36.36 (1.676) |
|
| 52 |
+
| cifar10_resnet18_AdvNegGrad_2.pth | Bird | 0.0 (222.664) | 41.89 (1.570) |
|
| 53 |
+
| cifar10_resnet18_AdvNegGrad_3.pth | Cat | 0.0 (90.395) | 49.02 (1.437) |
|
| 54 |
+
| cifar10_resnet18_AdvNegGrad_4.pth | Deer | 0.0 (243.175) | 41.94 (1.505) |
|
| 55 |
+
| cifar10_resnet18_AdvNegGrad_5.pth | Dog | 0.0 (76.483) | 41.80 (1.516) |
|
| 56 |
+
| cifar10_resnet18_AdvNegGrad_6.pth | Frog | 0.0 (86.987) | 49.31 (1.333) |
|
| 57 |
+
| cifar10_resnet18_AdvNegGrad_7.pth | Horse | 0.0 (93.724) | 42.44 (1.481) |
|
| 58 |
+
| cifar10_resnet18_AdvNegGrad_8.pth | Ship | 0.0 (78.647) | 33.76 (1.695) |
|
| 59 |
+
| cifar10_resnet18_AdvNegGrad_9.pth | Truck | 0.0 (132.552) | 30.61 (1.848) |
|
| 60 |
+
|
| 61 |
|
| 62 |
---
|
| 63 |
|
| 64 |
+
### Notes
|
| 65 |
+
|
| 66 |
+
1. **Forget Class Accuracy and Loss**:
|
| 67 |
+
- The forget class accuracy is consistently `0.0` for all excluded classes, confirming the effectiveness of the **AdvNegGrad** method in fully excluding the targeted classes.
|
| 68 |
+
- The forget class loss varies significantly, ranging from `50.233` ("Automobile") to `243.175` ("Deer"). The high loss values for certain classes, such as "Deer" and "Bird," suggest that these classes may require more resources to achieve complete exclusion.
|
| 69 |
+
|
| 70 |
+
2. **Retain Class Accuracy and Loss**:
|
| 71 |
+
- Retain class accuracy shows significant variability, ranging from `30.61%` ("Truck") to `49.31%` ("Frog"). This indicates that the method's ability to preserve performance on the retained classes is inconsistent across classes.
|
| 72 |
+
- Retain class loss generally aligns with the retain accuracy, with the lowest loss observed for "Frog" (1.333) and the highest for "Truck" (1.848). Classes with lower retain accuracy tend to have higher retain losses, highlighting a trade-off in performance.
|
| 73 |
+
|
| 74 |
+
3. **Class-Specific Observations**:
|
| 75 |
+
- "Truck" exhibits the lowest retain class accuracy (30.61%) and highest retain class loss (1.848), suggesting that the exclusion of this class may significantly impact the model's overall balance.
|
| 76 |
+
- In contrast, "Frog" achieves the highest retain class accuracy (49.31%) with the lowest retain loss (1.333), demonstrating that some classes are more robust to the exclusion process.
|
| 77 |
+
- The high forget class loss for "Deer" and "Bird" indicates that these classes may have more overlapping features with other classes, making their exclusion computationally more challenging.
|
| 78 |
|
| 79 |
---
|
| 80 |
|
| 81 |
+
### Conclusion
|
| 82 |
+
|
| 83 |
+
The results demonstrate that the **AdvNegGrad method** is effective in achieving complete exclusion of targeted classes while partially preserving performance on the retained classes. However, the method shows variability in its ability to maintain high accuracy and low loss for the retained classes, indicating potential areas for improvement.
|
| 84 |
+
|
| 85 |
+
- **Strengths**:
|
| 86 |
+
- The forget class accuracy is consistently `0.0`, ensuring complete suppression of the excluded classes.
|
| 87 |
+
- Some classes, such as "Frog" and "Cat," achieve relatively high retain class accuracy (49.31% and 49.02%, respectively), demonstrating that the method can preserve knowledge in certain scenarios.
|
| 88 |
+
|
| 89 |
+
- **Weaknesses**:
|
| 90 |
+
- Retain class accuracy is low overall, with several classes scoring below 40%. This suggests that the method struggles to maintain performance on the retained classes.
|
| 91 |
+
- High forget class loss for certain classes (e.g., "Deer" and "Bird") indicates that the exclusion process is more resource-intensive for these classes, which may reflect class-specific feature overlaps or shared characteristics.
|
| 92 |
+
|
| 93 |
+
- **Future Work**:
|
| 94 |
+
- Explore adaptive strategies to improve retain class accuracy and reduce loss, particularly for classes like "Truck" and "Ship."
|
| 95 |
+
- Investigate why certain classes, such as "Deer" and "Bird," have significantly higher forget class loss and optimize the exclusion process for such challenging classes.
|
| 96 |
+
- Test the **AdvNegGrad method** on other datasets and architectures to evaluate its generalizability and identify consistent patterns in performance.
|
| 97 |
+
|