Max-Margin Token Selection in Attention Mechanism - Ataee Tarzanagh, Davoud; Li, Yingcong; Zhang, Xuechen; Oymak, Samet (Advances in Neural Information Processing Systems, 2023)