Marco Rossi. “Multi-Head Attention-Based OOM Risk Prediction for CNN and Transformer Training Tasks on GPUMemNet”. Journal of Artificial Intelligence Review 7, no. 2 (May 3, 2026): 148–163. Accessed October 11, 2026. https://learnedvertex.com/index.php/JAIR/article/view/421.