ProME: Prototype-Margin Environments with Repair-Aware Selection for Group-Robust Learning

Qianqian Wang, Yunshan Li, Dawei Huang +2

Group-robust learning is crucial for maintaining accuracy on rare subpopulations when training-group labels are unavailable. However, existing methods often infer environments from…