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), pp. 148–163. doi:10.69987/AIMLR.2026.70211.