A Novel Orthogonality Loss for Deep Hierarchical Multi-Task Learning

被引:3
作者
He, Guiqing [1 ]
Huo, Yincheng [1 ]
He, Mingyao [1 ]
Zhang, Haixi [1 ]
Fan, Jianping [2 ]
机构
[1] Northwestern Polytech Univ, Sch Elect & Informat, Xian 710072, Peoples R China
[2] Univ North Carolina Charlotte, Dept Comp Sci, Charlotte, NC 28223 USA
基金
中国国家自然科学基金;
关键词
Task analysis; Feature extraction; Convolutional neural networks; Vegetation; Object recognition; Birds; Loss measurement; Orthogonality loss; multi-task learning; orthogonal distribution regularization; FEATURE-EXTRACTION; IMAGE; ATTENTION; FEATURES;
D O I
10.1109/ACCESS.2020.2985991
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
In this paper, a novel loss function is proposed to measure the correlation among different learning tasks and select useful feature components for each classification task. Firstly, the knowledge map we proposed is used for organizing the affiliation relationship between objects in natural world. Secondly, a novel loss function & x2013;orthogonality loss is proposed to make the deep features more discriminative by removing useless feature components. Furthermore, in order to prevent the extracted feature maps from being too divergent and causing over-fitting which will reduce network performance, this paper also added the orthogonal distribution regularization term to constrain the distribution of network parameters. Finally, the proposed orthogonality loss is applied in a multi-task network structure to learn more discriminative deep feature, and also to evaluate the validity of the proposed loss function.The results show that compared with the traditional deep convolutional neural network and a multi-task network without orthogonality loss, the multi -task based orthogonality loss is significantly better than the other two types of networks on image classification.
引用
收藏
页码:67735 / 67744
页数:10
相关论文
共 59 条
[1]   Real-time head pose estimation using multi-task deep neural network [J].
Ahn, Byungtae ;
Choi, Dong-Geol ;
Park, Jaesik ;
Kweon, In So .
ROBOTICS AND AUTONOMOUS SYSTEMS, 2018, 103 :1-12
[2]  
[Anonymous], P CVPR WORKSH ITH NY
[3]  
[Anonymous], 2001, COMPUTER VISION
[4]  
[Anonymous], 2015, SIMILARITY BASED PAT
[5]  
[Anonymous], 2011, The caltech-ucsd birds-200-2011 dataset
[6]  
[Anonymous], 2015, PROC CVPR IEEE
[7]  
[Anonymous], 2015, MACH LEARN
[8]  
[Anonymous], ENCY SCI LEARNING
[9]  
[Anonymous], 2010, P ADV NEURAL INFORM
[10]  
[Anonymous], 2014, P 2 INT C LEARNING R