Test-time data augmentation$-$averaging the predictions of a machine learning
model across multiple augmented samples of data$-$is a widely used technique
that improves the predictive performance. While many advanced learnable data
augmentation techniques have emerged in recent years, they are focused on the
training phase. Such techniques are not necessarily optimal for test-time
augmentation and can be outperformed by a policy consisting of simple crops and
flips. The primary goal of this paper is to demonstrate that test-time
augmentation policies can be successfully learned too. We introduce greedy
policy search (GPS), a simple but high-performing method for learning a policy
of test-time augmentation. We demonstrate that augmentation policies learned
with GPS achieve superior predictive performance on image classification
problems, provide better in-domain uncertainty estimation, and improve the
robustness to domain shift.

本文介绍了一种名为贪心策略搜索（GPS）的高性能方法，可用于学习数据增强的策略，以提高机器学习模型的预测性能、领域内不确定性估计以及提高对领域转移的鲁棒性分析。