Target-Aware Linear Regression Under Distribution Shift

Zhewen Hou, Tian Zheng

Distribution shift between training and deployment is a pervasive challenge for modern AI systems. In many cases, the target marginals of covariates and response are known or speci…