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

InfiFusion: A Unified Framework for Enhanced Cross-Model Reasoning via LLM Fusion

About

We introduce InfiFusion, an efficient training pipeline designed to integrate multiple domain-specialized Large Language Models (LLMs) into a single pivot model, effectively harnessing the strengths of each source model. Traditional fusion methods either merge model parameters directly or rely on knowledge distillation with rigid assumptions, limiting their flexibility and efficiency. InfiFusion overcomes these limitations by enhancing Universal Logit Distillation (ULD) with Top-K selection and Logits Standardization. We propose two fusion strategies: Pairwise Fusion (InfiFusion$_p$), where each source model knowledge is distilled individually into the pivot model followed by merging and Unified Fusion (InfiFusion$_u$), where knowledge from all source models is distilled simultaneously into the pivot model. InfiFusion outperforms the state-of-the-art models, such as Qwen-2.5-14B-Instruct and Phi-4, across 11 widely applied benchmarks covering reasoning, coding, mathematics, and instruction-following tasks. Notably, InfiFusion achieves this superior performance while significantly reduces computational costs, completing full training with only 160 H800 GPU hours compared to the millions typically required for traditional LLM training.

Zhaoyi Yan, Yiming Zhang, Baoyi He, Yuhao Fu, Qi Zhou, Zhijie Sang, Chunlin Ji, Shengyu Zhang, Fei Wu, Hongxia Yang• 2025

Related benchmarks

TaskDatasetResultRank
ReasoningHellaSwag (HS)
HellaSwag Accuracy87.91
209
General ReasoningBBH
Accuracy80.94
190
General ReasoningMMLU
MMLU Accuracy85.81
180
CodingMBPP
Accuracy79.63
145
Instruction FollowingIFEval
Accuracy (IFEval)76.02
89
General ReasoningBIG-Bench Hard--
68
General ReasoningMMLU
Accuracy85.81
51
Natural Language ReasoningDROP
Accuracy89.27
43
ReasoningDROP
Score89.27
42
General ReasoningARC Challenge
ARC Score94.24
33
Showing 10 of 15 rows

Other info

Follow for update