Bayesian Attention Modules - Fan, Xinjie; Zhang, Shujian; Chen, Bo; Zhou, Mingyuan (Advances in Neural Information Processing Systems, 2020)