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