From Parameters to Feature Space: Task Arithmetic for Backdoor Mitigation in Model Merging
About
Model merging (MM) has gained significant attention as a cost-effective approach to integrate multiple task-specific models into a unified model. However, recent work reveals that MM is highly susceptible to backdoor attacks. Existing defenses based on task arithmetic often fail to eliminate backdoors without substantially degrading clean-task performance, owing to their reliance on direct parameter-space editing. To address this gap, we propose Linear Feature Path Minimization (LFPM), a backdoor mitigation framework for model merging, which introduces an anti-backdoor task vector into the backdoored merged model. Unlike prior approaches, LFPM formulates the backdoor robustness of the merged model from a unified feature-space perspective under the Cross-Task Linearity (CTL) framework, which leverages the approximate linearity of features across tasks. This perspective guides the optimization of the anti-backdoor task to suppress backdoors while preserving clean-task performance. Furthermore, we introduce an effective optimization mechanism based on gradient accumulation and loss path-integral, ensuring robust backdoor suppression along the interpolation path. Extensive experiments demonstrate that LFPM consistently exhibits strong robustness against backdoor attacks in both full fine-tuning and Parameter-Efficient Fine-Tuning (PEFT) settings.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Classification | GTSRB | CA33.28 | 135 | |
| Backdoor Defense | GTSRB | CA43.04 | 118 | |
| Image Classification | STL10 | Robustness0.73 | 30 | |
| Image Classification | Pets | Class Accuracy (CA)85.96 | 28 | |
| Image Classification | CARS196 | Accuracy55.15 | 21 | |
| Image Classification | SUN397 | CA64.88 | 14 | |
| Image Classification | EuroSAT | Classification Accuracy (CA)45.66 | 14 | |
| Backdoor Defense | SUN397 | Clean Accuracy65.44 | 7 | |
| Backdoor Defense in Model Merging | Cars196 Task Order 2 | Clean Accuracy (CA)50.21 | 7 | |
| Backdoor Defense in Model Merging | SVHN Task Order 2 | Clean Accuracy50.32 | 7 |