Test-Time Training with Self-Supervision for Generalization under Distribution Shifts
arXiv (Cornell University)2019
Citations Over Time
Abstract
In this paper, we propose Test-Time Training, a general approach for improving the performance of predictive models when training and test data come from different distributions. We turn a single unlabeled test sample into a self-supervised learning problem, on which we update the model parameters before making a prediction. This also extends naturally to data in an online stream. Our simple approach leads to improvements on diverse image classification benchmarks aimed at evaluating robustness to distribution shifts.
Related Papers
- → Deep Residual Learning for Image Recognition(2016)216,943 cited
- → Dataset Shift in Machine Learning(2008)1,474 cited
- Test-Time Training for Out-of-Distribution Generalization(2019)
- → Test-Time Training with Self-Supervision for Generalization under\n Distribution Shifts(2019)199 cited