arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.08883cs.CV

TractGraphFormer:用于弥散MRI纤维束成像中可解释性别和年龄预测的解剖学信息混合图CNN-Transformer网络

TractGraphFormer: Anatomically Informed Hybrid Graph CNN-Transformer Network for Interpretable Sex and Age Prediction from Diffusion MRI Tractography

Brigham and Women's Hospital, Harvard Medical School(哈佛医学院布莱根妇女医院);University of Electronic Science and Technology of China(电子科技大学);Beth Israel Deaconess Medical Center, Harvard Medical School(哈佛医学院贝斯以色列女执事医疗中心);The University of Sydney(悉尼大学);University of New South Wales(新南威尔士大学);Boston University School of Medicine(波士顿大学医学院);Massachusetts General Hospital, Harvard Medical School(哈佛医学院麻省总医院)
Yuqian Chen, Fan Zhang, Meng Wang, Leo R. Zekelman, Suheyla Cetin-Karayumak, Tengfei Xue, Chaoyi Zhang, Yang Song, Jarrett Rushmore, Nikos Makris, Yogesh Rathi,… 展开作者

Yuqian Chen, Fan Zhang, Meng Wang, Leo R. Zekelman, Suheyla Cetin-Karayumak, Tengfei Xue, Chaoyi Zhang, Yang Song, Jarrett Rushmore, Nikos Makris, Yogesh Rathi, Weidong Cai, Lauren J. O'Donnell

AI总结:

提出TractGraphFormer混合图CNN-Transformer网络,融合局部解剖特征与全局依赖,在儿童和成人弥散MRI数据上实现可解释的性别和年龄预测。

摘要:

脑连接与非影像表型之间的关系越来越多地通过深度神经网络进行研究。然而,在卷积网络设计中,脑白质网络的局部和全局属性常常被忽视。我们引入了TractGraphFormer,一种专为弥散MRI纤维束成像设计的混合图CNN-Transformer深度学习框架。该模型利用白质结构的局部解剖特征和全局特征依赖性。图CNN模块捕获白质几何形状和灰质连接性,以从解剖学相似的白质连接中聚合局部特征,而Transformer模块使用自注意力机制增强全局信息学习。此外,TractGraphFormer包含一个注意力模块,用于解释预测性白质连接。我们将TractGraphFormer应用于性别和年龄预测任务。TractGraphFormer在大型儿童数据集(n=9345)和年轻成人数据集(n=1065)中表现出强大的性能。总体而言,我们的方法表明,白质中广泛的连接可预测个体的性别和年龄。对于每个预测任务,在两个数据集中识别出一致的预测性解剖纤维束。所提出的方法突显了在弥散MRI纤维束成像的机器学习中,整合局部解剖信息和全局特征依赖性以提高预测性能的潜力。

英文摘要:

The relationship between brain connections and non-imaging phenotypes is increasingly studied using deep neural networks. However, the local and global properties of brain white matter networks are often overlooked in convolutional network design. We introduce TractGraphFormer, a hybrid Graph CNN-Transformer deep learning framework tailored for diffusion MRI tractography. This model leverages local anatomical characteristics and global feature dependencies of white matter structures. The Graph CNN module captures white matter geometry and grey matter connectivity to aggregate local features from anatomically similar white matter connections, while the Transformer module uses self-attention to enhance global information learning. Additionally, TractGraphFormer includes an attention module for interpreting predictive white matter connections. We apply TractGraphFormer to tasks of sex and age prediction. TractGraphFormer shows strong performance in large datasets of children (n=9345) and young adults (n=1065). Overall, our approach suggests that widespread connections in the WM are predictive of the sex and age of an individual. For each prediction task, consistent predictive anatomical tracts are identified across the two datasets. The proposed approach highlights the potential of integrating local anatomical information and global feature dependencies to improve prediction performance in machine learning with diffusion MRI tractography.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑