Evolutionary Strategies AI Addresses Multiple Technical Challenges in Deep Learning Deployment: Proof-of-Principle Demonstration for Neuroblastoma Brain Metastasis Detection

被引:0
|
作者
Purkayastha, Subhanik [1 ]
Shalu, Hrithwik [2 ]
Gutman, David [1 ]
Holodny, Andrei [1 ]
Modak, Shakeel [3 ]
Basu, Ellen [3 ]
Kushner, Brian [3 ]
Kramer, Kim [3 ]
Haque, Sofia [1 ]
Stember, Joseph N. [1 ]
机构
[1] Mem Sloan Kettering Canc Ctr, Dept Radiol, New York, NY 10065 USA
[2] Indian Inst Technol Madras, Dept Aerosp Engn, Chennai 600036, India
[3] Mem Sloan Kettering Canc Ctr, Dept Pediat, New York, NY 10065 USA
关键词
Artificial intelligence; Convolutional neural network; Deep neuroevolution; Evolutionary strategies; Genetic algorithms supervised deep learning; Neuroblastoma; Magnetic resonance imaging; CENTRAL-NERVOUS-SYSTEM; FEATURES;
D O I
10.1007/s10278-024-01165-z
中图分类号
R8 [特种医学]; R445 [影像诊断学];
学科分类号
1002 ; 100207 ; 1009 ;
摘要
Two significant obstacles hinder the advancement of Radiology AI. The first is the challenge of overfitting, where small training data sets can result in unreliable outcomes. The second challenge is the need for more generalizability, the lack of which creates difficulties in implementing the technology across various institutions and practices. A recent innovation, deep neuroevolution (DNE), has been introduced to tackle the overfitting issue by training on small data sets and producing accurate predictions. However, the generalizability of DNE has yet to be proven. This paper strives to overcome this barrier by demonstrating that DNE can achieve satisfactory results in diverse external validation sets. The main innovation of the work is thus showing that DNE can generalize to varied outside data. Our example use case is predicting brain metastasis from neuroblastoma, emphasizing the importance of AI with limited data sets. Despite image collection and labeling advancements, rare diseases will always constrain data availability. We optimized a convolutional neural network (CNN) with DNE to demonstrate generalizability. We trained the CNN with 60 MRI images and tested it on a separate diverse collection of images from over 50 institutions. For comparison, we also trained with the more traditional stochastic gradient descent (SGD) method, with the two variants of (1) training from scratch and (2) transfer learning. Our results show that DNE demonstrates excellent generalizability with 97% accuracy on the heterogeneous testing set, while neither form of SGD could reach 60% accuracy. DNE's ability to generalize from small training sets to external and diverse testing sets suggests that it or similar approaches may play an integral role in improving the clinical performance of AI.
引用
收藏
页数:11
相关论文
empty
未找到相关数据