Directed evolution algorithm drives neural prediction
This work addresses domain shift and label scarcity in neural prediction for personalized medical AI, specifically for children with cochlear implants, but it is incremental as it builds on existing methods like reinforcement learning and replay buffers.
The authors tackled the problem of predicting post-operative language outcomes for children with cochlear implants using neural MRI data, which suffers from domain shift and label scarcity across datasets. Their directed evolution model (DEM) improved cross-domain prediction performance by enhancing generalization and balancing exploitation-exploration in continuous learning.
Neural prediction offers a promising approach to forecasting the individual variability of neurocognitive functions and disorders and providing prognostic indicators for personalized invention. However, it is challenging to translate neural predictive models into medical artificial intelligent applications due to the limitations of domain shift and label scarcity. Here, we propose the directed evolution model (DEM), a novel computational model that mimics the trial-and-error processes of biological directed evolution to approximate optimal solutions for predictive modeling tasks. We demonstrated that the directed evolution algorithm is an effective strategy for uncertainty exploration, enhancing generalization in reinforcement learning. Furthermore, by incorporating replay buffer and continual backpropagate methods into DEM, we provide evidence of achieving better trade-off between exploitation and exploration in continuous learning settings. We conducted experiments on four different datasets for children with cochlear implants whose spoken language developmental outcomes vary considerably on the individual-child level. Preoperative neural MRI data has shown to accurately predict the post-operative outcome of these children within but not across datasets. Our results show that DEM can efficiently improve the performance of cross-domain pre-implantation neural predictions while addressing the challenge of label scarcity in target domain.