Test-Time Training with Self-Supervision for Generalization under Distribution Shifts
Test-Time Training with Self-Supervision for Generalization under Distribution Shifts
复制标题
DOI:
--
复制
发表时间:
2019-09
期刊:
影响因子:
--
通讯作者:
Yu Sun;X. Wang;Zhuang Liu;John Miller;Alexei A. Efros;Moritz Hardt
中科院分区:
文献类型:
--
作者:
Yu Sun;X. Wang;Zhuang Liu;John Miller;Alexei A. Efros;Moritz Hardt
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.