XProtoNet: Diagnosis in Chest Radiography with Global and Local Explanations
About
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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Thoracic Disease Classification | NIH ChestX-ray14 (test) | AUROC82.2 | 44 | |
| Multi-Label Classification | NIH Chest X-ray | Atel AUC0.782 | 17 | |
| Aortic Stenosis Classification | Private dataset Cine-level (test) | bACC0.741 | 7 | |
| Aortic Stenosis Classification | Private dataset Study-level (N=258) (test) | bACC77.2 | 7 |