用健康婴儿脑龄模型预测缺氧缺血性脑病的发育结局
AGE2HIE: Transfer Learning from Brain Age to Predicting Neurocognitive Outcome for Infant Brain Injury
- 从健康脑影像学中学习脑龄,迁移用于早产儿病情预测
- 跨站点预测准确率提升5%,2岁神经发育结果预测更准
- 适合新生儿神经损伤早期评估与治疗决策支持
缺氧缺血性脑病(HIE)影响每1000名新生儿中的1至5名,其中30%至50%会出现不良神经认知后果。然而,这些后果直到2岁才能可靠评估。因此,利用深度学习模型早期准确预测HIE相关神经认知结局对改善临床决策、指导治疗和评估新疗法至关重要。但主要挑战在于缺乏大规模标注的HIE数据集。我们构建了首个公开的大型数据集,包含156例具有2岁神经认知结局标签的病例。相比之下,我们收集了8,859例正常婴儿及成人脑部磁共振成像(MRI),年龄范围为0至97岁,可用于深度学习模型进行脑龄估计。本文提出AGE2HIE,将深度学习模型从健康对照组脑部MRI中学习到的知识,迁移至患病群体,从结构到弥散MRI,从连续脑龄回归到二分类神经认知结局预测,从全生命周期年龄(0-97岁)到新生儿期(0-2周)。相比从零训练,基于脑龄估计的迁移学习不仅显著提升预测准确率(同站点或多站点分别提升3%或2%),还大幅增强模型在不同医院间的泛化能力(跨站点验证提升5%)。
原文摘要 · Abstract (English)
Hypoxic-Ischemic Encephalopathy (HIE) affects 1 to 5 out of every 1,000 newborns, with 30% to 50% of cases resulting in adverse neurocognitive outcomes. However, these outcomes can only be reliably assessed as early as age 2. Therefore, early and accurate prediction of HIE-related neurocognitive outcomes using deep learning models is critical for improving clinical decision-making, guiding treatment decisions and assessing novel therapies. However, a major challenge in developing deep learning models for this purpose is the scarcity of large, annotated HIE datasets. We have assembled the first and largest public dataset, however it contains only 156 cases with 2-year neurocognitive outcome labels. In contrast, we have collected 8,859 normal brain black Magnetic Resonance Imagings (MRIs) with 0-97 years of age that are available for brain age estimation using deep learning models. In this paper, we introduce AGE2HIE to transfer knowledge learned by deep learning models from healthy controls brain MRIs to a diseased cohort, from structural to diffusion MRIs, from regression of continuous age estimation to prediction of the binary neurocognitive outcomes, and from lifespan age (0-97 years) to infant (0-2 weeks). Compared to training from scratch, transfer learning from brain age estimation significantly improves not only the prediction accuracy (3% or 2% improvement in same or multi-site), but also the model generalization across different sites (5% improvement in cross-site validation).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。