Unsupervised domain adaptation by learning using privileged information
Adam Breitholtz, Anton Matsson, Fredrik D. Johansson
TL;DR
DALUPI tackles unsupervised domain adaptation under limited input overlap by leveraging privileged information (PI) available only during training. It develops a two-stage estimator h = g \circ f and a practical end-to-end ROI-based variant to exploit PI types such as binary attributes and regions of interest in image data, with theoretical identifiability of the target risk and a PAC-style generalization perspective. The approach yields empirical gains across CelebA, a synthetic digit task, MS-COCO, and Chest X-ray classification, demonstrating improved adaptation, robustness to spurious source correlations, and enhanced sample efficiency, especially in small-sample regimes. These results underscore the practical value of training-time PI for domain transfer and motivate broader data collection of PI-annotated examples for robust UDA.
Abstract
Successful unsupervised domain adaptation is guaranteed only under strong assumptions such as covariate shift and overlap between input domains. The latter is often violated in high-dimensional applications like image classification which, despite this limitation, continues to serve as inspiration and benchmark for algorithm development. In this work, we show that training-time access to side information in the form of auxiliary variables can help relax restrictions on input variables and increase the sample efficiency of learning at the cost of collecting a richer variable set. As this information is assumed available only during training, not in deployment, we call this problem unsupervised domain adaptation by learning using privileged information (DALUPI). To solve this problem, we propose a simple two-stage learning algorithm, inspired by our analysis of the expected error in the target domain, and a practical end-to-end variant for image classification. We propose three evaluation tasks based on classification of entities in photos and anomalies in medical images with different types of available privileged information (binary attributes and single or multiple regions of interest). We demonstrate across these tasks that using privileged information in learning can reduce errors in domain transfer compared to baselines, be robust to spurious correlations in the source domain, and increase sample efficiency.
