Machine Learning-Based Performance Modeling for Efficient Large Language Model Serving
Keywords:
machine learning, large language models, performance modeling, inference serving, system architecture, sustainability, governance, fairnessAbstract
The rapid proliferation of large language models has created unprecedented demands on computing infrastructure, necessitating sophisticated performance modeling techniques to ensure efficient serving. This paper examines the role of machine learning-based performance modeling in optimizing large language model serving systems. We analyze the architectural trade-offs inherent in serving these models, including latency, throughput, memory management, and energy consumption. The paper discusses how supervised and reinforcement learning approaches can predict inference delays, resource utilization, and quality-of-service metrics, enabling dynamic scheduling and resource allocation. We explore data requirements, feature engineering, and model selection challenges, emphasizing the need for robust, generalizable models that can adapt to heterogeneous hardware and evolving workloads. The discussion extends to deployment considerations, including edge-cloud continuum, sustainability, and governance frameworks. We highlight the importance of fairness, transparency, and policy interventions to mitigate risks of algorithmic bias and environmental impact. Through case illustrations and cross-domain comparisons, we demonstrate that machine learning-based performance modeling is not merely a technical optimization but a socio-technical infrastructure requiring careful governance. The paper concludes with forward-looking perspectives on automated orchestration, federated learning for performance prediction, and the need for standardized benchmarks and auditing mechanisms.
References
1. Kwon, W., Li, Z., Zhuang, S., Sheng, Y., Zheng, L., Yu, C. H., ... & Stoica, I. (2023). Efficient memory management for large language model serving with PagedAttention. In Proceedings of the 29th Symposium on Operating Systems Principles (pp. 611-626). ACM.
2. Yu, G. I., Jeong, J. S., Kim, G. W., Kim, S., & Chun, B. G. (2022). Orca: A distributed serving system for Transformer-based generative models. In 16th USENIX Symposium on Operating Systems Design and Implementation (OSDI 22) (pp. 521-538). USENIX Association.
3. Pope, R., Douglas, S., Chowdhery, A., Devlin, J., Bradbury, J., Heek, J., ... & Dean, J. (2023). Efficiently scaling transformer inference. Proceedings of Machine Learning and Systems, 5, 606-624.
4. Chen, L., Zaharia, M., & Zou, J. (2023). FrugalGPT: How to use large language models while reducing cost and improving performance. arXiv preprint arXiv:2305.05176.
5. Dao, T., Fu, D. Y., Ermon, S., Rudra, A., & Ré, C. (2022). FlashAttention: Fast and memory-efficient exact attention with IO-awareness. Advances in Neural Information Processing Systems, 35, 16344-16359.
6. Leviathan, Y., Kalman, M., & Matias, Y. (2023). Fast inference from transformers via speculative decoding. In International Conference on Machine Learning (pp. 19274-19286). PMLR.
7. Miao, X., Oliaro, G., Zhang, Z., Cheng, X., Wang, Z., Zhang, Z., ... & Jia, Z. (2023). SpecInfer: Accelerating generative large language model serving with tree-based speculative inference and verification. In Proceedings of the 29th ACM International Conference on Architectural Support for Programming Languages and Operating Systems (pp. 1-16). ACM.
8. Agrawal, A., Kedia, N., Panwar, A., Mohan, J., Kwatra, N., Gulavani, B. S., ... & Ramjee, R. (2024). Taming throughput-latency tradeoff in LLM inference with Sarathi-Serve. In 18th USENIX Symposium on Operating Systems Design and Implementation (OSDI 24) (pp. 117-134). USENIX Association.
9. Zhong, Y., Liu, S., Chen, J., Hu, J., Zhu, Y., Liu, X., ... & Li, D. (2024). Distributed inference and serving of large language models: A survey. IEEE Transactions on Parallel and Distributed Systems, 35(8), 1423-1441.
10. Zhang, J., & Gao, H. (2026, March). LLM inference delay prediction based on machine learning algorithms. In 2026 International Conference on Generative Artificial Intelligence and Information Security (GAIIS) (pp. 37-41). IEEE.
11. Crankshaw, D., Wang, X., Zhou, G., Franklin, M. J., Gonzalez, J. E., & Stoica, I. (2017). Clipper: A low-latency online prediction serving system. In 14th USENIX Symposium on Networked Systems Design and Implementation (NSDI 17) (pp. 613-627). USENIX Association.
12. Gujarati, A., Karimi, R., Alzayat, S., Hao, W., Kaufmann, A., Vigfusson, Y., & Mace, J. (2020). Serving DNNs like clockwork: Performance predictability from the bottom up. In 14th USENIX Symposium on Operating Systems Design and Implementation (OSDI 20) (pp. 443-462). USENIX Association.
13. Romero, F., Li, Q., Yadwadkar, N. J., & Kozyrakis, C. (2021). INFaaS: Automated model-less inference serving. In 2021 USENIX Annual Technical Conference (USENIX ATC 21) (pp. 397-411). USENIX Association.
14. Zhang, Y., Wang, Y., & Li, H. (2019). POS: A practical approach to estimate the inference latency of DNNs. arXiv preprint arXiv:1907.01098.
15. Chen, T., Moreau, T., Jiang, Z., Zheng, L., Yan, E., Shen, H.,Chen, T., Moreau, T., Jiang, Z., Zheng, L., Yan, E., Shen, H., Cowan, M., Wang, L., Hu, Y., Ceze, L., Guestrin, C., & Krishnamurthy, A. (2018). TVM: An automated end-to-end optimizing compiler for deep learning. In 13th USENIX Symposium on Operating Systems Design and Implementation (OSDI 18) (pp. 578-594). USENIX Association.
16. Schwartz, R., Dodge, J., Smith, N. A., & Etzioni, O. (2020). Green AI. Communications of the ACM, 63(12), 54-63.
17. Strubell, E., Ganesh, A., & McCallum, A. (2019). Energy and policy considerations for deep learning in NLP. In Proceedings of the 57th Annual Meeting of the Association for Computational Linguistics (pp. 3645-3650). Association for Computational Linguistics.
18. Bender, E. M., Gebru, T., McMillan-Major, A., & Shmitchell, S. (2021). On the dangers of stochastic parrots: Can language models be too big? In Proceedings of the 2021 ACM Conference on Fairness, Accountability, and Transparency (pp. 610-623). ACM.
19. Bommasani, R., Hudson, D. A., Adeli, E., Altman, R., Arora, S., von Arx, S., ... & Liang, P. (2021). On the opportunities and risks of foundation models. arXiv preprint arXiv:2108.07258.
20. Sculley, D., Holt, G., Golovin, D., Davydov, E., Phillips, T., Ebner, D., ... & Dennison, D. (2015). Hidden technical debt in machine learning systems. Advances in Neural Information Processing Systems, 28, 2503-2511.
Downloads
Published
Issue
Section
License
Copyright (c) 2026 Computer Science and Engineering Transactions

This work is licensed under a Creative Commons Attribution 4.0 International License.
