Marco Rossi. “Multi-Head Attention-Based OOM Risk Prediction for CNN and Transformer Training Tasks on GPUMemNet”. Artificial Intelligence and Machine Learning Review 7, no. 2 (May 3, 2026): 148–163. Accessed July 27, 2026. https://scipublication.com/index.php/AIMLR/article/view/421.