共 51 条
Class Rectification Hard Mining for Imbalanced Deep Learning
被引:145
作者:
Dong, Qi
[1
]
Gong, Shaogang
[1
]
Zhu, Xiatian
[2
]
机构:
[1] Queen Mary Univ London, London, England
[2] Vis Semant Ltd, London, England
来源:
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV)
|
2017年
关键词:
SMOTE;
D O I:
10.1109/ICCV.2017.205
中图分类号:
TP18 [人工智能理论];
学科分类号:
081104 ;
0812 ;
0835 ;
1405 ;
摘要:
Recognising detailed facial or clothing attributes in images of people is a challenging task for computer vision, especially when the training data are both in very large scale and extremely imbalanced among different attribute classes. To address this problem, we formulate a novel scheme for batch incremental hard sample mining of minority attribute classes from imbalanced large scale training data. We develop an end-to-end deep learning framework capable of avoiding the dominant effect of majority classes by discovering sparsely sampled boundaries of minority classes. This is made possible by introducing a Class Rectification Loss (CRL) regularising algorithm. We demonstrate the advantages and scalability of CRL over existing state-of-the-art attribute recognition and imbalanced data learning models on two large scale imbalanced benchmark datasets, the CelebA facial attribute dataset and the X-Domain clothing attribute dataset.
引用
收藏
页码:1869 / 1878
页数:10
相关论文