Learning Invariant Representations with a Nonparametric Nadaraya-Watson Head - Wang, Alan; Nguyen, Minh; Sabuncu, Mert (Advances in Neural Information Processing Systems, 2023)