Clustering in Causal Attention Masking - Karagodin, Nikita; Polyanskiy, Yury; Rigollet, Philippe (Advances in Neural Information Processing Systems, 2024)