多标签分类
判别式
人工智能
计算机科学
模式识别(心理学)
特征(语言学)
相关性
机器学习
一致性(知识库)
约束(计算机辅助设计)
数学
哲学
语言学
几何学
作者
Kai Zhang,Wei Liang,Peng Cao,Xiaoli Liu,Jinzhu Yang,Osmar R. Zäıane
标识
DOI:10.1016/j.cmpb.2024.108032
摘要
Multi-label Chest X-ray (CXR) images often contain rich label relationship information, which is beneficial to improve classification performance. However, because of the intricate relationships among labels, most existing works fail to effectively learn and make full use of the label correlations, resulting in limited classification performance. In this study, we propose a multi-label learning framework that learns and leverages the label correlations to improve multi-label CXR image classification.In this paper, we capture the global label correlations through the self-attention mechanism. Meanwhile, to better utilize label correlations for guiding feature learning, we decompose the image-level features into label-level features. Furthermore, we enhance label-level feature learning in an end-to-end manner by a consistency constraint between global and local label correlations, and a label correlation guided multi-label supervised contrastive loss.To demonstrate the superior performance of our proposed approach, we conduct three times 5-fold cross-validation experiments on the CheXpert dataset. Our approach obtains an average F1 score of 44.6% and an AUC of 76.5%, achieving a 7.7% and 1.3% improvement compared to the state-of-the-art results.More accurate label correlations and full utilization of the learned label correlations help learn more discriminative label-level features. Experimental results demonstrate that our approach achieves exceptionally competitive performance compared to the state-of-the-art algorithms.
科研通智能强力驱动
Strongly Powered by AbleSci AI