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

注意力引导课程学习在胸部X线片弱监督分类定位中的应用

Attention-Guided Curriculum Learning for Weakly Supervised Classification and Localization of Thoracic Diseases on Chest Radiographs

National Institutes of Health, Clinical Center(美国国立卫生研究院临床中心);Ping An Technology Co., Ltd.(平安科技有限公司)

Yuxing Tang, Xiaosong Wang, Adam P. Harrison, Le Lu, Jing Xiao, Ronald M. Summers

摘要:

在这项工作中,我们利用的任务,联合分类和弱监督定位胸部疾病的胸部X光片,只有图像级的疾病标签加上疾病严重程度(DSL)的信息的子集。提出了一种基于卷积神经网络(CNN)的注意力引导课程学习(AGCL)框架,该框架利用了从放射学报告中挖掘的严重程度属性。按照难度顺序(按不同的严重程度分组)的图像被馈送到CNN,以逐步提高学习。此外,提取高置信度样本(通过分类概率测量)及其相应的类条件热图(由CNN生成),并进一步馈送到AGCL框架中,以指导在下一次迭代中学习更独特的卷积特征。设计了一种双路径网络架构,除了原始分类任务之外,还可以从选定的种子样本中回归热图。联合学习方案可以提高分类和定位性能,同时为下一次迭代提供更多的种子样本。我们通过对公开的ChestXray 14数据集进行广泛的实验评估,证明了这种迭代细化框架的有效性。与现有技术相比,AGCL在分类AUC方面实现了超过5.7%(平均超过14种疾病)的增加,并且在定位任务的召回率/精确度方面实现了7%/11%的增加。

英文摘要:

In this work, we exploit the task of joint classification and weakly supervised localization of thoracic diseases from chest radiographs, with only image-level disease labels coupled with disease severity-level (DSL) information of a subset. A convolutional neural network (CNN) based attention-guided curriculum learning (AGCL) framework is presented, which leverages the severity-level attributes mined from radiology reports. Images in order of difficulty (grouped by different severity-levels) are fed to CNN to boost the learning gradually. In addition, highly confident samples (measured by classification probabilities) and their corresponding class-conditional heatmaps (generated by the CNN) are extracted and further fed into the AGCL framework to guide the learning of more distinctive convolutional features in the next iteration. A two-path network architecture is designed to regress the heatmaps from selected seed samples in addition to the original classification task. The joint learning scheme can improve the classification and localization performance along with more seed samples for the next iteration. We demonstrate the effectiveness of this iterative refinement framework via extensive experimental evaluations on the publicly available ChestXray14 dataset. AGCL achieves over 5.7\% (averaged over 14 diseases) increase in classification AUC and 7%/11% increases in Recall/Precision for the localization task compared to the state of the art.

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

↑