CheXtriev: Anatomy-Centered Representation for Case-Based Retrieval of Chest Radiographs
Organizations: Center for Visual Information Technology, International Institute of Information Technology Hyderabad, India
Abstract
We present CheXtriev, a graph-based, anatomy-aware framework for chest radiograph retrieval. Unlike prior methods focussed on global features, our method leverages graph transformers to extract informative features from specific anatomical regions. Furthermore, it captures spatial context and the interplay between anatomical location and findings. This contextualization, grounded in evidence-based anatomy, results in a richer anatomy-aware representation and leads to more accurate, effective and efficient retrieval, particularly for less prevalent findings. CheXtriv outperforms state-of-the-art global and local approaches by 18% to 26% in retrieval accuracy and 11% to 23% in ranking quality. The code is available at https://github.com/cvit-mip/chextriev.
Figures & tables
| Variants | IRM | Edge Connectivity | Location | MLF | mAP | mHR | mRR |
|---|---|---|---|---|---|---|---|
| V0 | ✕ | ✕ | ✕ | ✕ | 51.8 | 39.7 | 54.0 |
| V1 | GT | Shared Binary | ✕ | Global | 52.9 | 39.9 | 55.2 |
| V2 | GT | Shared Uniform | ✕ | Global | 53.9 | 40.8 | 56.3 |
| V3 | GT | Shared Uniform | Learnable | Global | 54.0 | 40.9 | 56.4 |
| V4 | GT | Unique | Learnable | Local | 51.8 | 39.8 | 53.7 |
| V5 | GT | Unique | ✕ | Global | 54.6 | 40.9 | 57.1 |
| Global CNN | ATH [ 1 ] | AnaXNet [ 2 ] | CheXtriev | |||||||||
| Findings | AP | HR | RR | AP | HR | RR | AP | HR | RR | AP | HR | RR |
| LO | 92.2 | 90.9 | 91.2 | 91.7 | 87.6 | 93.3 | ||||||
| PE | 72.6 | 66.1 | 67.2 | 71.4 | 59.4 | 74.7 | ||||||
| AT | 64.2 | 57.8 | 62.8 | 64.4 | 49.1 | 67.6 | ||||||
| ECS | 67.9 | 64.9 | 68.4 | 69.4 | 56.0 | 73.7 | ||||||
| PE/HO | 59.2 | 54.7 | 57.4 | 62.5 | 47.8 | 65.8 | ||||||
| Global CNN | ATH [ 1 ] | AnaXNet [ 2 ] | CheXtriev | |||||||||
| Findings | AP | HR | RR | AP | HR | RR | AP | HR | RR | AP | HR | RR |
| LO | 91.4 ∗ | 85.8 ∗ | 91.9 ∗ | 89.7 ∗ | 82.5 ∗ | 90.6 ∗ | 90.7 ∗ | 84.0 ∗ | 91.7 ∗ | 92.5 | 88.2 | 93.0 |
| PE | 70.0 ∗ | 56.2 ∗ | 71.2 | 62.6 ∗ | 47.6 ∗ | 63.9 ∗ | 64.8 ∗ | 48.6 ∗ | 65.7 ∗ | 72.1 | 59.4 | 73.0 |
| AT | 60.8 ∗ | 45.2 ∗ | 61.9 ∗ | 53.9 ∗ | 36.8 ∗ | 54.5 ∗ | 59.7 ∗ | 42.5 ∗ | 61.1 ∗ | 64.9 | 48.5 | 65.8 |
| ECS | 64.5 ∗ | 49.3 ∗ | 65.6 ∗ | 61.6 ∗ | 45.9 ∗ | 62.5 ∗ | 64.5 ∗ | 48.3 ∗ | 65.2 ∗ | 70.5 | 55.4 | 71.7 |
| PE/HO | 55.3 ∗ | 40.7 ∗ | 56.1 ∗ | 51.0 ∗ | 36.0 ∗ | 51.7 ∗ | 53.7 ∗ | 36.3 ∗ | 54.5 ∗ | 64.0 | 49.1 | 65.2 |
| Global CNN | ATH [ 1 ] | AnaXNet [ 2 ] | CheXtriev | |||||||||
| Findings | AP | HR | RR | AP | HR | RR | AP | HR | RR | AP | HR | RR |
| LO | 89.0 ∗ | 86.1 ∗ | 92.2 ∗ | 86.2 ∗ | 82.0 ∗ | 90.9 ∗ | 87.3 ∗ | 83.5 ∗ | 91.9 ∗ | 90.5 | 87.7 | 93.2 |
| PE | 64.4 ∗ | 54.1 ∗ | 73.2 | 58.2 ∗ | 45.2 ∗ | 67.2 ∗ | 59.3 ∗ | 46.4 ∗ | 68.9 ∗ | 67.6 | 57.5 | 75.1 |
| AT | 55.7 ∗ | 43.5 ∗ | 65.0 ∗ | 50.5 ∗ | 36.1 ∗ | 59.2 ∗ | 55.1 ∗ | 41.4 ∗ | 65.0 ∗ | 59.7 | 47.6 | 68.6 |
| ECS | 59.8 ∗ | 48.5 ∗ | 68.5 ∗ | 57.3 ∗ | 43.8 ∗ | 65.7 ∗ | 59.5 ∗ | 46.1 ∗ | 68.3 ∗ | 64.6 | 53.1 | 74.1 |
| PE/HO | 52.3 ∗ | 39.1 ∗ | 60.2 ∗ | 48.4 ∗ | 32.8 ∗ | 56.1 ∗ | 50.2 ∗ | 34.0 ∗ | 59.1 ∗ | 59.2 | 46.3 | 68.0 |
| Radiological Findings | #Radiographs |
|---|---|
| Lung Opacity | 2679 |
| Pleural Effusion | 1340 |
| Atelectasis | 1241 |
| Enlarged Cardiac Silhouette | 1212 |
| Pulmonary Edema / Hazy Opacity | 819 |
| Pneumonia | 643 |