Differentiable Top-k with Optimal Transport - Xie, Yujia; Dai, Hanjun; Chen, Minshuo; Dai, Bo; Zhao, Tuo; Zha, Hongyuan; Wei, Wei; Pfister, Tomas (Advances in Neural Information Processing Systems, 2020)