arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.05938cs.LGcs.CV

GradTail: Learning Long-Tailed Data Using Gradient-based Sample Weighting

Waymo LLC(Waymo有限责任公司)

Zhao Chen, Vincent Casser, Henrik Kretzschmar, Dragomir Anguelov

英文摘要:

We propose GradTail, an algorithm that uses gradients to improve model performance on the fly in the face of long-tailed training data distributions. Unlike conventional long-tail classifiers which operate on converged - and possibly overfit - models, we demonstrate that an approach based on gradient dot product agreement can isolate long-tailed data early on during model training and improve performance by dynamically picking higher sample weights for that data. We show that such upweighting leads to model improvements for both classification and regression models, the latter of which are relatively unexplored in the long-tail literature, and that the long-tail examples found by gradient alignment are consistent with our semantic expectations.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑