English

XProtoNet: Diagnosis in Chest Radiography with Global and Local Explanations

Computer Vision and Pattern Recognition 2021-03-22 v1

Abstract

Automated diagnosis using deep neural networks in chest radiography can help radiologists detect life-threatening diseases. However, existing methods only provide predictions without accurate explanations, undermining the trustworthiness of the diagnostic methods. Here, we present XProtoNet, a globally and locally interpretable diagnosis framework for chest radiography. XProtoNet learns representative patterns of each disease from X-ray images, which are prototypes, and makes a diagnosis on a given X-ray image based on the patterns. It predicts the area where a sign of the disease is likely to appear and compares the features in the predicted area with the prototypes. It can provide a global explanation, the prototype, and a local explanation, how the prototype contributes to the prediction of a single image. Despite the constraint for interpretability, XProtoNet achieves state-of-the-art classification performance on the public NIH chest X-ray dataset.

Keywords

Cite

@article{arxiv.2103.10663,
  title  = {XProtoNet: Diagnosis in Chest Radiography with Global and Local Explanations},
  author = {Eunji Kim and Siwon Kim and Minji Seo and Sungroh Yoon},
  journal= {arXiv preprint arXiv:2103.10663},
  year   = {2021}
}

Comments

10 pages, 7 figures. Accepted to CVPR2021