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), 148-163. https://doi.org/10.69987/