2 papers
cs.LG2026
Environment-Conditioned Tail Reweighting for Invariant Learning under Mixed Shifts
Yuanchao Wang, Zhao-Rong Lai, Tianqi Zhong +1
Out-of-distribution generalization becomes challenging when spurious correlations vary across environments while difficult or underrepresented samples remain insufficiently emphasi…
cs.LG2025
Out-of-distribution Generalization for Total Variation based Invariant Risk Minimization
Yuanchao Wang, Zhao-Rong Lai, Tianqi Zhong
Invariant risk minimization is an important general machine learning framework that has recently been interpreted as a total variation model (IRM-TV). However, how to improve out-o…