arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2007.10859cs.CV
收藏

基于交叉注意网络的多标记胸部疾病图像分类

Multi-label Thoracic Disease Image Classification with Cross-Attention Networks

Singapore Management University(新加坡管理大学);South China University of Technology(华南理工大学);The University of Adelaide(阿德莱德大学);Salesforce Research Asia(赛富时亚洲研究院)

Congbo Ma, Hu Wang, Steven C. H. Hoi

摘要:

放射学图像的自动疾病分类已经成为支持临床诊断和治疗计划的有前途的技术。与一般的图像分类任务不同,真实世界的放射学图像分类任务明显更具挑战性,因为收集训练数据要昂贵得多,其中标记的数据在本质上是多标记的;并且来自简单类别的更严重的样本通常占主导地位;训练数据是高度类别不平衡的问题在实践中也存在。为了克服这些挑战,在本文中,我们提出了一种新的交叉注意网络(CAN)方案,用于从胸部X射线图像中自动进行胸部疾病分类,该方案可以有效地从数据中挖掘更有意义的表示,从而通过仅使用图像级注释的交叉注意来提高性能。我们还设计了一个新的损失函数,它超越了交叉熵损失,有助于交叉注意过程,并能够克服类间和类内易支配样本的不平衡。所提出的方法实现了国家的最先进的结果。

英文摘要:

Automated disease classification of radiology images has been emerging as a promising technique to support clinical diagnosis and treatment planning. Unlike generic image classification tasks, a real-world radiology image classification task is significantly more challenging as it is far more expensive to collect the training data where the labeled data is in nature multi-label; and more seriously samples from easy classes often dominate; training data is highly class-imbalanced problem exists in practice as well. To overcome these challenges, in this paper, we propose a novel scheme of Cross-Attention Networks (CAN) for automated thoracic disease classification from chest x-ray images, which can effectively excavate more meaningful representation from data to boost the performance through cross-attention by only image-level annotations. We also design a new loss function that beyond cross-entropy loss to help cross-attention process and is able to overcome the imbalance between classes and easy-dominated samples within each class. The proposed method achieves state-of-the-art results.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑