Bridged Semantic Alignment for Zero-shot 3D Medical Image Diagnosis
About
3D medical images such as computed tomography are widely used in clinical practice, offering a great potential for automatic diagnosis. Supervised learning-based approaches have achieved significant progress but rely heavily on extensive manual annotations, limited by the availability of training data and the diversity of abnormality types. Vision-language alignment (VLA) offers a promising alternative by enabling zero-shot learning without additional annotations. However, we empirically discover that the visual and textural embeddings after alignment endeavors from existing VLA methods form two well-separated clusters, presenting a wide gap to be bridged. To bridge this gap, we propose a Bridged Semantic Alignment (BrgSA) framework. First, we utilize a large language model to perform semantic summarization of reports, extracting high-level semantic information. Second, we design a Cross-Modal Knowledge Interaction module that leverages a cross-modal knowledge bank as a semantic bridge, facilitating interaction between the two modalities, narrowing the gap, and improving their alignment. To comprehensively evaluate our method, we construct a benchmark dataset that includes 15 underrepresented abnormalities as well as utilize two existing benchmark datasets. Experimental results demonstrate that BrgSA achieves state-of-the-art performances on both public benchmark datasets and our custom-labeled dataset, with significant improvements in zero-shot diagnosis of underrepresented abnormalities.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Text-to-Image Retrieval | CT-RATE (val) | AUC82.9 | 15 | |
| Abnormality diagnosis | CT-RATE (internal val) | AUC82.9 | 13 | |
| Text-to-Image Retrieval | Rad-ChestCT (RC) (val) | AUC74.2 | 13 | |
| Abnormality diagnosis | Rad-ChestCT (external val) | AUC74.2 | 11 | |
| Report-to-volume retrieval | CT-RATE (test) | Recall@1010.1 | 9 | |
| Report-to-volume retrieval | AH-Chest n=63,378 | Recall@52 | 3 | |
| Diagnostic Classification | AH-Chest | AUC61.2 | 2 |