AbdomenAtlas/RSuperMaskPretrained

Scaling Artificial Intelligence for Multi-Tumor Early Detection with More Reports, Fewer Masks

This repository contains a segmentation model (MedFormer architecture) which served as a baseline for the paper Scaling Artificial Intelligence for Multi-Tumor Early Detection with More Reports, Fewer Masks.

This model was trained for organ and tumor (kidney, liver and pancreas tumors) segmentation in the Beta version of AbdomenAtlas 3.0. Performance improvements are expected for models trained on the released version of AbdomenAtlas 3.0. For the official release of AbdomenAtlas 3.0 (ICCV 2025), please check our GitHub: https://github.com/MrGiovanni/RadGPT.

This model was only trained with per-voxel segmentation masks. In the MICCAI 2025 paper "Learning Segmentation from Radiology Report", it served as the "segmentation" baseline.

Also, this is the model is a starting point for our R-Super: you can can fine-tune it with radiology reports, please see our Report Supervision (R-Super) GitHub.

Training and inference code: https://github.com/MrGiovanni/R-Super

Label order

['adrenal_gland_left',
 'adrenal_gland_right',
 'aorta',
 'bladder',
 'celiac_trunk',
 'colon',
 'common_bile_duct',
 'duodenum',
 'esophagus',
 'femur_left',
 'femur_right',
 'gall_bladder',
 'hepatic_vessel',
 'intestine',
 'kidney_left',
 'kidney_lesion',
 'kidney_right',
 'liver',
 'liver_lesion',
 'liver_segment_1',
 'liver_segment_2',
 'liver_segment_3',
 'liver_segment_4',
 'liver_segment_5',
 'liver_segment_6',
 'liver_segment_7',
 'liver_segment_8',
 'lung_left',
 'lung_right',
 'pancreas',
 'pancreas_body',
 'pancreas_head',
 'pancreas_tail',
 'pancreatic_lesion',
 'portal_vein_and_splenic_vein',
 'postcava',
 'prostate',
 'rectum',
 'spleen',
 'stomach',
 'superior_mesenteric_artery',
 'veins']

---

Papers

Scaling Artificial Intelligence for Multi-Tumor Early Detection with More Reports, Fewer Masks
Pedro R. A. S. Bassi, Xinze Zhou, Wenxuan Li, Szymon Płotka, Jieneng Chen, Qi Chen, Zheren Zhu, Jakub Prządo, Ibrahim E. Hamacı, Sezgin Er, Yuhan Wang, Ashwin Kumar, Bjoern Menze, Jarosław B. Ćwikła, Yuyin Zhou, Akshay S. Chaudhari, Curtis P. Langlotz, Sergio Decherchi, Andrea Cavalli, Kang Wang, Yang Yang, Alan Yuille, Zongwei Zhou*
Johns Hopkins University

Learning Segmentation from Radiology Reports
Pedro R. A. S. Bassi, Wenxuan Li, Jieneng Chen, Zheren Zhu, Tianyu Lin, Sergio Decherchi, Andrea Cavalli, Kang Wang, Yang Yang, Alan Yuille, Zongwei Zhou*
Johns Hopkins University
MICCAI 2025
Finalist, Best Paper and Young Scientist Awards

RadGPT: Constructing 3D Image-Text Tumor Datasets
Pedro R. A. S. Bassi, Mehmet Yavuz, Kang Wang, Sezgin Er, Ibrahim E. Hamamci, Wenxuan Li, Xiaoxi Chen, Sergio Decherchi, Andrea Cavalli, Yang Yang, Alan Yuille, Zongwei Zhou*
Johns Hopkins University
ICCV, 2025

Citations

If you use this data, please cite the 3 paper below:

@article{bassi2025learning,
  title={Learning Segmentation from Radiology Reports},
  author={Bassi, Pedro RAS and Li, Wenxuan and Chen, Jieneng and Zhu, Zheren and Lin, Tianyu and Decherchi, Sergio and Cavalli, Andrea and Wang, Kang and Yang, Yang and Yuille, Alan L and others},
  journal={arXiv preprint arXiv:2507.05582},
  year={2025}
}

@article{bassi2025radgpt,
  title={Radgpt: Constructing 3d image-text tumor datasets},
  author={Bassi, Pedro RAS and Yavuz, Mehmet Can and Wang, Kang and Chen, Xiaoxi and Li, Wenxuan and Decherchi, Sergio and Cavalli, Andrea and Yang, Yang and Yuille, Alan and Zhou, Zongwei},
  journal={arXiv preprint arXiv:2501.04678},
  year={2025}
}

@misc{bassi2025scaling,
      title={Scaling Artificial Intelligence for Multi-Tumor Early Detection with More Reports, Fewer Masks}, 
      author={Pedro R. A. S. Bassi and Xinze Zhou and Wenxuan Li and Szymon Płotka and Jieneng Chen and Qi Chen and Zheren Zhu and Jakub Prządo and Ibrahim E. Hamacı and Sezgin Er and Yuhan Wang and Ashwin Kumar and Bjoern Menze and Jarosław B. Ćwikła and Yuyin Zhou and Akshay S. Chaudhari and Curtis P. Langlotz and Sergio Decherchi and Andrea Cavalli and Kang Wang and Yang Yang and Alan L. Yuille and Zongwei Zhou},
      year={2025},
      eprint={2510.14803},
      archivePrefix={arXiv},
      primaryClass={cs.CV},
      url={https://arxiv.org/abs/2510.14803}, 
}

Acknowledgement

This work was supported by the Lustgarten Foundation for Pancreatic Cancer Research, the Patrick J. McGovern Foundation Award, and the National Institutes of Health (NIH) under Award Number R01EB037669. We would like to thank the Johns Hopkins Research IT team in IT@JH for their support and infrastructure resources where some of these analyses were conducted; especially DISCOVERY HPC. Paper content is covered by patents pending.