Why resampling outperforms reweighting for correcting sampling bias with stochastic gradients
Jing An, Lexing Ying, Yuhua Zhu
Abstract
A data set sampled from a certain population is biased if the subgroups of the population are sampled at proportions that are significantly different from their underlying proportions. Training machine learning models on biased data sets requires correction techniques to compensate for the bias. We consider two commonly-used techniques, resampling and reweighting, that rebalance the proportions of the subgroups to maintain the desired objective function. Though statistically equivalent, it has been observed that resampling outperforms reweighting when combined with stochastic gradient algorithms. By analyzing illustrative examples, we explain the reason behind this phenomenon using tools from dynamical stability and stochastic asymptotics. We also present experiments from regression, classification, and off-policy prediction to demonstrate that this is a general phenomenon. We argue that it is imperative to consider the objective function design and the optimization algorithm together while addressing the sampling bias.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 66901524-3a08-459d-afc8-087b10c2b3cbCited by top-tier papers14
- ABC: Auxiliary Balanced Classifier for Class-imbalanced Semi-supervised LearningHyuck Lee, Seungjae Shin, Heeyoung KimNeurIPS 2021 · 131 citations
- Smoothed Adaptive Weighting for Imbalanced Semi-Supervised Learning: Improve Reliability Against Unknown Distribution DataZhengfeng Lai, Chao Wang, Henrry Gunawan, Sen-Ching S. Cheung et al.ICML 2022 · 52 citations
- RLSbench: Domain Adaptation Under Relaxed Label ShiftSaurabh Garg, Nick Erickson, James Sharpnack, Alex Smola et al.ICML 2023 · 44 citations
- Dirichlet-based Per-Sample Weighting by Transition Matrix for Noisy Label LearningHeeSun Bae, Seungjae Shin, Byeonghu Na, Il-Chul MoonICLR 2024 · 10 citations
- Mitigating Spurious Correlations via Disagreement ProbabilityHyeonggeun Han, Sehwan Kim, Hyungjun Joo, Sangwoo Hong et al.NeurIPS 2024 · 8 citations
Builds on2
Related papers
- Fairness with Adaptive WeightsJunyi Chai, Xiaoqian WangICML 2022 · 47 citations
- Adaptive Sampling for Minimax Fair ClassificationShubhanshu Shekhar, Greg Fields, Mohammad Ghavamzadeh, Tara JavidiNeurIPS 2021 · 46 citations
- A Theoretical Analysis of the Learning Dynamics under Class ImbalanceEmanuele Francazi, Marco Baity-Jesi, Aurélien LucchiICML 2023 · 33 citations
- Group-robust Sample Reweighting for Subpopulation Shifts via Influence FunctionsRui Qiao, Zhaoxuan Wu, Jingtan Wang, Pang Wei Koh et al.ICLR 2025
- Stable Learning via Sample ReweightingZheyan Shen, Peng Cui, Tong Zhang, Kun KuangAAAI 2020 · 155 citations
