On the Optimization and Generalization of Multi-head Attention. - Puneesh Deora; Rouzbeh Ghaderi; Hossein Taheri; Christos Thrampoulidis (Trans. Mach. Learn. Res., 2024)