Entropic Projection Alignment: Estimating, Explaining, and Improving Model Performance Under Distribution Shift
熵投影对齐:估计、解释和改进分布偏移下的模型性能
Salim I. Amoukou, Emanuele Albini, Tom Bewley, Saumitra Mishra, Manuela Veloso
AI总结 提出熵投影对齐(EPA)方法,通过匹配选定矩并最小化KL散度来对齐源分布与目标分布,从而统一解决分布偏移下的性能估计、解释和改进问题。
详情
- Comments
- Accepted at the 29th International Conference on Artificial Intelligence and Statistics (AISTATS 2026)
我们提出了一个统一框架,用于解决分布偏移的三个关键挑战:(1)估计模型在未标记目标域上的性能,(2)通过识别导致偏移的特征来解释偏移,以及(3)提高目标域性能。我们的方法,熵投影对齐(EPA),通过匹配精心选择的矩同时最小化与源分布的KL散度,将源分布与目标分布对齐。该公式为重要性权重提供了唯一的闭式解,通过隐式方差控制实现鲁棒性。借鉴领域适应理论,我们证明矩匹配足以实现可靠的估计和适应,避免了完全密度比恢复的需要。大量实验以及强有力的理论保证表明,EPA在提供显著计算效率的同时,始终优于最先进的基线方法。
We propose a unified framework for addressing three key challenges of distribution shift: (1) estimating a model's performance on an unlabeled target domain, (2) explaining the shift by identifying the features responsible, and (3) improving the target domain performance. Our method, Entropic Projection Alignment (EPA), aligns the source distribution to the target by matching carefully selected moments while simultaneously minimising the KL divergence from the source. This formulation yields a unique closed-form solution for importance weights, achieving robustness through implicit variance control. Drawing on domain adaptation theory, we establish that moment matching is sufficient for reliable estimation and adaptation, avoiding the need for full density ratio recovery. Extensive experiments, together with strong theoretical guarantees, demonstrate that EPA consistently outperforms state-of-the-art baselines while offering substantial computational efficiency.