Improving Transformers with Probabilistic Attention Keys. - Tam Minh Nguyen; Tan Minh Nguyen; Dung D. D. Le; Duy Khuong Nguyen; Viet-Anh Tran; Richard G. Baraniuk; Nhat Ho; Stanley J. Osher (ICML, 2022)