A clinically applicable and generalizable deep learning model for anterior mediastinal tumors in CT images across multiple institutions.
Authors
Affiliations (9)
Affiliations (9)
- Department of Thoracic Surgery, National Cancer Center Hospital, 5-1-1 Tsukiji, Chuo-ku, 104-0045, Tokyo, Japan.
- Department of Diagnostic Radiology, National Cancer Center Hospital, 5-1-1 Tsukiji, Chuo-ku, 104-0045, Tokyo, Japan.
- Division of Medical AI Research and Development, National Cancer Center Research Institute, 5-1-1 Tsukiji, Chuo-ku, 104-0045, Tokyo, Japan.
- Digital Content and Media Sciences Research Division, National Institute of Informatics, 2-1-2 Hitotsubashi, Chiyoda-ku, 101-8430, Tokyo, Japan.
- AI Medical Engineering Team, RIKEN Center for Advanced Intelligence Project, 1-4-1 Nihonbashi, Chuo-ku, 103-0027, Tokyo, Japan.
- Division of Radiological Sciences, Graduate School of Health Sciences, Ibaraki Prefectural University of Health Sciences, 4669-2, Ami-machi, Inashiki-gun, 300-0394, Ibaraki, Japan.
- Department of Thoracic Oncology, National Cancer Center Hospital, 5-1-1 Tsukiji, Chuo-ku, 104-0045, Tokyo, Japan.
- Department of Diagnostic Pathology, National Cancer Center Hospital, 5-1-1 Tsukiji, Chuo-ku, 104-0045, Tokyo, Japan.
- Department of Diagnostic Radiology, National Cancer Center Hospital, 5-1-1 Tsukiji, Chuo-ku, 104-0045, Tokyo, Japan. [email protected].
Abstract
Rare diseases are often difficult to diagnose, and their scarcity also makes it challenging to develop deep learning models for them due to limited large-scale datasets. Anterior mediastinal tumors-including thymoma and thymic carcinoma-represent such rare entities. A few diagnostic support systems for these tumors have been proposed; however, no prior studies have tested them across multiple institutions, and clinically applicable and generalizable models remain lacking. A total of 711 computed tomography (CT) images were collected from 136 hospitals, each from a different patient with pathologically proven anterior mediastinal tumors (339 males, 372 females). Of these, 485 images were used for training, 62 for tuning, and 164 for external testing. The external testing dataset comprised CT images from 121 unique institutions not involved in the other datasets. A 3D U-Net-based model was trained on the training dataset, and the model with the best performance on the tuning dataset was selected. This model was then evaluated on the external testing dataset for its segmentation and detection performance across different institutions. Based on the reference standards provided by board-certified diagnostic radiologists, the trained model achieved average Dice scores of 0.82, Intersection over Union (IoU) of 0.72, Precision of 0.85, and Recall of 0.82 for tumor segmentation at the CT-image level. The free-response receiver operating characteristic curve-derived from lesion-wise IoU thresholds-demonstrated high sensitivity and a low false-positive rate for tumor detection. Even under a stricter IoU threshold of 0.50, the model maintained a sensitivity of 0.87 with only 0.61 false positives per scan. Our model achieved clinically applicable segmentation and detection performance for anterior mediastinal tumors, demonstrating broad generalizability across 121 institutions and overcoming the data-scarcity challenges inherent to such rare diseases.