Hanshu Rao
Author directory2026
AIA2: Attribute-Agnostic Imbalance Augmentation for Subgroup Robustness
Hanshu Rao | Guangzeng Han | Xiaolei Huang
Findings of the Association for Computational Linguistics: EMNLP 2026
Hanshu Rao | Guangzeng Han | Xiaolei Huang
Findings of the Association for Computational Linguistics: EMNLP 2026
Attributes describing data content and context can induce diverse imbalance patterns that go beyond label imbalance alone. However, existing studies primarily address label imbalance while overlooking data attributes, such as topics and demographics, which can induce meaningful subgroup structure while causing model degradation on underrepresented subgroups. We propose Attribute-Agnostic Imbalance Augmentation (AIA2), a framework for improving model robustness under varying subgroup imbalances without explicit subgroup annotations. AIA2 automatically discovers varying imbalances via latent semantic distributions, obtains slices with both learning difficulty and subgroup imbalance deficits, and deploys a large language model (LLM) for subgroup-aware imbalance augmentation. We have evaluated AIA2 on 5 popular corpora with rich domains and their attribute values, covering social issues and diverse topics. Results show improved performance on the lowest-performing subgroups and consistent gains over competitive baselines. Ablation studies confirm complementary contributions from each component, and additional analyses show that AIA2 provides a practical and consistent way to improve worst-group robustness under data subgroup imbalance. Code is available at https://github.com/trust-nlp/AIA2-Subgroup-Robustness.
Model-Agnostic Meta Learning for Class Imbalance Adaptation
Hanshu Rao | Guangzeng Han | Xiaolei Huang
Findings of the Association for Computational Linguistics: ACL 2026
Hanshu Rao | Guangzeng Han | Xiaolei Huang
Findings of the Association for Computational Linguistics: ACL 2026
Class imbalance is a widespread challenge in NLP tasks, significantly hindering robust performance across diverse domains and applications. We introduce Hardness-Aware Meta-Resample (HAMR), a unified framework that adaptively addresses both class imbalance and data difficulty. HAMR employs bi-level optimizations to dynamically estimate instance-level weights that prioritize genuinely challenging samples and minority classes, while a neighborhood-aware resampling mechanism amplifies training focus on hard examples and their semantically similar neighbors. We validate HAMR on six imbalanced datasets covering multiple tasks and spanning biomedical, disaster response, and sentiment domains. Experimental results show that HAMR achieves substantial improvements for minority classes and consistently outperforms strong baselines. Extensive ablation studies demonstrate that our proposed modules synergistically contribute to performance gains and highlight HAMR as a flexible and generalizable approach for class imbalance adaptation.