RadImageNet: An Open Radiologic Deep Learning Research Dataset for Effective Transfer Learning

被引:117
|
作者
Mei, Xueyan [1 ]
Liu, Zelong [1 ]
Robson, Philip M. [1 ,2 ]
Marinelli, Brett [2 ]
Huang, Mingqian [2 ]
Doshi, Amish [2 ]
Jacobi, Adam [2 ]
Cao, Chendi [1 ]
Link, Katherine E. [1 ]
Yang, Thomas [1 ]
Wang, Ying [3 ]
Greenspan, Hayit [1 ]
Deyer, Timothy [4 ,5 ]
Fayad, Zahi A. [1 ,2 ]
Yang, Yang [1 ,2 ]
机构
[1] Icahn Sch Med Mt Sinai, Leon & Norma Hess Ctr Sci & Med, Biomed Engn & Imaging Inst, 1470 Madison Ave, New York, NY 10029 USA
[2] Icahn Sch Med Mt Sinai, Leon & Norma Hess Ctr Sci & Med, Dept Diagnost Intervent & Mol Radiol, 1470 Madison Ave, New York, NY 10029 USA
[3] Univ Oklahoma, Dept Math, Okla, SK, Canada
[4] Cornell Med, Dept Radiol, New York, NY USA
[5] East River Med Imaging, Dept Radiol, New York, NY USA
基金
美国国家科学基金会;
关键词
CANCER;
D O I
10.1148/ryai.210315
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Purpose: To demonstrate the value of pretraining with millions of radiologic images compared with ImageNet photographic images on downstream medical applications when using transfer learning. Materials and Methods: This retrospective study included patients who underwent a radiologic study between 2005 and 2020 at an outpatient imaging facility. Key images and associated labels from the studies were retrospectively extracted from the original study interpretation. These images were used for RadImageNet model training with random weight initiation. The RadImageNet models were compared with ImageNet models using the area under the receiver operating characteristic curve (AUC) for eight classification tasks and using Dice scores for two segmentation problems. Results: The RadImageNet database consists of 1.35 million annotated medical images in 131 872 patients who underwent CT, MRI, and US for musculoskeletal, neurologic, oncologic, gastrointestinal, endocrine, abdominal, and pulmonary pathologic conditions. For transfer learning tasks on small datasets-thyroid nodules (US), breast masses (US), anterior cruciate ligament injuries (MRI), and meniscal tears (MRI)-the RadImageNet models demonstrated a significant advantage (P < .001) to ImageNet models (9.4%, 4.0%, 4.8%, and 4.5% AUC improvements, respectively). For larger datasets-pneumonia (chest radiography), COVID-19 (CT), SARS-CoV-2 (CT), and intracranial hemorrhage (CT)-the RadImageNet models also illustrated improved AUC (P < .001) by 1.9%, 6.1%, 1.7%, and 0.9%, respectively. Additionally, lesion localizations of the RadImageNet models were improved by 64.6% and 16.4% on thyroid and breast US datasets, respectively. Conclusion: RadImageNet pretrained models demonstrated better interpretability compared with ImageNet models, especially for smaller radiologic datasets.
引用
收藏
页数:9
相关论文
共 50 条
  • [1] Breast Mass Detection and Classification Using Transfer Learning on OPTIMAM Dataset Through RadImageNet Weights
    Kassahun, Ruth Kehali
    Molinara, Mario
    Bria, Alessandro
    Marrocco, Claudio
    Tortorella, Francesco
    IMAGE ANALYSIS AND PROCESSING - ICIAP 2023 WORKSHOPS, PT II, 2024, 14366 : 71 - 82
  • [2] Moving from ImageNet to RadImageNet for Improved Transfer Learning and Generalizability
    Cadrin-Chenevert, Alexandre
    RADIOLOGY-ARTIFICIAL INTELLIGENCE, 2022, 4 (05)
  • [3] A UAV Open Dataset of Rice Paddies for Deep Learning Practice
    Yang, Ming-Der
    Tseng, Hsin-Hung
    Hsu, Yu-Chun
    Yang, Chin-Ying
    Lai, Ming-Hsin
    Wu, Dong-Hong
    REMOTE SENSING, 2021, 13 (07)
  • [4] CrackVision: Effective Concrete Crack Detection With Deep Learning and Transfer Learning
    Alkannad, Abdulrahman A.
    Al Smadi, Ahmad
    Yang, Shuyuan
    Al-Smadi, Mutasem K.
    Al-Makhlafi, Moeen
    Feng, Zhixi
    Yin, Zhenlong
    IEEE ACCESS, 2025, 13 : 29554 - 29576
  • [5] Generation of microbial colonies dataset with deep learning style transfer
    Pawlowski, Jaroslaw
    Majchrowska, Sylwia
    Golan, Tomasz
    SCIENTIFIC REPORTS, 2022, 12 (01)
  • [6] On the Effective Transfer Learning Strategy for Medical Image Analysis in Deep Learning
    Wen, Yang
    Chen, Leiting
    Zhou, Chuan
    Deng, Yu
    Zeng, Huiru
    Xi, Shuo
    Guo, Rui
    2020 IEEE INTERNATIONAL CONFERENCE ON BIOINFORMATICS AND BIOMEDICINE, 2020, : 827 - 834
  • [7] Cross-dataset Deep Transfer Learning for Activity Recognition
    Gjoreski, Martin
    Kalabakov, Stefan
    Lustrek, Mitja
    Gams, Matjaz
    Gjoreski, Hristijan
    UBICOMP/ISWC'19 ADJUNCT: PROCEEDINGS OF THE 2019 ACM INTERNATIONAL JOINT CONFERENCE ON PERVASIVE AND UBIQUITOUS COMPUTING AND PROCEEDINGS OF THE 2019 ACM INTERNATIONAL SYMPOSIUM ON WEARABLE COMPUTERS, 2019, : 714 - 718
  • [8] Generation of microbial colonies dataset with deep learning style transfer
    Jarosław Pawłowski
    Sylwia Majchrowska
    Tomasz Golan
    Scientific Reports, 12
  • [9] Impact of dataset size and variety on the effectiveness of deep learning and transfer learning for plant disease classification
    Arnal Barbedo, Jayme Garcia
    COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2018, 153 : 46 - 53
  • [10] Research on the Deep Learning of the Small Sample Data based on Transfer Learning
    Zhao, Wei
    GREEN ENERGY AND SUSTAINABLE DEVELOPMENT I, 2017, 1864