Diagnosis by Composition: Detecting and Localizing Unseen Diseases by Composing Learned Radiological Findings
Organizations: University of California, Berkeley · Alibaba Group · Fudan University · Zhejiang University ·
Abstract
Deep models for radiological diagnosis are typically trained to predict disease labels directly from images, which limits them to diseases seen during training and makes their decisions hard to interpret in radiologists' terms. Radiology, however, proceeds from visual findings to disease through published guidelines that make this decision rule explicit, interpretable, and reproducible. We introduce diagnosis by composition (DbC), a representation learning framework that encodes this findings-first logic. DbC learns a vocabulary of radiological findings by aligning image features with the text embeddings of the findings, each finding represented as a dense map over the scan. A disease is defined by a set of required findings derived from its guidelines, and its diagnosis map is a fixed, parameter-free composition of the corresponding finding maps. Because the finding-to-disease mapping is derived from guidelines rather than learned, DbC is interpretable and auditable by design, can extend to new diseases by specifying their findings without retraining, and can diagnose unseen diseases by composing their findings. On multi-phase liver CT, DbC detects and localizes unseen diseases, rare diseases from a different institution, and an external cohort significantly better than all comparators, including models that align images with disease names. DbC also improves detection and localization in the fully supervised setting and remains superior across training-set sizes. Although demonstrated on liver CT, DbC applies wherever radiological guidelines define diseases through findings.
Figures & tables
| Localization | Detection | ||||||
|---|---|---|---|---|---|---|---|
| Model | Dice | IoU | Coverage | AP 50 | AUROC | AP | Sens.@ 95% Spec. |
| SegVol | |||||||
| SAT | |||||||
| VoxTell | |||||||
| Disease-V † | |||||||
| Disease-VL † | |||||||
| Localization | Detection | ||||||
|---|---|---|---|---|---|---|---|
| Unseen disease | Model | Dice | IoU | AP 50 | AUROC | AP | Sens.@ 95% Spec. |
| HCC | SegVol | ||||||
| SAT | |||||||
| VoxTell | |||||||
| Disease-VL † | |||||||
| DbC | |||||||
| Localization | Detection | ||||
|---|---|---|---|---|---|
| Rare disease | Model | Dice | IoU | AUROC | AP |
| Abscess | SegVol | ||||
| SAT | |||||
| VoxTell | |||||
| Disease-VL † | |||||
| DbC | |||||
| Localization | Detection | ||||
|---|---|---|---|---|---|
| Setup | Model | Dice | IoU | AUROC | AP |
| Zero-shot | SegVol | ||||
| SAT | |||||
| VoxTell | |||||
| Disease-VL † | |||||
| DbC | |||||
Appendix figures & tables9 assets
Supplementary material from the paper’s appendix.
Appendix
| Category | Finding | Phase | Required in |
| Density | low density | any | ICC, Meta, Heman |
| low density on NC | NC | HCC, FNH | |
| water density | any | Cyst | |
| very high density | any | Calc | |
| iso-density | any | – | |
| heterogeneous density | any | – |
| Disease | Diagnostic composition (required findings) |
|---|---|
| HCC | low density on NC non-rim arterial hyperenhancement non-peripheral washout † enhancing capsule † |
| ICC | low density rim arterial enhancement progressive delayed enhancement upstream biliary dilatation † infiltrative growth pattern † |
| Metastasis | low density rim arterial enhancement peripheral washout † ill-defined margin † |
| Hemangioma | low density peripheral nodular discontinuous enhancement † centripetal fill-in † progressive delayed enhancement well-defined margin |
| FNH | low density on NC non-rim arterial hyperenhancement isoenhancing portal phase † well-defined margin homogeneous enhancement † |
| Cyst | water density † no arterial enhancement † well-defined margin homogeneous density † round or oval shape † |
| Hyperparameter | Value |
|---|---|
| Hardware | 2 NVIDIA H200 GPUs, distributed data parallel |
| Micro-batch and accumulation | 2 CTs per GPU, 8 micro-batches |
| Effective batch | 32 CTs per optimizer step |
| Optimizer | AdamW [ Loshchilov and Hutter, 2019 ] , peak learning rate , weight decay 0.1 |
| Schedule | linear warm-up over the first 3% of optimizer steps, then constant |
| Gradient clipping | global norm 1.0 |
| Localization | Detection | ||||
|---|---|---|---|---|---|
| Target | Model | Dice | IoU | AUROC | AP |
| HCC 91 CT / 191 L | SegVol | ||||
| SAT | |||||
| VoxTell | |||||
| Disease-VL † | |||||
| DbC | |||||
| Localization | Detection | ||||
|---|---|---|---|---|---|
| Target | Model | Dice | IoU | AUROC | AP |
| HCC 91 CT / 191 L | SegVol | ||||
| SAT | |||||
| VoxTell | |||||
| Disease-VL † | |||||
| DbC | |||||
| Training fraction | |||||||
|---|---|---|---|---|---|---|---|
| Curve | Row | 10% | 20% | 40% | 60% | 80% | 100% |
| Detection AUROC all diseases | DbC | ||||||
| Disease-VL † | |||||||
| Significance | ns | ||||||
| Localization Dice all diseases | DbC | ||||||
| Disease-VL † | |||||||
Medical Image Analysis
Medical image analysis uses computational methods to interpret clinical images, measure anatomy and disease, and support diagnosis and treatment.
See all 300 papers