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

DLLG: Dynamic Logit-Level Gating of LLM Experts

About

Leveraging multiple specialized LLMs can combine complementary strengths, but existing approaches trade adaptability for stability: routing commits prematurely, heuristic ensembling depends on fragile proxies, and parameter merging introduces interference. We propose DLLG (Dynamic Logit-Level Gating), a dynamic logit-level ensembling framework that learns token-level expert fusion from sparse response-level supervision. A lightweight gating module predicts step-wise fusion weights, linking trajectory-level correctness to generation without token-level labels or expert retraining. Across diverse reasoning and code benchmarks, DLLG consistently outperforms strong routing, heuristic ensembling, and parameter-merging baselines across model scales, highlighting learned logit-level fusion as a robust and scalable paradigm for integrating specialized experts.

Bingnan Li, Zhaoyang Zhang, Xiaoze Liu, Yantao Shen, Shuli Jiang, Shuo Yang, Wei Xia, Zhuowen Tu, Stefano Soatto• 2026

Related benchmarks

TaskDatasetResultRank
Code GenerationHumanEval
pass@165.24
329
General ReasoningBBH
BBH General Reasoning Accuracy36.11
117
ReasoningBBH
BBH Score30.56
53
Code GenerationMBPP
Top-1 Acc.52.6
48
Mathematical ReasoningMATH
Overall Score43.4
43
Code GenerationBigCodeBench
pass@14.1
32
Code GenerationCode R1
Score9.55
14
Code GenerationCode R1
Pass@119.96
14
General Reasoning and Code GenerationCombined Benchmarks AVG
Average Score33.75
14
Mathematical ReasoningMinervaMath
Accuracy51.79
14
Showing 10 of 11 rows

Other info

Follow for update