Marco Rossi 2026. Multi-Head Attention-Based OOM Risk Prediction for CNN and Transformer Training Tasks on GPUMemNet. Journal of Artificial Intelligence Review. 7, 2 (May 2026), 148–163. DOI:https://doi.org/10.69987/.