Maximum Likelihood Estimation is All You Need for Well-Specified Covariate Shift
Not on the map. Why?
Abstract (arXiv)
A key challenge of modern machine learning systems is to achieve Out-of-Distribution (OOD) generalization -- generalizing to target data whose distribution differs from that of source data. Despite its significant importance, the fundamental question of ``what are the most effective algorithms for OOD generalization'' remains open even under the standard setting of covariate shift. This paper addresses this fundamental question by proving that, surprisingly, classical Maximum Likelihood Estimation (MLE) purely using source data (without any modification) achieves the minimax optimality for covariate shift under the well-specified setting. That is, no algorithm performs better than MLE in this setting (up to a constant factor), justifying MLE is all you need. Our result holds for a very rich class of parametric models, and does not require any boundedness condition on the density ratio. We illustrate the wide applicability of our framework by instantiating it to three concrete examples -- linear regression, logistic regression, and phase retrieval. This paper further complement the study by proving that, under the misspecified setting, MLE is no longer the optimal choice, whereas Maximum Weighted Likelihood Estimator (MWLE) emerges as minimax optimal in certain scenarios.
Key facts. Cited by 15 papers in the map (data through 14 Oct 2026). Builds on 13 of its 42 references that are in the map. First posted on arXiv 27 Nov 2023.
Builds on
13 of 42 references.
References · 13
- 2019Invariant Risk Minimizationcited by 1,854
- 2019Distributionally Robust Neural Networks for Group Shifts: On the Importance of Regularization for Worst-Case Generalizationcited by 1,247
- 2020In Search of Lost Domain Generalizationcited by 1,054
- 2021Towards Out-Of-Distribution Generalization: A Surveycited by 436
- 2020The Risks of Invariant Risk Minimizationcited by 262
- 2020Neural Unsupervised Domain Adaptation in NLP---A Surveycited by 144
- 2019Domain Adaptation with Asymmetrically-Relaxed Distribution Alignmentcited by 61
- 2021Near-Optimal Linear Regression under Distribution Shiftcited by 42
- 2022A new similarity measure for covariate shift with applications to nonparametric regressioncited by 33
- 2020Minimax Lower Bounds for Transfer Learning with Linear and One-hidden Layer Neural Networkscited by 30
- 2023Pseudo-Labeling for Kernel Ridge Regression under Covariate Shiftcited by 17
- 2023Importance-Weighted Offline Learning Done Rightcited by 15
Built on by
Citing papers per year: 2024: 7; 2025: 6; 2026: 2.
15 papers.
Sort:
By area:
- 2025When Shift Happens - Confounding Is to Blamecited by 5
- 2024Mitigating Covariate Shift in Misspecified Regression with Applications to Reinforcement Learningcited by 4
- 2024High-Dimensional Kernel Methods under Covariate Shift: Data-Dependent Implicit Regularizationcited by 4
- 2024Help Me Help You: The Aggregate Value of Source and Target Data in Transfer Learningcited by 4
- 2024On the design-dependent suboptimality of the Lassocited by 3
- 2024Distributionally robust risk evaluation with an isotonic constraintcited by 3
- 2025Bilevel optimization for learning hyperparameters: Application to solving PDEs and inverse problems with Gaussian processescited by 3
- 2024Trans-Glasso: A Transfer Learning Approach to Precision Matrix Estimationcited by 2
- 2025Representative, Informative, and De-Amplifying: Requirements for Robust Bayesian Active Learning under Model Misspecificationcited by 2
- 2025Reweighting Improves Conditional Risk Boundscited by 1
- 2025Experimental Design for Semiparametric Banditscited by 1
- 2025Mixed-Sample SGD: an End-to-end Analysis of Supervised Transfer Learningcited by 1
- 2026SMART Fine-tuning Factor Augmented Neural Lasso14 Apr 2026
- 2026Learning from Biased and Costly Data Sources: Minimax-optimal Data Collection under a Budget19 Feb 2026
- 2025Bilevel optimization for learning hyperparameters: Application to solving PDEs and inverse problems with Gaussian processes7 Oct 2025
- 2025Mixed-Sample SGD: an End-to-end Analysis of Supervised Transfer Learning6 Jul 2025
- 2025Experimental Design for Semiparametric Bandits16 Jun 2025
- 2025Representative, Informative, and De-Amplifying: Requirements for Robust Bayesian Active Learning under Model Misspecification9 Jun 2025
- 2025When Shift Happens - Confounding Is to Blame27 May 2025
- 2025Reweighting Improves Conditional Risk Bounds4 Jan 2025
- 2024Trans-Glasso: A Transfer Learning Approach to Precision Matrix Estimation23 Nov 2024
- 2024Help Me Help You: The Aggregate Value of Source and Target Data in Transfer Learning29 Aug 2024
- 2024Distributionally robust risk evaluation with an isotonic constraint9 Jul 2024
- 2024Learning When the Concept Shifts: Confounding, Invariance, and Dimension Reduction22 Jun 2024
- 2025When Shift Happens - Confounding Is to Blamecited by 5
- 2024High-Dimensional Kernel Methods under Covariate Shift: Data-Dependent Implicit Regularizationcited by 4
- 2024Help Me Help You: The Aggregate Value of Source and Target Data in Transfer Learningcited by 4
- 2024On the design-dependent suboptimality of the Lassocited by 3
- 2024Distributionally robust risk evaluation with an isotonic constraintcited by 3
- 2025Bilevel optimization for learning hyperparameters: Application to solving PDEs and inverse problems with Gaussian processescited by 3
- 2024Trans-Glasso: A Transfer Learning Approach to Precision Matrix Estimationcited by 2
- 2025Representative, Informative, and De-Amplifying: Requirements for Robust Bayesian Active Learning under Model Misspecificationcited by 2
- 2025Reweighting Improves Conditional Risk Boundscited by 1
- 2025Experimental Design for Semiparametric Banditscited by 1
- 2025Mixed-Sample SGD: an End-to-end Analysis of Supervised Transfer Learningcited by 1
- 2026Learning from Biased and Costly Data Sources: Minimax-optimal Data Collection under a Budgetcited by 1
12 most cited of 14 All 14 (JSON)
- 2026SMART Fine-tuning Factor Augmented Neural Lasso14 Apr 2026
- 2026Learning from Biased and Costly Data Sources: Minimax-optimal Data Collection under a Budget19 Feb 2026
- 2025Bilevel optimization for learning hyperparameters: Application to solving PDEs and inverse problems with Gaussian processes7 Oct 2025
- 2025Mixed-Sample SGD: an End-to-end Analysis of Supervised Transfer Learning6 Jul 2025
- 2025Experimental Design for Semiparametric Bandits16 Jun 2025
- 2025Representative, Informative, and De-Amplifying: Requirements for Robust Bayesian Active Learning under Model Misspecification9 Jun 2025
- 2025When Shift Happens - Confounding Is to Blame27 May 2025
- 2025Reweighting Improves Conditional Risk Bounds4 Jan 2025
- 2024Trans-Glasso: A Transfer Learning Approach to Precision Matrix Estimation23 Nov 2024
- 2024Help Me Help You: The Aggregate Value of Source and Target Data in Transfer Learning29 Aug 2024
- 2024Distributionally robust risk evaluation with an isotonic constraint9 Jul 2024
- 2024Learning When the Concept Shifts: Confounding, Invariance, and Dimension Reduction22 Jun 2024
12 newest of 14 All 14 (JSON)