Marco Rossi. “Multi-Head Attention-Based OOM Risk Prediction for CNN and Transformer Training Tasks on GPUMemNet”. Journal of Artificial Intelligence Review, vol. 7, no. 2, May 2026, pp. 148-63, https://doi.org/10.69987/.