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

Language Model Networks: Supervision-Efficient Learning through Dense Communication

About

Language models are increasingly used not only as standalone predictors but also as components in larger inference systems, from test-time scaling to multi-agent collaboration. We study language model networks, where pre-trained language models serve as reusable nodes and intelligence emerges from their topology, communication, and optimization. Existing systems mostly communicate through natural language: easy to deploy, but discrete, inefficient, and hard to optimize from end-task supervision. We propose LMNet, a dense and differentiable realization of this paradigm. LMNet uses stripped LLMs as vertex modules and trainable seq2seq modules as communication edges, enabling intermediate nodes to exchange dense vectors while preserving natural-language input and output at the system boundary. By bypassing intermediate embedding and de-embedding, LMNet enables efficient information transfer, end-to-end gradient optimization, and learned communication beyond hand-designed protocols. Experiments show performance with small additional training cost and effective adaptation under limited supervision.

Shiguang Wu, Yaqing Wang, Quanming Yao• 2025

Related benchmarks

TaskDatasetResultRank
Language UnderstandingMMLU
MMLU Accuracy53.9
147
Language UnderstandingMMLU-Pro
Accuracy26.2
116
Code GenerationHumanEval
Accuracy39
115
FactualityTruthfulQA
Accuracy47.9
97
Science Question AnsweringGPQA
Accuracy25.6
69
Language UnderstandingMMLU
Delta Accuracy37.28
30
Language ReasoningBBH (BIG-Bench Hard)
Average BBH Score47.3
20
Multi-task Language UnderstandingMMLU STEM
Accuracy46
13
Language GenerationE2E
BLEU70.5
9
Mathematical ReasoningOpenR1-Math-220k unseen
Accuracy46
6
Showing 10 of 10 rows

Other info

Follow for update