Domain Adaptation with Adaptive $f$-Divergence: Tighter Variational Representation and Generalization Bounds
Abstract
Lay Summary
Machine learning models often work well on data similar to their training data, but their accuracy can drop when used in a new setting, such as moving from synthetic images to real camera images or from one image collection to another. A common way to handle this problem is to make the training and new data look more similar to the model, but existing methods usually rely on one fixed way of measuring the difference between the two datasets. Our work builds a method that can learn a better measure of this difference while training. It also adjusts the type of measure it uses based on the data, instead of choosing it in advance. We provide mathematical analysis showing when this approach should reduce the risk of errors on the new data. In experiments on standard image adaptation benchmarks, the method improved accuracy over several strong baselines. This can help reduce the need to label large amounts of new data whenever a model is deployed in a changed environment.