TEAS: TOKEN- AND ENERGY-AWARE AUTOSCALING FOR COST-EFFICIENT LLM SERVING

Authors

  • Dai Teng (Corresponding Author) University of Illinois Urbana-Champaign, Illinois, USA.

Keywords:

Large language models, Autoscaling, Inference serving, Energy efficiency, Cloud computing, Service level objectives

Abstract

Production platforms that serve large language models (LLMs) must continuously decide how many GPU replicas to keep online. Conventional autoscalers inherited from microservice stacks scale on request rate or coarse utilization, but LLM requests differ by up to three orders of magnitude in the number of tokens they process, arrive in extreme bursts, and run on replicas whose cold start takes minutes and whose idle power remains a large fraction of peak. We present TEAS (Token- and Energy-Aware Scaling), a horizontal autoscaler for LLM serving that (i) measures load in phase-weighted effective tokens rather than requests, (ii) forecasts load over the replica cold-start horizon with an empirical-residual safety margin, (iii) adapts its utilization headroom to the measured burstiness of the arrival process, and (iv) applies an energy-aware asymmetric hysteresis that delays scale-in until the projected idle-energy waste exceeds the energy cost of a replica restart. We evaluate TEAS in a trace-driven cluster simulator with a calibrated power and cost model, replaying real production traces from Azure LLM inference services alongside synthetic diurnal and composition-drift workloads derived from the real token distributions. Across all four workloads TEAS is the only dynamic policy that sustains at least 96% SLO attainment; on a bursty production code-assistant trace it improves attainment by 32 percentage points over a Kubernetes-HPA-style baseline and by 11 points over static peak provisioning, and on a 24-hour diurnal workload it meets a 95% SLO target at 11.6% lower monetary cost than peak provisioning. We release the simulator, policies, and experiment scripts.

References

[1] Kwon W, Li Z, Zhuang S, et al. Efficient memory management for large language model serving with PagedAttention. Proceedings of the 29th ACM Symposium on Operating Systems Principles (SOSP), 2023: 611-626.

[2] Yu GI, Jeong J S, Kim GW, et al. Orca: A distributed serving system for transformer-based generative models. Proceedings of the 16th USENIX Symposium on Operating Systems Design and Implementation (OSDI), 2022: 521-538.

[3] Patel P, Choukse E, Zhang C, et al. Splitwise: Efficient generative LLM inference using phase splitting. Proceedings of the 51st ACM/IEEE International Symposium on Computer Architecture (ISCA), 2024: 118-132.

[4] Patel P, Choukse E, Zhang C, et al. Characterizing power management opportunities for LLMs in the cloud. Proceedings of the 29th ACM International Conference on Architectural Support for Programming Languages and Operating Systems (ASPLOS), 2024: 207-222.

[5] Stojkovic J, Zhang C, Goiri Í, et al. DynamoLLM: Designing LLM inference clusters for performance and energy efficiency. Proceedings of the 31st IEEE International Symposium on High-Performance Computer Architecture (HPCA), 2025: 1348-1362.

[6] Qiu H, Mao W, Patke A, et al. Power-aware deep learning model serving with μ-Serve. Proceedings of the USENIX Annual Technical Conference (ATC), 2024: 75-93.

[7] Wang Y, Chen Y, Li Z, et al. BurstGPT: A real-world workload dataset to optimize LLM serving systems. arXiv preprint arXiv:2401.17644, 2024.

[8] Fu Y, Xue L, Huang Y, et al. ServerlessLLM: Low-latency serverless inference for large language models. Proceedings of the 18th USENIX Symposium on Operating Systems Design and Implementation (OSDI), 2024: 135-153.

[9] Griggs T, Liu X, Yu J, et al. Mélange: Cost efficient large language model serving by exploiting GPU heterogeneity. arXiv preprint arXiv:2404.14527, 2024.

[10] Zhong Y, Liu S, Chen J, et al. DistServe: Disaggregating prefill and decoding for goodput-optimized large language model serving. Proceedings of the 18th USENIX Symposium on Operating Systems Design and Implementation (OSDI), 2024: 193-210.

[11] Agrawal A, Kedia N, Panwar A, et al. Taming throughput-latency tradeoff in LLM inference with Sarathi-Serve. Proceedings of the 18th USENIX Symposium on Operating Systems Design and Implementation (OSDI), 2024: 117-134.

[12] Zhang C, Yu M, Wang W, et al. MArk: Exploiting cloud services for cost-effective, SLO-aware machine learning inference serving. Proceedings of the USENIX Annual Technical Conference (ATC), 2019: 1049-1062.

[13] Rzadca K, Findeisen P, Swiderski J, et al. Autopilot: Workload autoscaling at Google. Proceedings of the 15th European Conference on Computer Systems (EuroSys), 2020: 1-16.

[14] Gujarati A, Karimi R, Alzayat S, et al. Serving DNNs like Clockwork: Performance predictability from the bottom up. Proceedings of the 14th USENIX Symposium on Operating Systems Design and Implementation (OSDI), 2020: 443-462.

[15] Jaiswal S, Jain K, Simmhan Y, et al. SageServe: Optimizing LLM serving on cloud data centers with forecast aware auto-scaling. Proceedings of ACM Measurement and Analysis of Computing Systems (POMACS), 2025, 9(3): Article 61.

[16] Patke A, Reddy D, Jha S, et al. Hierarchical autoscaling for large language model serving with Chiron. arXiv preprint arXiv:2501.08090, 2025.

[17] Li Z, Zheng L, Zhong Y, et al. AlpaServe: Statistical multiplexing with model parallelism for deep learning serving. Proceedings of the 17th USENIX Symposium on Operating Systems Design and Implementation (OSDI), 2023: 663-679.

Downloads

Published

2025-12-29

How to Cite

Dai Teng. Teas: Token- And Energy-Aware Autoscaling For Cost-Efficient Llm Serving. AI and Data Science Journal. 2025, 2(3): 47-53. DOI: https://doi.org/10.61784/adsj03036.