中国科学院机构知识库网格
Chinese Academy of Sciences Institutional Repositories Grid
Distribution Unified and Probability Space Aligned Teacher-Student Learning for Imbalanced Visual Recognition

文献类型:期刊论文

作者Zhang, Shaoyu1,3; Chen, Chen1,3; Xie, Qiong1,3; Sun, Haigang2; Dong, Fei2; Peng, Silong1,3
刊名IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS FOR VIDEO TECHNOLOGY
出版日期2024-04-01
卷号34期号:4页码:2414-2425
关键词Class-imbalanced learning distribution mismatch probability space mismatch teacher-student learning
ISSN号1051-8215
DOI10.1109/TCSVT.2023.3311142
英文摘要

Imbalanced label distribution is usually the case for real-world data, which poses a challenge for training unbiased recognition model. In this paper, we study two underlying mismatches, i.e., distribution mismatch and probability space mismatch, present in class-imbalanced learning. Firstly, we analyze the label distribution mismatch between imbalanced training data and balanced test data, and introduce a distribution unified framework to unify the two distributions through probability conversion. Secondly, we analyze that the utilization of cross-entropy loss under the proposed framework may lead to probability space mismatch, where the conversion of the predictive probability is implemented in softmax probability space while the comparison with one-hot label is implemented in true probability space. To alleviate this dilemma, we involve a teacher model and formulate a teacher-student learning strategy, which contains two novel techniques. The Teacher Guided Label Smoothing (TGLS) is first proposed to relax the one-hot label to smoother pseudo softmax probability, which is more aligned with the softmax probability space. Additionally, we propose Distribution Unified Knowledge Distillation (DU-KD) under the proposed framework to further reduce both the mismatches. Experiments on several benchmarks confirm the top-level performance of the proposed method.

WOS研究方向Engineering
语种英语
WOS记录号WOS:001197960500075
出版者IEEE-INST ELECTRICAL ELECTRONICS ENGINEERS INC
源URL[http://ir.ia.ac.cn/handle/173211/57041]  
专题自动化研究所_智能制造技术与系统研究中心_多维数据分析团队
通讯作者Chen, Chen
作者单位1.Chinese Acad Sci, Inst Automat, Beijing 100190, Peoples R China
2.CNPC, GWDC Mud Logging Co, Panjin 124010, Peoples R China
3.School of Artificial Intelligence, University of Chinese Academy of Sciences, Beijing 100049, China
推荐引用方式
GB/T 7714
Zhang, Shaoyu,Chen, Chen,Xie, Qiong,et al. Distribution Unified and Probability Space Aligned Teacher-Student Learning for Imbalanced Visual Recognition[J]. IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS FOR VIDEO TECHNOLOGY,2024,34(4):2414-2425.
APA Zhang, Shaoyu,Chen, Chen,Xie, Qiong,Sun, Haigang,Dong, Fei,&Peng, Silong.(2024).Distribution Unified and Probability Space Aligned Teacher-Student Learning for Imbalanced Visual Recognition.IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS FOR VIDEO TECHNOLOGY,34(4),2414-2425.
MLA Zhang, Shaoyu,et al."Distribution Unified and Probability Space Aligned Teacher-Student Learning for Imbalanced Visual Recognition".IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS FOR VIDEO TECHNOLOGY 34.4(2024):2414-2425.

入库方式: OAI收割

来源:自动化研究所

浏览0
下载0
收藏0
其他版本

除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。