Training Dynamics of Transformers to Recognize Word Co-occurrence via Gradient Flow Analysis - Yang, Hongru; Kailkhura, Bhavya; Wang, Zhangyang; Liang, Yingbin (Advances in Neural Information Processing Systems, 2024)