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

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.

Zhenqian Zhu, Yamin Hu, Yiya Diao, Weixiang Li, Haodong Li, Wenjian Luo• 2026

Related benchmarks

TaskDatasetResultRank
Image ClassificationGTSRB
CA33.28
135
Backdoor DefenseGTSRB
CA43.04
118
Image ClassificationSTL10
Robustness0.73
30
Image ClassificationPets
Class Accuracy (CA)85.96
28
Image ClassificationCARS196
Accuracy55.15
21
Image ClassificationSUN397
CA64.88
14
Image ClassificationEuroSAT
Classification Accuracy (CA)45.66
14
Backdoor DefenseSUN397
Clean Accuracy65.44
7
Backdoor Defense in Model MergingCars196 Task Order 2
Clean Accuracy (CA)50.21
7
Backdoor Defense in Model MergingSVHN Task Order 2
Clean Accuracy50.32
7
Showing 10 of 28 rows

Other info

Follow for update