Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Ying-Hua Huang, Rui Fang, Hsi-Wen Chen, Ming-Syan Chen• 2026

Related benchmarks

TaskDatasetResultRank
Membership Inference AttackNYU V2
AUC98.58
90
Semantic segmentationNYU v2 (val)
mIoU75.41
75
Depth EstimationNYU v2 (val)--
65
Semantic segmentationNYU v2 (Retained set)
mIoU93.23
37
Multi-task Unlearning InterferenceNYU V2
UIS9.1
34
Surface Normal EstimationNYU v2 (Retained set)
A3063.21
33
Depth EstimationNYU v2 (Retained set)
Acc (sigma 1.25)85.56
33
Depth EstimationNYU Forget set v2 (train)
Accuracy (σ1.25)87.57
30
Semantic segmentationNYUv2 Forget set (train)
mIoU94.63
30
Surface Normal PredictionNYU v2 (val)
A30 Accuracy55.62
30
Showing 10 of 12 rows

Other info

Follow for update