mlsanityData to 14 Oct 2026

ReForge: Refining Merged Models with Anchor-Regularized Regression

arXiv 2605.12843 ↗First posted 13 May 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Model merging aims to combine multiple task-specific expert models into a single model without joint retraining, offering a practical alternative to multi-task learning when data access or computational budget is limited. Existing model merging methods rarely exploit strong merged models as priors for further improvement. To address this limitation, we propose ReForge, a bilevel optimization framework that formulates module-wise refinement as Bayesian linear regression with an anchor-centered prior. The inner level yields a closed-form MAP estimate from unlabeled calibration activations. The outer level uses Bayesian optimization to jointly select heterogeneous regularization strengths and assembly scales using held-out validation data. Furthermore, we develop a data-free variant of ReForge that replaces activation statistics with task-vector Grams, eliminating the need for calibration examples. Across extensive benchmarks, including up to 20-task merging in vision and 5-task merging in language, ReForge consistently outperforms all evaluated plug-and-play anchor baselines (e.g., TA, WUDI-Merging, and TSV). On 20-task ViT-B/32, ReForge improves the strongest evaluated baseline, ISO-CTS, from 77.6% to 82.8% in the data-assisted setting and to 81.5% in the data-free setting. On eight-task ViT-L/14, the data-assisted variant achieves 95.1% mean accuracy, compared with 95.8% for the individual task experts. Our source code will be released soon.

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

Builds on

33 of 56 references.

References · 33

  1. 2017On Calibration of Modern Neural Networkscited by 4,960
  2. 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,349
  3. 2021Program Synthesis with Large Language Modelscited by 3,637
  4. 2017EuroSAT: A Novel Dataset and Deep Learning Benchmark for Land Use and Land Cover Classificationcited by 1,496
  5. 2024HarmBench: A Standardized Evaluation Framework for Automated Red Teaming and Robust Refusalcited by 1,429
  6. 2023Is Your Code Generated by ChatGPT Really Correct? Rigorous Evaluation of Large Language Models for Code Generationcited by 1,391
  7. 2023Instruction-Following Evaluation for Large Language Modelscited by 1,228
  8. 2022Editing Models with Task Arithmeticcited by 1,120
  9. 2024Tulu 3: Pushing Frontiers in Open Language Model Post-Trainingcited by 1,053
  10. 2023TIES-Merging: Resolving Interference When Merging Modelscited by 818
  11. 2017An Overview of Multi-Task Learning in Deep Neural Networkscited by 803
  12. 2023Language Models are Super Mario: Absorbing Abilities from Homologous Models as a Free Lunchcited by 694
21 more
  1. 2023XSTest: A Test Suite for Identifying Exaggerated Safety Behaviours in Large Language Modelscited by 505
  2. 2024WildGuard: Open One-Stop Moderation Tools for Safety Risks, Jailbreaks, and Refusals of LLMscited by 496
  3. 2018Rotation Equivariant CNNs for Digital Pathologycited by 422
  4. 2018Deep Learning for Classical Japanese Literaturecited by 422
  5. 2022Dataless Knowledge Fusion by Merging Weights of Language Modelscited by 387
  6. 2013Challenges in Representation Learning: A report on three machine learning contestscited by 302
  7. 2023Magicoder: Empowering Code Generation with OSS-Instructcited by 256
  8. 2023AdaMerging: Adaptive Model Merging for Multi-Task Learningcited by 252
  9. 2020Being Bayesian, Even Just a Bit, Fixes Overconfidence in ReLU Networkscited by 231
  10. 2024WildTeaming at Scale: From In-the-Wild Jailbreaks to (Adversarially) Safer Language Modelscited by 223
  11. 2023Okapi: Instruction-tuned Large Language Models in Multiple Languages with Reinforcement Learning from Human Feedbackcited by 190
  12. 2024Aya Dataset: An Open-Access Collection for Multilingual Instruction Tuningcited by 177
  13. 2024Task Singular Vectors: Reducing Task Interference in Model Mergingcited by 147
  14. 2025No Task Left Behind: Isotropic Model Merging with Common and Task-Specific Subspacescited by 96
  15. 2024Parameter Competition Balancing for Model Mergingcited by 72
  16. 2025Whoever Started the Interference Should End It: Guiding Data-Free Model Merging via Task Vectorscited by 62
  17. 2024Localize-and-Stitch: Efficient Model Merging via Sparse Task Arithmeticcited by 50
  18. 2025Modeling Multi-Task Model Merging as Adaptive Projective Gradient Descentcited by 39
  19. 2024Knowledge Composition using Task Vectors with Learned Anisotropic Scalingcited by 37
  20. 2025MergeBench: A Benchmark for Merging Domain-Specialized LLMscited by 33
  21. 2026ACE-Merging: Data-Free Model Merging with Adaptive Covariance Estimationcited by 6

Built on by

Not cited yet.