arXiv:2411.05825q-bio.NCcs.AI2024-11被引 8

提出可解释的脑表面图神经网络,精准预测新生儿脑龄并定位关键区域。

SurfGNN: A robust surface-based prediction model with interpretability for coactivation maps of spatial and cortical features

  • 将皮层表面建模为稀疏图,结合拓扑采样与区域特异性学习。
  • 在481名新生儿数据上实现0.827±0.056的平均绝对误差,提升超9.0%。
  • 生成特征级激活图,揭示不同形态学贡献的区域差异,适合临床研究者。

当前基于脑表面的预测模型常忽视皮层特征层级的区域属性变异性。尽管图神经网络(GNN)擅长捕捉区域差异,但在处理复杂高密度图结构时面临挑战。本文将皮层表面网格视为稀疏图,提出可解释的表面图神经网络(SurfGNN)。SurfGNN采用拓扑采样学习(TSL)和区域特异性学习(RSL)结构,在表面网格的低阶与高阶尺度上分别管理个体皮层特征,有效应对节点过多问题并缓解皮层区域异质性。在此基础上,引入新颖的评分加权融合(SWF)方法,整合各皮层特征的节点表示以进行预测。模型应用于481名受试者(共503次扫描)的标准化磁共振图像新生儿脑龄预测任务。SurfGNN优于所有现有先进方法,提升幅度至少9.0%,达到0.827±0.056的平均绝对误差(以胎龄周计)。此外,模型生成特征级激活图,展现其识别不同形态学贡献区域变异的能力。

原文摘要 · Abstract (English)

Current brain surface-based prediction models often overlook the variability of regional attributes at the cortical feature level. While graph neural networks (GNNs) excel at capturing regional differences, they encounter challenges when dealing with complex, high-density graph structures. In this work, we consider the cortical surface mesh as a sparse graph and propose an interpretable prediction model-Surface Graph Neural Network (SurfGNN). SurfGNN employs topology-sampling learning (TSL) and region-specific learning (RSL) structures to manage individual cortical features at both lower and higher scales of the surface mesh, effectively tackling the challenges posed by the overly abundant mesh nodes and addressing the issue of heterogeneity in cortical regions. Building on this, a novel score-weighted fusion (SWF) method is implemented to merge nodal representations associated with each cortical feature for prediction. We apply our model to a neonatal brain age prediction task using a dataset of harmonized MR images from 481 subjects (503 scans). SurfGNN outperforms all existing state-of-the-art methods, demonstrating an improvement of at least 9.0% and achieving a mean absolute error (MAE) of 0.827+0.056 in postmenstrual weeks. Furthermore, it generates feature-level activation maps, indicating its capability to identify robust regional variations in different morphometric contributions for prediction.

脑图谱图神经网络可解释性新生儿

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