RT-Super: Learning Tumor Segmentation from Longitudinal Images and Reports
Organizations: Johns Hopkins University · Harvard Medical School · Massachusetts General Hospital · University of California, San Francisco · University of Zurich · Istanbul Medipol University · Johns Hopkins Medicine
Abstract
Multi-tumor segmentation is important for early cancer detection and allows radiologists to visualize, verify, and understand AI predictions. However, tumor segmentation masks are expensive, time-consuming, and unavailable for many tumor types in public data. Instead, hospitals have vast, readily available data that can guide segmentation: radiology reports, longitudinal images, and multi-phase images. We use this readily available data to substitute for tumor masks in training AI for tumor segmentation. To this end, we propose a new architecture, RT-Super. It has a teacher network, which analyzes the patient's longitudinal images and reports to create high-quality tumor masks. These masks train a student network, which sees a single image and no report. At inference, when longitudinal images and reports are unavailable, we use the student. RT-Super uses a new CNN-Transformer architecture and novel Consistency Losses that exploit tumor location consistency across longitudinal images. We train RT-Super to segment esophagus, uterus and spleen tumors, which have few or no public masks. Even without training masks, RT-Super can segment these tumors and surpass public AI models. Overall, we demonstrate that learning from longitudinal images, multi-phase images, and reports can overcome mask scarcity and advance multi-cancer detection and segmentation. Code: https://github.com/MrGiovanni/RT-Super
Figures & tables
| train | spleen | esophagus | uterus | average | |||||||||||
| model | longi. | report | mask | Se | Sp | F1 | Se | Sp | F1 | Se | Sp | F1 | Se | Sp | F1 |
| public AI models | |||||||||||||||
| Merlin [ 5 ] | x | 0 | 100 | 0 | 0 | 100 | 0 | 0 | 100 | 0 | 0 | 100 | 0 | ||
| ULS [ 13 ] | x | 28 | 89 | 34 | 5 | 98 | 10 | 32 | 85 | 42 | 22 | 91 | 29 | ||
| MedGemma [ 23 ] | x | 6 | 95 | 10 | 0 | 100 | 0 | 0 | 100 | 0 | 2 | 98 | 3 | ||
| trained on our dataset — report only | |||||||||||||||
| train | spleen | esophagus | uterus | average | |||||||||||||||
| model | longi. | report | mask | Se | Sp | AUC | DSC | Se | Sp | AUC | DSC | Se | Sp | AUC | DSC | Se | Sp | AUC | DSC |
| public AI models | |||||||||||||||||||
| Merlin [ 5 ] | x | 1 | 99 | 50 | - | 0 | 100 | 50 | - | 33 | 87 | 60 | - | 11 | 95 | 53 | - | ||
| ULS [ 13 ] | x | 58 | 51 | 53 | 4 | 59 | 87 | 74 | 11 | 50 | 74 | 66 | 12 | 56 | 71 | 64 | 9 | ||
| MedGemma [ 23 ] | x | 0 | 100 | 50 | - | 0 | 99 | 50 | - | 0 | 100 | 50 | - | 0 | 100 | 50 | - | ||
| trained on our dataset — report only | |||||||||||||||||||