Hybrid Transformer–LSTM Forecasting and PPO-Based Intelligent Kubernetes Auto-Scaling Framework for Cloud Data Science Pipelines Using Predictive Workload Analytics and Multi-Objective Performance Optimization
Article Sidebar
Main Article Content
With the rise of cloud-native data science pipelines, dynamic infrastructure is needed to cope with the variable nature of AI/ML workloads, however, traditional auto-scalers based on Kubernetes use reactive threshold mechanisms leading to inefficient resource utilization and service level agreement (SLA) violations. In this paper, we introduce a novel prediction-based framework with an integrated intelligent auto-scaler based on a Hybrid Transformer-LSTM forecasting model and a Proximal Policy Optimization (PPO) reinforcement learning (RL) agent. The forecasting engine leverages historical time series data on performance metrics, such as CPU, memory, GPU utilization, and request latency, to provide accurate workload predictions. These predictions are used by the RL agent to determine optimal scaling decisions, pod placement and resource allocation strategies based on multiple objectives including minimizing latency, maximizing throughput, energy savings and lowering cloud expenditures. In extensive evaluations performed using multiple node clusters performing distributed deep learning, stream processing and real-time inference pipelines, the proposed solution outperforms conventional HPA/VPA and baseline ML scalers by offering improved accuracy, speed of scaling, reducing unnecessary allo-cations by 38% and meeting strict SLA requirements for bursty workloads.
Downloads
References
Kubernetes Authors, “Horizontal and vertical pod autoscaling,” Kubernetes Official Documentation, 2024. [Online]. Available: https://kubernetes.io/docs/tasks/run-application/horizontal-pod-autoscale/
J. Schulman, F. Wolski, P. Dhariwal, A. Radford, and O. Klimov, “Prox-imal policy optimization algorithms,” arXiv preprint arXiv:1707.06347, 2017.
A. Vaswani et al., “Attention is all you need,” in Adv. Neural Inf. Process. Syst. (NeurIPS), vol. 30, 2017, pp. 5998–6008.
S. Hochreiter and J. Schmidhuber, “Long short-term memory,” Neural Comput., vol. 9, no. 8, pp. 1735–1780, Nov. 1997.
R. Gupta, M. Arora, and S. K. Panda, “Predictive cloud resource management using hybrid deep learning models,” IEEE Trans. Cloud Comput., vol. 11, no. 2, pp. 1452–1465, Apr.–Jun. 2023.
H. Mao, M. Alizadeh, I. Menache, and S. Kandula, “Resource manage-ment with deep reinforcement learning,” in Proc. 15th ACM Workshop Hot Topics Netw., Atlanta, GA, USA, 2016, pp. 50–56.
Y. Chen, L. Zhang, and X. Wang, “AI-driven Kubernetes scheduling for heterogeneous AI workloads,” Future Gener. Comput. Syst., vol. 134,
pp. 112–125, Sep. 2022.
Kubeflow Authors, “Kubeflow: Portable, scalable machine learn-ing on Kubernetes,” GitHub Repository, 2024. [Online]. Available: https://github.com/kubeflow/kubeflow
D. Sculley et al., “Hidden technical debt in machine learning systems,” in Adv. Neural Inf. Process. Syst. (NeurIPS), vol. 28, 2015, pp. 2503–2511.
T. P. Lillicrap et al., “Continuous control with deep reinforcement learning,” arXiv preprint arXiv:1509.02971, 2015.
M. Liao, J. Liu, and H. Wang, “Energy-aware container scheduling in cloud data centers using reinforcement learning,” J. Syst. Archit., vol. 138, p. 102745, Jul. 2023.
Amazon Web Services, “Amazon EKS best practices guide,” AWS Docu-mentation, 2024. [Online]. Available: https://aws.github.io/aws-eks-best-practices/
C. Amato, F. S. Melo, and M. Eger, “Reinforcement learning for dynamic resource allocation in cloud environments,” IEEE Trans. Serv. Comput., vol. 15, no. 4, pp. 2103–2116, Jul.–Aug. 2022.
D. Bahdanau, K. Cho, and Y. Bengio, “Neural machine translation by jointly learning to align and translate,” arXiv preprint arXiv:1409.0473, 2014.
M. Abadi et al., “TensorFlow: A system for large-scale machine learning,” in Proc. 12th USENIX Conf. Oper. Syst. Design Implement., Savannah, GA, USA, 2016, pp. 265–283.
MLflow Contributors, “MLflow documentation: Experiment tracking and model deployment,” 2024. [Online]. Available: https://www.mlflow.org/docs/latest/index.html
P. Stone, R. S. Sutton, and G. Kuhlmann, “Reinforcement learning for autonomous systems: A comprehensive survey,” IEEE Trans. Robot., vol. 38, no. 1, pp. 1–18, Feb. 2022.
K. Li, G. Xu, Y. Dai, and K. Hara, “Time series forecasting with Transformer-LSTM hybrid networks for cloud workload prediction,” J. Cloud Comput., vol. 12, no. 3, p. 45, Mar. 2023.
J. Dean and L. A. Barroso, “The tail at scale,” Commun. ACM, vol. 56, no. 2, pp. 74–80, Feb. 2013.
S. Dey, A. Patel, and R. Kumar, “Autonomous cloud orchestration: A survey of AI-driven Kubernetes management,” Future Gener. Comput. Syst., vol. 145, pp. 312–330, Aug. 2023.

This work is licensed under a Creative Commons Attribution 4.0 International License.
All articles published in our journal are licensed under CC-BY 4.0, which permits authors to retain copyright of their work. This license allows for unrestricted use, sharing, and reproduction of the articles, provided that proper credit is given to the original authors and the source.