LLM2D
NeuroGraph:脑连接组学中图机器学习的基准测试
NeuroGraph: Benchmarks for Graph Machine Learning in Brain Connectomics
作者: Anwar Said, Roza G. Bayrak, Tyler Derr, Mudassir Shabbir, Daniel Moyer, Catie Chang, Xenofon Koutsoukos
发布日期: 11/25/2024
arXiv ID: oai:arXiv.org:2306.06202v4

摘要

机器学习为分析高维功能性神经影像数据提供了一种宝贵的工具,并被证明可有效预测各种神经系统疾病、精神障碍和认知模式。在功能性磁共振成像 (MRI) 研究中,大脑区域之间的相互作用通常使用基于图的表示进行建模。基于图的机器学习方法的效力已在无数领域得到证实,标志着数据解释和预测建模的变革性步骤。然而,尽管这些技术很有前景,但由于潜在的预处理流程数量庞大以及基于图的数据集构建的大型参数搜索空间,将其应用于神经影像领域一直面临挑战。在本文中,我们介绍了 NeuroGraph,这是一个基于图的神经影像数据集集合,并证明了其预测多种行为和认知特征的效用。我们通过创建包含静态和动态大脑连接的 35 个数据集,运行超过 15 种基线方法进行基准测试,深入研究了数据集生成搜索空间。此外,我们还提供了在静态图和动态图上学习的通用框架。我们广泛的实验得出了一些关键的观察结果。值得注意的是,使用相关向量作为节点特征、结合更多感兴趣区域以及使用更稀疏的图可以提高性能。为了促进基于图的数据驱动神经影像分析的进一步发展,我们提供了一个全面的开源 Python 包,其中包括基准数据集、基线实现、模型训练和标准评估。