Tjayush commited on
Commit
802faba
Β·
verified Β·
1 Parent(s): c2b48c7

Add README with paper summary and results

Browse files
Files changed (1) hide show
  1. README.md +67 -0
README.md ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # UFUSC: Unified Federated Unlearning via Sensitivity-Guided Contrastive Forgetting
2
+
3
+ **A Novel Research Contribution in Machine Unlearning**
4
+
5
+ ## Paper
6
+
7
+ πŸ“„ **Title:** Sensitivity-Guided Contrastive Forgetting: Unified Label and Feature Unlearning in Vertical Federated Learning
8
+
9
+ This repository contains the complete research artifacts for UFUSC β€” the **first framework to simultaneously perform label AND feature unlearning in Vertical Federated Learning (VFL)**.
10
+
11
+ ## Key Innovation
12
+
13
+ Existing federated unlearning methods address either:
14
+ - **Label unlearning in VFL** (Manifold Mixup, arxiv:2410.10922) β€” forgets class labels but not features
15
+ - **Feature unlearning in HFL** (Ferrari, arxiv:2405.17462) β€” forgets features but in horizontal FL only
16
+
17
+ **UFUSC unifies both** through three innovations:
18
+ 1. 🎯 **Contrastive Forgetting Loss (CFL)** β€” Repels forget-set embeddings from class centroids while anchoring retain-set representations
19
+ 2. πŸ“ **Lipschitz Feature Sensitivity** β€” Minimizes model responsiveness to target features via perturbation-based sensitivity
20
+ 3. βœ… **Dual-Variable Certification** β€” Provides convergence-based forgetting guarantees via primal-dual optimization
21
+
22
+ ## Results Summary
23
+
24
+ | Dataset | Method | Retain Acc ↑ | Forget Acc ↓ | MIA ASR ↓ |
25
+ |---|---|---|---|---|
26
+ | MNIST | UFUSC-Joint | 85.45% | **13.00%** | **29.70%** |
27
+ | F-MNIST | UFUSC-Joint | 70.18% | **3.00%** | **19.10%** |
28
+ | CIFAR-10 | UFUSC-Joint | 50.96% | **4.80%** | **38.80%** |
29
+
30
+ UFUSC-Joint achieves the **lowest forget accuracy and MIA attack success rate** across all baselines.
31
+
32
+ ## Repository Contents
33
+
34
+ - `paper.md` β€” Full conference-ready research paper (NeurIPS/ICML style)
35
+ - `research_paper.py` β€” Complete self-contained implementation (baselines + UFUSC + experiments + visualization)
36
+ - `results/` β€” All experimental results in JSON format
37
+ - `figures/` β€” Publication-quality visualizations
38
+
39
+ ## Quick Start
40
+
41
+ ```bash
42
+ pip install torch torchvision numpy matplotlib seaborn pandas scikit-learn
43
+ python research_paper.py
44
+ ```
45
+
46
+ ## Citation
47
+
48
+ If you use this work, please cite both anchor papers:
49
+ ```bibtex
50
+ @article{bryan2024vfl_label_unlearning,
51
+ title={Towards Privacy-Guaranteed Label Unlearning in Vertical Federated Learning},
52
+ author={Bryan, H.X. et al.},
53
+ journal={arXiv preprint arXiv:2410.10922},
54
+ year={2024}
55
+ }
56
+
57
+ @article{ong2024ferrari,
58
+ title={Ferrari: Federated Feature Unlearning via Optimizing Feature Sensitivity},
59
+ author={Ong, W.K. et al.},
60
+ journal={arXiv preprint arXiv:2405.17462},
61
+ year={2024}
62
+ }
63
+ ```
64
+
65
+ ## License
66
+
67
+ This research is released for academic purposes. See paper for full details.