Marco Rossi, “Multi-Head Attention-Based OOM Risk Prediction for CNN and Transformer Training Tasks on GPUMemNet”, AIMLR, vol. 7, no. 2, pp. 148–163, May 2026, doi: 10.69987/AIMLR.2026.70211.