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

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/)

Kyungjin Im, Miru Kim, Chanin Eom, Minhae Kwon• 2026

Related benchmarks

TaskDatasetResultRank
Question AnsweringPubMedQA
Accuracy72.8
40
Question AnsweringCoQA
Exact Match71.4
24
Model Merging7-task NLP benchmark
Avg Performance1.18
20
General Language UnderstandingOverall IFEval, GSM8K, M-MMLU, M-ARC, M-HellaSwag, XSTest
Average Normalized Performance1.015
13
Multilingual Understandingm-ARC
Normalized Performance1.155
13
SafetyXSTest
Normalized Performance1.245
13
Instruction FollowingIFEval
Normalized performance91.7
13
Mathematical ReasoningGSM8K
Normalized performance0.872
13
Multilingual UnderstandingM-Hellaswag
Performance0.991
13
Multilingual Understandingm-MMLU
Normalized Performance0.91
13
Showing 10 of 14 rows

Other info

Follow for update