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 条
  • [21] A Review of Dataset Distillation for Deep Learning
    Thi-Thu-Huong Le
    Larasati, Harashta Tatimma
    Prihatno, Aji Teguh
    Kim, Howon
    2022 INTERNATIONAL CONFERENCE ON PLATFORM TECHNOLOGY AND SERVICE (PLATCON22), 2022, : 34 - 37
  • [22] Transfer Learning in Deep Reinforcement Learning
    Islam, Tariqul
    Abid, Dm. Mehedi Hasan
    Rahman, Tanvir
    Zaman, Zahura
    Mia, Kausar
    Hossain, Ramim
    PROCEEDINGS OF SEVENTH INTERNATIONAL CONGRESS ON INFORMATION AND COMMUNICATION TECHNOLOGY, ICICT 2022, VOL 1, 2023, 447 : 145 - 153
  • [23] DeepFire: A Novel Dataset and Deep Transfer Learning Benchmark for Forest Fire Detection
    Khan, Ali
    Hassan, Bilal
    Khan, Somaiya
    Ahmed, Ramsha
    Abuassba, Adnan
    MOBILE INFORMATION SYSTEMS, 2022, 2022
  • [24] A Novel Classification Model of Date Fruit Dataset Using Deep Transfer Learning
    Alsirhani, Amjad
    Siddiqi, Muhammad Hameed
    Mostafa, Ayman Mohamed
    Ezz, Mohamed
    Mahmoud, Alshimaa Abdelraof
    ELECTRONICS, 2023, 12 (03)
  • [25] Quantifying Uncertainty in Deep Learning of Radiologic Images
    Faghani, Shahriar
    Moassefi, Mana
    Rouzrokh, Pouria
    Khosravi, Bardia
    Baffour, Francis I.
    Ringler, Michael D.
    Erickson, Bradley J.
    RADIOLOGY, 2023, 308 (02)
  • [26] Open Research and Open Learning
    Williams, Roy Trevor
    Mackness, Jenny
    CAMPUS VIRTUALES, 2013, 2 (01): : 40 - 53
  • [27] Deep learning and transfer learning to understand emotions: a PoliEMO dataset and multi-label classification in Indian elections
    Surolia, Anuradha
    Mehta, Shikha
    Kumaraguru, Ponnurangam
    INTERNATIONAL JOURNAL OF DATA SCIENCE AND ANALYTICS, 2025,
  • [28] A Novel Approach to Classify Brain Tumor with an Effective Transfer Learning based Deep Learning Model
    Khushi, Hafiz Muhammad Tayyab
    Jaffar, Arfan
    Masood, Tehreem
    Akram, Sheeraz
    BRAZILIAN ARCHIVES OF BIOLOGY AND TECHNOLOGY, 2024, 67
  • [29] The Methodological Pitfall of Dataset-Driven Research on Deep Learning: An IoT Example
    Wang, Tianshi
    Kara, Denizhan
    Li, Jinyang
    Liu, Shengzhong
    Abdelzaher, Tarek
    Jalaian, Brian
    2022 IEEE MILITARY COMMUNICATIONS CONFERENCE (MILCOM), 2022,
  • [30] A comprehensive survey of deep learning research on medical image analysis with focus on transfer learning
    Atasever, Sema
    Azginoglu, Nuh
    Terzi, Duygu Sinanc
    Terzi, Ramazan
    CLINICAL IMAGING, 2023, 94 : 18 - 41