mlsanityData to 14 Oct 2026

Maximum Likelihood Estimation is All You Need for Well-Specified Covariate Shift

arXiv 2311.15961 ↗First posted 27 Nov 2023stat.ML cs.LG math.STcited by 15

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

  1. 2019Invariant Risk Minimizationcited by 1,854
  2. 2019Distributionally Robust Neural Networks for Group Shifts: On the Importance of Regularization for Worst-Case Generalizationcited by 1,247
  3. 2020In Search of Lost Domain Generalizationcited by 1,054
  4. 2021Towards Out-Of-Distribution Generalization: A Surveycited by 436
  5. 2020The Risks of Invariant Risk Minimizationcited by 262
  6. 2020Neural Unsupervised Domain Adaptation in NLP---A Surveycited by 144
  7. 2019Domain Adaptation with Asymmetrically-Relaxed Distribution Alignmentcited by 61
  8. 2021Near-Optimal Linear Regression under Distribution Shiftcited by 42
  9. 2022A new similarity measure for covariate shift with applications to nonparametric regressioncited by 33
  10. 2020Minimax Lower Bounds for Transfer Learning with Linear and One-hidden Layer Neural Networkscited by 30
  11. 2023Pseudo-Labeling for Kernel Ridge Regression under Covariate Shiftcited by 17
  12. 2023Importance-Weighted Offline Learning Done Rightcited by 15
1 more
  1. 2022A Class of Geometric Structures in Transfer Learning: Minimax Bounds and Optimalitycited by 9

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2024: 7; 2025: 6; 2026: 2.

15 papers.

Sort:

By area: