Attentional-Biased Stochastic Gradient Descent

Qi Qi, Yi Xu, Wotao Yin, Rong Jin, Tianbao Yang

Abstract: In this paper, we present a simple yet effective provable method (named ABSGD) for addressing the data imbalance or label noise problem in deep learning. Our method is a simple modification to momentum SGD where we assign an individual importance weight to each sample in the mini-batch. The individual-level weight of a sampled data is systematically proportional to the exponential of a scaled loss value of the data, where the scaling factor is interpreted as the regularization parameter in the framework of distributionally robust optimization (DRO). Depending on whether the scaling factor is positive or negative, ABSGD is guaranteed to converge to a stationary point of an information-regularized min-max or min-min DRO problem, respectively. Compared with existing class-level weighting schemes, our method can capture the diversity between individual examples within each class. Compared with existing individual-level weighting methods using meta-learning that require three backward propagations for computing mini-batch stochastic gradients, our method is more efficient with only one backward propagation at each iteration as in standard deep learning methods. ABSGD is flexible enough to combine with other robust losses without any additional cost. Our empirical studies on several benchmark datasets demonstrate the effectiveness of the proposed method.

Submission Length: Long submission (more than 12 pages of main content)

Changes Since Last Submission: Modify the paper according to the AC Recommendations.

Code: https://github.com/qiqi-helloworld/ABSGD/

Assigned Action Editor: ~Changyou_Chen1

License: Creative Commons Attribution 4.0 International (CC BY 4.0)

Submission Number: 689