Dynamically adaptive adjustment loss function biased towards few‐class learning. Issue 2 (17th October 2022)

• Record Type: Journal Article
• Authors: Liu, Guoqi; Bai, Lu; Li, Junlin; Li, Xusheng; Ru, Linyuan; Chang, Baofang
• Is Part Of: IET image processing. Volume 17:Issue 2(2023)
• Page Start: 627
• Page End: 635
• Publication Date: 2022-10-17
• Subjects: Image processing -- Periodicals; 621.36705
• DOI: 10.1049/ipr2.12661
• Languages: English
• ISSNs: 1751-9659

Abstract

Abstract: Convolution neural networks have been widely used in the field of computer vision, which effectively solve practical problems. However, the loss function with fixed parameters will affect the training efficiency and even lead to poor prediction accuracy. In particular, when there is a class imbalance in the data, the final result tends to favor the large‐class. In detection and recognition problems, the large‐class will dominate due to its quantitative advantage, and the features of few‐class can be not fully learned. In order to learn few‐class, batch nuclear‐norm maximization is introduced to the deep neural networks, and the mechanism of the adaptive composite loss function is established to increase the diversity of the network and thus improve the accuracy of prediction. The proposed loss function is added to the crowd counting, and verified on ShanghaiTech and UCF_CC_50 datasets. Experimental results show that the proposed loss function improves the prediction accuracy and convergence speed of deep neural networks.