Metric Transforms and Low Rank Representations of Kernels for Fast Attention - Chu, Timothy; Alman, Josh; Miller, Gary; Narayanan, Shyam; Sellke, Mark; Song, Zhao (Advances in Neural Information Processing Systems, 2024)