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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Language Understanding | MMLU | MMLU Accuracy53.9 | 147 | |
| Language Understanding | MMLU-Pro | Accuracy26.2 | 116 | |
| Code Generation | HumanEval | Accuracy39 | 115 | |
| Factuality | TruthfulQA | Accuracy47.9 | 97 | |
| Science Question Answering | GPQA | Accuracy25.6 | 69 | |
| Language Understanding | MMLU | Delta Accuracy37.28 | 30 | |
| Language Reasoning | BBH (BIG-Bench Hard) | Average BBH Score47.3 | 20 | |
| Multi-task Language Understanding | MMLU STEM | Accuracy46 | 13 | |
| Language Generation | E2E | BLEU70.5 | 9 | |
| Mathematical Reasoning | OpenR1-Math-220k unseen | Accuracy46 | 6 |