Back to Search Start Over

Multi-task convolutional neural network with coarse-to-fine knowledge transfer for long-tailed classification.

Authors :
Li, Zhengyu
Zhao, Hong
Lin, Yaojin
Source :
Information Sciences. Aug2022, Vol. 608, p900-916. 17p.
Publication Year :
2022

Abstract

Long-tailed classifications make it very challenging to deal with class-imbalanced problems using deep convolutional neural networks (CNNs). Existing solutions based on re-balancing methods perform well and use single-task CNNs to train each fine-grained class independently. However, classification tasks are multiplex and involve a coarse-to-fine hierarchical relation. In this paper, we propose a coarse-to-fine knowledge transfer based multi-task CNN, which utilizes the coarse-to-fine structure to promote long-tailed learning. First, we construct a tail hierarchical structure in a coarse-to-fine way to pay greater attention to tail classes than head classes. Second, a multi-task CNN is adopted to simultaneously train coarse- and fine-grained tasks to extract a more generalized knowledge representation than the single-task CNN. Third, we design a coarse-to-fine knowledge transfer strategy to adaptively adjust the task weights to improve fine-grained performance. Extensive experiments on benchmark datasets show that our model achieves better gains than the re-balancing methods. In particular, the proposed model is 3.25% more accurate than the second-best method on the long-tailed tieredImageNet dataset. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
00200255
Volume :
608
Database :
Academic Search Index
Journal :
Information Sciences
Publication Type :
Periodical
Accession number :
159234454
Full Text :
https://doi.org/10.1016/j.ins.2022.07.015