Neural Networks2022,Vol.15111.DOI:10.1016/j.neunet.2022.03.027

Dynamic Auxiliary Soft Labels for decoupled learning

Shen, Furao Zhao, Jian Wang, Yan Zhang, Yongshun
Neural Networks2022,Vol.15111.DOI:10.1016/j.neunet.2022.03.027

Dynamic Auxiliary Soft Labels for decoupled learning

Shen, Furao 1Zhao, Jian 2Wang, Yan 1Zhang, Yongshun1
扫码查看

作者信息

  • 1. State Key Lab Novel Software Technol,Nanjing Univ
  • 2. Sch Elect Sci & Engn,Nanjing Univ
  • 折叠

Abstract

The long-tailed distribution in the dataset is one of the major challenges of deep learning. Convolutional Neural Networks have poor performance in identifying classes with only a few samples. For this problem, it has been proved that separating the feature learning stage and the classifier learning stage improves the performance of models effectively, which is called decoupled learning. We use soft labels to improve the performance of the decoupled learning framework by proposing a Dynamic Auxiliary Soft Labels (DaSL) method. Specifically, we design a dedicated auxiliary network to generate auxiliary soft labels for the two different training stages. In the feature learning stage, it helps to learn features with smaller variance within the class, and in the classifier learning stage it helps to alleviate the overconfidence of the model prediction. We also introduce a feature-level distillation method for the feature learning, and improve the learning of general features through multi-scale feature fusion. We conduct extensive experiments on three long-tailed recognition benchmark datasets to demonstrate the effectiveness of our DaSL.(C) 2022 Elsevier Ltd. All rights reserved.

Key words

Neural network/Decoupled learning/Long-tailed

引用本文复制引用

出版年

2022
Neural Networks

Neural Networks

EISCI
ISSN:0893-6080
被引量1
参考文献量54
段落导航相关论文