Mamba4Net: Distilled Hybrid Mamba Large Language Models For Networking
Transformer-based large language models (LLMs) are increasingly being adopted in networking research to address domain-specific challenges. However, their quadratic time complexity and substantial model sizes often result in significant computational overhead and memory constraints, particularly in resource-constrained environments. Drawing inspiration from the efficiency and performance of the Deepseek-R1 model within the knowledge distillation paradigm, this paper introduces Mamba4Net, a novel cross-architecture distillation framework. Mamba4Net transfers networking-specific knowledge from transformer-based LLMs to student models built on the Mamba architecture, which features linear time complexity. This design substantially enhances computational efficiency compared to the quadratic complexity of transformer-based models, while the reduced model size further minimizes computational demands, improving overall performance and resource utilization. To evaluate its effectiveness, Mamba4Net was tested across three diverse networking tasks: viewport prediction, adaptive bitrate streaming, and cluster job scheduling. Compared to existing methods that do not leverage LLMs, Mamba4Net demonstrates superior task performance. Furthermore, relative to direct applications of transformer-based LLMs, it achieves significant efficiency gains, including a throughput 3.96 times higher and a storage footprint of only 5.48% of that required by previous LLM-based approaches. These results highlight Mamba4Net’s potential to enable the cost-effective application of LLM-derived knowledge in networking contexts. The source code is openly available to support further research and development.
Authors
- Long Xia (ORCID: https://orcid.org/0000-0002-1936-7617)
- Zhimin Yan (ORCID: https://orcid.org/0000-0002-0751-7374)
- Kai Lei (ORCID: https://orcid.org/0000-0001-9197-895X)
- Jingjing Wang (ORCID: https://orcid.org/0009-0003-5543-878X)
- Min Yang (ORCID: https://orcid.org/0000-0002-2319-8086)
- Yakun Ren
- Guo Yu
Institutions
- Lifetech Scientific (China) (CN)
- Peking University Shenzhen Hospital (CN)
- Shenzhen Technology University (CN)
Publication Details
- Published
- 2025-09-22
- DOI
- https://doi.org/10.1109/icnp65844.2025.11192368
- Primary Topic
- Natural Language Processing Techniques
- Type
- article
- Field-Weighted Citation Impact
- 0.00