Interference-Aware Multi-Task Unlearning
About
Machine unlearning aims to remove the contribution of designated training data from a trained model while preserving performance on the remaining data. Existing work mainly focuses on single-task settings, whereas modern models often operate in multi-task setups with shared backbones, where removing supervision for one task or instance can unintentionally affect others. We introduce multi-task unlearning with two settings: full-task unlearning, which removes a target instance from all tasks, and partial-task unlearning, which removes supervision only from selected tasks. We show that shared parameters couple the forget and retain sets, causing task-level interference on non-target tasks and instance-level interference on other instances. To address this issue, we propose an interference-aware framework that combines task-aware gradient projection, which constrains updates within task-specific subspaces, with instance-level gradient orthogonalization, which reduces conflicts between forget and retain signals. Experiments on two multi-task computer vision benchmarks across five tasks show that our method achieves effective unlearning while maintaining strong generalization, reducing UIS compared with the strongest baseline by 30.3% in full-task unlearning and 52.9% in partial-task unlearning.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Membership Inference Attack | NYU V2 | AUC98.58 | 90 | |
| Semantic segmentation | NYU v2 (val) | mIoU75.41 | 75 | |
| Depth Estimation | NYU v2 (val) | -- | 65 | |
| Semantic segmentation | NYU v2 (Retained set) | mIoU93.23 | 37 | |
| Multi-task Unlearning Interference | NYU V2 | UIS9.1 | 34 | |
| Surface Normal Estimation | NYU v2 (Retained set) | A3063.21 | 33 | |
| Depth Estimation | NYU v2 (Retained set) | Acc (sigma 1.25)85.56 | 33 | |
| Depth Estimation | NYU Forget set v2 (train) | Accuracy (σ1.25)87.57 | 30 | |
| Semantic segmentation | NYUv2 Forget set (train) | mIoU94.63 | 30 | |
| Surface Normal Prediction | NYU v2 (val) | A30 Accuracy55.62 | 30 |