Post-Hoc Merging is Not Enough: Many-Shot Model Merging with Loss-Gap Balancing
About
Model merging has become a practical post-training strategy for building a single multi-task large language model (LLM) by combining multiple task-specialized models. However, most existing approaches rely on post-hoc merging, in which task-specific models are merged only once after training. This one-shot aggregation often suffers from task interference, leading to information erasure across individual tasks. In this work, we show that replacing post-hoc merging with an iterative many-shot merging protocol is effective in improving multi-task performance. Building on this insight, we propose METIS, Mitigating Erasure from Task Interference for Stable many-shot merging. METIS is a loss-aware many-shot merging method that addresses information erasure in post-hoc merging through task-wise loss-gap weighting and consensus-based masking. Notably, METIS exhibits significant performance improvement on the worst-performing task, effectively mitigating information erasure. (Project page: https://imkyungjin.github.io/METIS/)
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Question Answering | PubMedQA | Accuracy72.8 | 40 | |
| Question Answering | CoQA | Exact Match71.4 | 24 | |
| Model Merging | 7-task NLP benchmark | Avg Performance1.18 | 20 | |
| General Language Understanding | Overall IFEval, GSM8K, M-MMLU, M-ARC, M-HellaSwag, XSTest | Average Normalized Performance1.015 | 13 | |
| Multilingual Understanding | m-ARC | Normalized Performance1.155 | 13 | |
| Safety | XSTest | Normalized Performance1.245 | 13 | |
| Instruction Following | IFEval | Normalized performance91.7 | 13 | |
| Mathematical Reasoning | GSM8K | Normalized performance0.872 | 13 | |
| Multilingual Understanding | M-Hellaswag | Performance0.991 | 13 | |
| Multilingual Understanding | m-MMLU | Normalized Performance0.91 | 13 |