arXiv:2501.10153cs.LGcs.AI2025-01被引 1

用分区域堆叠集成提升脑龄预测准确率,兼顾隐私保护。

Region-wise stacking ensembles for estimating brain-age using MRI

  • 分区域建模后用二级模型融合预测,避免平均化信息损失。
  • 最佳结果MAE达4.75,优于基线均值的5.68,跨数据集训练效果更优。
  • 保留解剖特异性,增强生物可解释性,适合多中心研究应用。

利用结构磁共振成像(MRI)数据进行脑龄预测是研究脑老化的主流方法。机器学习与特征提取技术被广泛用于提升预测性能,并探索健康老化及加速老化(如神经退行性疾病和精神障碍)。高维MRI数据给构建泛化性强、可解释的模型以及保障数据隐私带来挑战。传统做法是对预定义区域内的体素进行重采样或平均,但会降低解剖特异性和生物学可解释性,因区域内体素对老化的关联可能不同。简单平均融合易导致信息丢失,降低准确性。本文提出一种概念新颖的两层堆叠集成(SE)方法:第一层为基于体素信息的区域模型,第二层模型融合各区域预测得到最终结果。在覆盖成年全生命周期的四个数据集上,以灰质体积(GMV)估计为输入,探索了八种数据融合方案。评估指标包括平均绝对误差(MAE)、R²、相关性和预测偏差,结果显示SE优于区域均值方法。最佳表现出现在第一层使用站点内留出样本预测、第二层模型基于独立且站点特异性数据训练时(MAE=4.75 vs 基线区域均值MAE=5.68)。随着训练数据集数量增加,性能持续提升。第一层预测展现出更优且更稳健的老化信号,提供新的生物学洞见并增强数据隐私。总体而言,该方法在提升准确性的同时,保持甚至增强了数据隐私。

原文摘要 · Abstract (English)

Predictive modeling using structural magnetic resonance imaging (MRI) data is a prominent approach to study brain-aging. Machine learning algorithms and feature extraction methods have been employed to improve predictions and explore healthy and accelerated aging e.g. neurodegenerative and psychiatric disorders. The high-dimensional MRI data pose challenges to building generalizable and interpretable models as well as for data privacy. Common practices are resampling or averaging voxels within predefined parcels, which reduces anatomical specificity and biological interpretability as voxels within a region may differently relate to aging. Effectively, naive fusion by averaging can result in information loss and reduced accuracy. We present a conceptually novel two-level stacking ensemble (SE) approach. The first level comprises regional models for predicting individuals' age based on voxel-wise information, fused by a second-level model yielding final predictions. Eight data fusion scenarios were explored using as input Gray matter volume (GMV) estimates from four datasets covering the adult lifespan. Performance, measured using mean absolute error (MAE), R2, correlation and prediction bias, showed that SE outperformed the region-wise averages. The best performance was obtained when first-level regional predictions were obtained as out-of-sample predictions on the application site with second-level models trained on independent and site-specific data (MAE=4.75 vs baseline regional mean GMV MAE=5.68). Performance improved as more datasets were used for training. First-level predictions showed improved and more robust aging signal providing new biological insights and enhanced data privacy. Overall, the SE improves accuracy compared to the baseline while preserving or enhancing data privacy.

脑龄预测堆叠集成MRI分析数据隐私

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。