Proceedings

DoMoE: Domain-Aware Semantic Expert Prediction for Efficient MoE Inference Under Expert Offloading

Download BibTeX Cite

Proceedings of the Thirty-Fifth International Joint Conference on Artificial Intelligence. Main Track. Pages 5901-5909.

https://doi.org/10.24963/ijcai.2026/657

Abstract

Mixture-of-Experts (MoE) large language models improve inference efficiency through sparse expert activation, but deployment on resource-constrained devices remains challenging due to the large expert parameter footprint. Expert offloading mitigates this issue by loading experts on demand, yet its effectiveness critically depends on accurate and efficient expert prediction: inaccurate predictions incur redundant expert transfers, while overly expensive predictors negate latency benefits. Existing trajectory-based methods suffer from low accuracy, whereas semantic-based approaches incur prohibitive similarity-matching overhead. We propose DoMoE, a domain-aware MoE inference system that exploits domain locality in inference workloads. DoMoE organizes routing information into domain-specific expert routing tables, restricts semantic matching to domain-relevant tokens and explicitly balances prediction accuracy against prediction overhead. Experiments show that DoMoE achieves a 1.32 X average throughput improvement and a 1.22 X increase in expert hit ratio across multiple MoE models and workloads, enabling efficient and accurate expert routing for practical inference.

  1. [Abdin et al., 2024] Marah Abdin, Jyoti Aneja, Hany Awadalla, Ahmed Awadallah, and Ammar Ahmad. Phi-3 technical report: A highly capable language model locally on your phone, 2024. [AlicanKiraz0, 2025] AlicanKiraz0. Agentic- chain-of-thought-coding-sft-dataset. https: //huggingface.co/datasets/AlicanKiraz0/ Agentic-Chain-of-Thought-Coding-SFT-Dataset, 2025. Accessed: 2026-05-25. Link
  2. [Casanueva et al., 2020] Iñigo Casanueva, Tadas Temcinas, Daniela Gerz, Matthew Henderson, and Ivan Vulic. Ef- ficient intent detection with dual sentence encoders. In Proceedings of the 2nd Workshop on NLP for Con- vAI -ACL 2020, mar 2020. Data available at https://github.com/PolyAI-LDN/task-specific-datasets. Link
  3. [Fieberg et al., 2024] Christian Fieberg, Lars Hornuf, David Streich, and Maximilian Meiler. Using large language models for financial advice. http://dx.doi.org/10.2139/ ssrn.4850039, 2024. DOI
  4. [Hadia et al., 2025] Aliyya Hadia, Tariq Afifa, and Fawzi Gamal. Moe at scale: From modular design to deploy- ment in large-scale machine learning systems. Preprints, April 2025.
  5. [He et al., 2023] Shwai He, Run-Ze Fan, Liang Ding, Li Shen, Tianyi Zhou, and Dacheng Tao. Merging experts into one: Improving computational efficiency of mixture of experts. arXiv preprint arXiv:2310.09832, 2023.
  6. [Hu et al., 2025a] Xing Hu, Zhixuan Chen, Dawei Yang, Zukang Xu, Chen Xu, Zhihang Yuan, Sifan Zhou, and Jiangyong Yu. Moequant: Enhancing quantization for mixture-of-experts large language models via expert- balanced sampling and affinity guidance. arXiv preprint arXiv:2505.03804, 2025.
  7. [Hu et al., 2025b] Yitao Hu, Xiulong Liu, Guotao Yang, Linxuan Li, Kai Zeng, Zhixin Zhao, Sheng Chen, Laiping Zhao, Wenxin Li, and Keqiu Li. Tightllm: Maximizing throughput for llm inference via adaptive offloading pol- icy. IEEE Transactions on Computers, 2025.
  8. [Hwang et al., 2024] Ranggi Hwang, Jianyu Wei, Shijie Cao, Changho Hwang, Xiaohu Tang, Ting Cao, and Mao Yang. Pre-gated moe: An algorithm-system co-design for fast and scalable mixture-of-expert inference. In 2024 ACM/IEEE 51st Annual International Symposium on Computer Architecture (ISCA), pages 1018-1031. IEEE, 2024.
  9. [Jiang et al., 2024] Albert Q. Jiang, Alexandre Sablay- rolles, Antoine Roux, Arthur Mensch, Blanche Savary, Chris Bamford, Devendra Singh Chaplot, Diego de las Casas, Emma Bou Hanna, Florian Bressand, Gianna Lengyel, Guillaume Bour, Guillaume Lample, Lélio Re- nard Lavaud, Lucile Saulnier, Marie-Anne Lachaux, Pierre Stock, Sandeep Subramanian, Sophia Yang, Szymon An- toniak, Teven Le Scao, Théophile Gervet, Thibaut Lavril, Thomas Wang, Timothée Lacroix, and William El Sayed. Mixtral of experts, 2024.
  10. [Kim et al., 2023] Young Jin Kim, Raffy Fahim, and Hany Hassan Awadalla. Mixture of quantized experts (moqe): Complementary effect of low-bit quantization and robustness. arXiv preprint arXiv:2310.02410, 2023.
  11. [Kong et al., 2025] Rui Kong, Yuanchun Li, Weijun Wang, Linghe Kong, and Yunxin Liu. Serving moe models on resource-constrained edge devices via dynamic expert swapping. IEEE Transactions on Computers, 2025.
  12. [Li et al., 2024] Pingzhi Li, Xiaolong Jin, Zhen Tan, Yu Cheng, and Tianlong Chen. Quantmoe-bench: Ex- amining post-training quantization for mixture-of-experts. arXiv preprint arXiv:2406.08155, 2024.
  13. [Lightman et al., 2023] Hunter Lightman, Vineet Kosaraju, Yura Burda, Harri Edwards, Bowen Baker, Teddy Lee, Jan Leike, John Schulman, Ilya Sutskever, and Karl Cobbe. Let's verify step by step. arXiv preprint arXiv:2305.20050, 2023.
  14. [Liu et al., 2025] Defu Liu, Yixiao Zhu, Zhe Liu, Yi Liu, Changlin Han, Jinkai Tian, Ruihao Li, and Wei Yi. A sur- vey of model compression techniques: past, present, and future. Frontiers in Robotics and AI, 2025.
  15. [Lu et al., 2024] Xudong Lu, Qi Liu, Yuhui Xu, Aojun Zhou, Siyuan Huang, Bo Zhang, Junchi Yan, and Hongsheng Li. Not all experts are equal: Efficient expert pruning and skipping for mixture-of-experts large language mod- els. arXiv preprint arXiv:2402.14800, 2024.
  16. [Ma et al., 2025] Songkai Ma, Zhaorui Zhang, Sheng Di, Benben Liu, Xiaodong Yu, Xiaoyi Lu, and Dan Wang. Compression error sensitivity analysis for different experts in moe model inference. In Proceedings of the SC'25 Workshops of the International Conference for High Per- formance Computing, Networking, Storage and Analysis, pages 339-348, 2025.
  17. [Maity and Saikia, 2025] Subhankar Maity and Manob Jyoti Saikia. Large language models in healthcare and medical applications: A review. Bioengineering (Basel, Switzer- land), 2025.
  18. [Masoudnia and Ebrahimpour, 2014] Saeed Masoudnia and Reza Ebrahimpour. Mixture of experts: a literature sur- vey. Artificial Intelligence Review, 2014.
  19. [Tang et al., 2024] Peng Tang, Jiacheng Liu, Xiaofeng Hou, Yifei Pu, Jing Wang, Pheng-Ann Heng, Chao Li, and Minyi Guo. Hobbit: A mixed precision expert of- floading system for fast moe inference. arXiv preprint arXiv:2411.01433, 2024.
  20. [Team, 2024] Qwen Team. Qwen1.5-moe: Matching 7b model performance with 1/3 activated parameters. https: //qwenlm.github.io/blog/qwen-moe/, 2024. Accessed: 2026-05-25. Link
  21. [Team, 2025] Qwen Team. Qwen3 technical report. https: //arxiv.org/abs/2505.09388, 2025. Accessed: 2026-05-25. Link
  22. [Xie et al., 2025] Haojie Xie, Yirong Chen, Xiaofen Xing, Jingkai Lin, and Xiangmin Xu. PsyDT: Using LLMs to construct the digital twin of psychological counselor with personalized counseling style for psychological counsel- ing. In Wanxiang Che, Joyce Nabende, Ekaterina Shutova, and Mohammad Taher Pilehvar, editors, Proceedings of the 63rd Annual Meeting of the Association for Com- putational Linguistics (Volume 1: Long Papers), pages 1081-1115. Association for Computational Linguistics, July 2025.
  23. [Xu et al., 2025] Fengyi Xu, Jun Ma, and Nan Li. Large lan- guage model applications in disaster management: An in- terdisciplinary review. International Journal of Disaster Risk Reduction, 127:105642, 2025.
  24. [Xu et al., 2026] Yunting Xu, Jiacheng Wang, Ruichen Zhang, Changyuan Zhao, Dusit Niyato, Jiawen Kang, Ze- hui Xiong, Bo Qian, Haibo Zhou, Shiwen Mao, Abbas Ja- malipour, Xuemin Shen, and Dong In Kim. Mixture of experts for decentralized generative ai and reinforcement learning in wireless networks: A comprehensive survey. IEEE Communications Surveys and Tutorials, 28:4051- 4085, 2026.
  25. [Xue et al., 2024] Leyang Xue, Yao Fu, Zhan Lu, Luo Mai, and Mahesh Marina. Moe-infinity: Offloading-efficient moe model serving. arXiv preprint arXiv:2401.14361, 2024.
  26. [Yang et al., 2024] Cheng Yang, Yang Sui, Jinqi Xiao, Lingyi Huang, Yu Gong, Yuanlin Duan, Wenqi Jia, Miao Yin, Yu Cheng, and Bo Yuan. Moe-i 2 : Compressing mixture of experts models through inter-expert pruning and intra-expert low-rank decomposition. arXiv preprint arXiv:2411.01016, 2024.
  27. [Yang et al., 2025] Aiqiang Yang, Jie Liu, Bo Yang, Zeyao Mo, and Keqin Li. Balancing communication overhead and accuracy in compression integration: a survey: A. yang et al. The Journal of Supercomputing, 81(8):964, 2025.
  28. [Yu et al., 2025] Hanfei Yu, Xingqi Cui, Hong Zhang, and Hao Wang. fmoe: Fine-grained expert offloading for large mixture-of-experts serving. arXiv preprint arXiv:2502.05370, 2025.
  29. [Yue et al., 2024] Shengbin Yue, Shujun Liu, Yuxuan Zhou, Chenchen Shen, Siyuan Wang, Yao Xiao, Bingxuan Li, Yun Song, Xiaoyu Shen, Wei Chen, et al. Lawllm: In- telligent legal system with legal reasoning and verifiable retrieval. In International Conference on Database Sys- tems for Advanced Applications, pages 304-321. Springer, 2024.
  30. [Yuksel et al., 2012] Seniha Esen Yuksel, Joseph N Wilson, and Paul D Gader. Twenty years of mixture of experts. IEEE transactions on neural networks and learning sys- tems, 23(8):1177-1193, 2012.
  31. [Zhang et al., 2025] Runhua Zhang, Hongxu Jiang, Wei Wang, and Jinhao Liu. Optimization methods, challenges, and opportunities for edge inference: A comprehensive survey. Electronics, 2025.
  32. [Zhong et al., 2024] Shuzhang Zhong, Ling Liang, Yuan Wang, Runsheng Wang, Ru Huang, and Meng Li. Adap- moe: Adaptive sensitivity-based expert gating and man- agement for efficient moe inference. In Proceedings of the 43rd IEEE/ACM International Conference on Computer- Aided Design, pages 1-9, 2024.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards