mlsanityData to 14 Oct 2026

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

arXiv 2604.28109 ↗First posted 30 Apr 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

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.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 56 of its 71 references that are in the map. First posted on arXiv 30 Apr 2026.

Builds on

56 of 71 references.

References · 56

  1. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
  2. 2024The Llama 3 Herd of Modelscited by 14,715
  3. 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,372
  4. 2021Training Verifiers to Solve Math Word Problemscited by 10,109
  5. 2025Qwen3 Technical Reportcited by 8,820
  6. 2020End-to-End Object Detection with Transformerscited by 7,414
  7. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
  8. 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,899
  9. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
  10. 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,148
  11. 2021Program Synthesis with Large Language Modelscited by 3,637
  12. 2022A ConvNet for the 2020scited by 3,579
38 more
  1. 2017A Broad-Coverage Challenge Corpus for Sentence Understanding through Inferencecited by 2,682
  2. 2019HuggingFace's Transformers: State-of-the-art Natural Language Processingcited by 2,218
  3. 2013Describing Textures in the Wildcited by 2,125
  4. 2024Gemma 2: Improving Open Language Models at a Practical Sizecited by 1,998
  5. 2025Gemma 3 Technical Reportcited by 1,824
  6. 2022Challenging BIG-Bench Tasks and Whether Chain-of-Thought Can Solve Themcited by 1,807
  7. 2017EuroSAT: A Novel Dataset and Deep Learning Benchmark for Land Use and Land Cover Classificationcited by 1,496
  8. 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,396
  9. 2019MMDetection: Open MMLab Detection Toolbox and Benchmarkcited by 1,385
  10. 2022Editing Models with Task Arithmeticcited by 1,120
  11. 2023LIMA: Less Is More for Alignmentcited by 1,100
  12. 2018Neural Network Acceptability Judgmentscited by 958
  13. 2023TIES-Merging: Resolving Interference When Merging Modelscited by 818
  14. 2023A Simple and Effective Pruning Approach for Large Language Modelscited by 800
  15. 2017Remote Sensing Image Scene Classification: Benchmark and State of the Artcited by 797
  16. 2024Phi-4 Technical Reportcited by 711
  17. 2023Language Models are Super Mario: Absorbing Abilities from Homologous Models as a Free Lunchcited by 694
  18. 2019Linear Mode Connectivity and the Lottery Ticket Hypothesiscited by 637
  19. 2021Merging Models with Fisher-Weighted Averagingcited by 621
  20. 2017To prune, or not to prune: exploring the efficacy of pruning for model compressioncited by 603
  21. 2021Knowledge Neurons in Pretrained Transformerscited by 601
  22. 2018Loss Surfaces, Mode Connectivity, and Fast Ensembling of DNNscited by 588
  23. 2020Federated Learning with Matched Averagingcited by 564
  24. 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 471
  25. 2022Decoupled Knowledge Distillationcited by 388
  26. 2022Dataless Knowledge Fusion by Merging Weights of Language Modelscited by 387
  27. 2019On the Efficacy of Knowledge Distillationcited by 381
  28. 2020Parameter-Efficient Transfer Learning with Diff Pruningcited by 352
  29. 2018Essentially No Barriers in Neural Network Energy Landscapecited by 330
  30. 2023AdaMerging: Adaptive Model Merging for Multi-Task Learningcited by 252
  31. 2021Training Neural Networks with Fixed Sparse Maskscited by 214
  32. 2016Trained Ternary Quantizationcited by 208
  33. 2023TheoremQA: A Theorem-driven Question Answering datasetcited by 192
  34. 2023ZipIt! Merging Models from Different Tasks without Trainingcited by 179
  35. 2021Composable Sparse Fine-Tuning for Cross-Lingual Transfercited by 144
  36. 2020Knowledge Distillation Meets Self-Supervisioncited by 142
  37. 2024Twin-Merging: Dynamic Integration of Modular Expertise in Model Mergingcited by 116
  38. 2024EMR-Merging: Tuning-Free High-Performance Model Mergingcited by 110

50 most cited of 56 All 56 (JSON)

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025Task Vector Quantization for Memory-Efficient Model Merging
  2. 2026Escaping Low-Dimensional Overlap: Multi-Task Model Merging via High-Dimensional Sparse Disentanglement
  3. 2025Decomposing Task Vectors for Refined Model Editing
  4. 2025DisTaC: Conditioning Task Vectors via Distillation for Robust Model Merging
  5. 2024Efficient and Effective Weight-Ensembling Mixture of Experts for Multi-Task Model Merging
  6. 2025RanDeS: Randomized Delta Superposition for Multi-Model Compression
  7. 20251bit-Merging: Dynamic Quantized Merging for Large Language Models
  8. 2025MINGLE: Mixture of Null-Space Gated Low-Rank Experts for Test-Time Continual Model Merging