Alignment Attention by Matching Key and Query Distributions - Zhang, Shujian; Fan, Xinjie; Zheng, Huangjie; Tanwisuth, Korawat; Zhou, Mingyuan (Advances in Neural Information Processing Systems, 2021)