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

Auto-FlexSwitch: Efficient Dynamic Model Merging via Learnable Task Vector Compression

About

Model merging has attracted attention as an effective path toward multi-task adaptation by integrating knowledge from multiple task-specific models. Among existing approaches, dynamic merging mitigates performance degradation caused by conflicting parameter updates across tasks by flexibly combining task-specific parameters at inference time, thereby maintaining high performance. However, these methods require storing independent parameters for each task, resulting in prohibitive storage overhead. To address this issue, we first experimentally demonstrate that the fine-tuned weight increments (referred to as task vectors) exhibit an impulse-like activation pattern and high robustness to low-bit representations. Driven by this insight, we propose T-Switch, which decomposes task vectors into three compact components: a binary sparse mask, a sign vector, and a scalar scaling factor, achieving high-fidelity approximation at high compression ratios. We then introduce Auto-Switch, a training-free merging scheme that automatically composes task vectors via feature similarity retrieval. Building on this, we develop Auto-Switch, a training-free merging scheme that automatically assembles task vectors through feature similarity retrieval. Furthermore, to transform task vector sparsification and quantization from static rules to adaptive learning, we propose FlexSwitch, a learnable framework which jointly optimizes the compression strategy for each model unit via Learnable Gating Sparsification (LGS) and Bit-width Adaptive Selection (BAS), while employing the Sparsity-Aware Storage Strategy (SASS) to select the optimal storage encoding structure. Finally, by incorporating a K-Nearest Neighbor (KNN) inference scheme with a learnable low-rank metric, we present Auto-FlexSwitch, a dynamic model merging approach that supports highly efficient task vector compression.

Junqi Gao, Dazhi Zhang, Zhichang Guo, Biqing Qi, Yi Ran, Wangmeng Zuo• 2026

Related benchmarks

TaskDatasetResultRank
Image ClassificationDTD
Accuracy84.57
599
ClassificationCars
Accuracy87.61
492
Image ClassificationRESISC45
Accuracy95.92
472
Image ClassificationEuroSAT
Accuracy99.74
226
Image ClassificationSVHN
Top-1 Accuracy97.45
186
Image ClassificationStanford Cars
Top-1 Accuracy92.61
104
Image ClassificationGTSRB
Accuracy98.7
81
Image ClassificationMNIST
Accuracy99.74
70
Image ClassificationSUN397
Accuracy84.86
28
Image ClassificationSVHN
Accuracy98.12
22
Showing 10 of 15 rows

Other info

Follow for update