DualOptim: Enhancing Efficacy and Stability in Machine Unlearning with Dual Optimizers
About
Existing machine unlearning (MU) approaches exhibit significant sensitivity to hyperparameters, requiring meticulous tuning that limits practical deployment. In this work, we first empirically demonstrate the instability and suboptimal performance of existing popular MU methods when deployed in different scenarios. To address this issue, we propose Dual Optimizer (DualOptim), which incorporates adaptive learning rate and decoupled momentum factors. Empirical and theoretical evidence demonstrates that DualOptim contributes to effective and stable unlearning. Through extensive experiments, we show that DualOptim can significantly boost MU efficacy and stability across diverse tasks, including image classification, image generation, and large language models, making it a versatile approach to empower existing MU algorithms.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Machine Unlearning | TOFU Forget05 Phi-1.5B model (5%) | Model Utility (MU)50.94 | 32 | |
| Machine Unlearning | TOFU (forget 5% data) | UFE97.65 | 24 | |
| Machine Unlearning | TOFU (forget 10% data) | UFE Score97.66 | 24 | |
| Machine Unlearning | TOFU (forget 1% data) | UFE97.46 | 24 | |
| Unlearning | TOFU Phi 1.5 1.0 (forget 1%) | UFE74.62 | 12 | |
| Unlearning | TOFU Llama 2 1.0 (forget 1%) | UFE92.02 | 12 | |
| Unlearning | TOFU Llama 2 1.0 (forget 5%) | UFE Score88.11 | 12 | |
| Unlearning | TOFU Llama 2 1.0 (forget 10%) | UFE85.14 | 12 | |
| Unlearning | TOFU Phi 1.5 1.0 (forget 10%) | UFE67.33 | 12 | |
| Model Utility | Alpaca-Llama 3 Utility Benchmarks | ARC-c Score47.36 | 7 |