Evolutionary Strategies AI Addresses Multiple Technical Challenges in Deep Learning Deployment: Proof-of-Principle Demonstration for Neuroblastoma Brain Metastasis Detection
被引:0
|
作者:
Purkayastha, Subhanik
论文数: 0引用数: 0
h-index: 0
机构:
Mem Sloan Kettering Canc Ctr, Dept Radiol, New York, NY 10065 USAMem Sloan Kettering Canc Ctr, Dept Radiol, New York, NY 10065 USA
Purkayastha, Subhanik
[1
]
Shalu, Hrithwik
论文数: 0引用数: 0
h-index: 0
机构:
Indian Inst Technol Madras, Dept Aerosp Engn, Chennai 600036, IndiaMem Sloan Kettering Canc Ctr, Dept Radiol, New York, NY 10065 USA
Shalu, Hrithwik
[2
]
Gutman, David
论文数: 0引用数: 0
h-index: 0
机构:
Mem Sloan Kettering Canc Ctr, Dept Radiol, New York, NY 10065 USAMem Sloan Kettering Canc Ctr, Dept Radiol, New York, NY 10065 USA
Gutman, David
[1
]
Holodny, Andrei
论文数: 0引用数: 0
h-index: 0
机构:
Mem Sloan Kettering Canc Ctr, Dept Radiol, New York, NY 10065 USAMem Sloan Kettering Canc Ctr, Dept Radiol, New York, NY 10065 USA
Holodny, Andrei
[1
]
Modak, Shakeel
论文数: 0引用数: 0
h-index: 0
机构:
Mem Sloan Kettering Canc Ctr, Dept Pediat, New York, NY 10065 USAMem Sloan Kettering Canc Ctr, Dept Radiol, New York, NY 10065 USA
Modak, Shakeel
[3
]
Basu, Ellen
论文数: 0引用数: 0
h-index: 0
机构:
Mem Sloan Kettering Canc Ctr, Dept Pediat, New York, NY 10065 USAMem Sloan Kettering Canc Ctr, Dept Radiol, New York, NY 10065 USA
Basu, Ellen
[3
]
Kushner, Brian
论文数: 0引用数: 0
h-index: 0
机构:
Mem Sloan Kettering Canc Ctr, Dept Pediat, New York, NY 10065 USAMem Sloan Kettering Canc Ctr, Dept Radiol, New York, NY 10065 USA
Kushner, Brian
[3
]
Kramer, Kim
论文数: 0引用数: 0
h-index: 0
机构:
Mem Sloan Kettering Canc Ctr, Dept Pediat, New York, NY 10065 USAMem Sloan Kettering Canc Ctr, Dept Radiol, New York, NY 10065 USA
Kramer, Kim
[3
]
Haque, Sofia
论文数: 0引用数: 0
h-index: 0
机构:
Mem Sloan Kettering Canc Ctr, Dept Radiol, New York, NY 10065 USAMem Sloan Kettering Canc Ctr, Dept Radiol, New York, NY 10065 USA
Haque, Sofia
[1
]
Stember, Joseph N.
论文数: 0引用数: 0
h-index: 0
机构:
Mem Sloan Kettering Canc Ctr, Dept Radiol, New York, NY 10065 USAMem Sloan Kettering Canc Ctr, Dept Radiol, New York, NY 10065 USA
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.