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

  • 2018-07-19 16:54:16
  • Yuxing Tang, Xiaosong Wang, Adam P. Harrison, Le Lu, Jing Xiao, Ronald M. Summers
  • 4

Abstract

In this work, we exploit the task of joint classification and weaklysupervised localization of thoracic diseases from chest radiographs, with onlyimage-level disease labels coupled with disease severity-level (DSL)information of a subset. A convolutional neural network (CNN) basedattention-guided curriculum learning (AGCL) framework is presented, whichleverages the severity-level attributes mined from radiology reports. Images inorder of difficulty (grouped by different severity-levels) are fed to CNN toboost the learning gradually. In addition, highly confident samples (measuredby classification probabilities) and their corresponding class-conditionalheatmaps (generated by the CNN) are extracted and further fed into the AGCLframework to guide the learning of more distinctive convolutional features inthe next iteration. A two-path network architecture is designed to regress theheatmaps from selected seed samples in addition to the original classificationtask. The joint learning scheme can improve the classification and localizationperformance along with more seed samples for the next iteration. We demonstratethe effectiveness of this iterative refinement framework via extensiveexperimental evaluations on the publicly available ChestXray14 dataset. AGCLachieves over 5.7\% (averaged over 14 diseases) increase in classification AUCand 7%/11% increases in Recall/Precision for the localization task compared tothe state of the art.

 

Quick Read (beta)

loading the full paper ...