arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2210.16670cs.CVcs.LG
收藏

图神经网络在神经影像形状分类中的比较研究

A Comparative Study of Graph Neural Networks for Shape Classification in Neuroimaging

Imperial College London(帝国理工学院)

Nairouz Shehata, Wulfie Bain, Ben Glocker

摘要:

图神经网络已成为一种有前途的方法,用于分析非欧几里德数据,如网格。在医学成像中,网格状数据对于解剖结构建模起着重要作用,并且形状分类可以用于计算机辅助诊断和疾病检测。然而,由于有太多的选择,使用GNN进行医学形状分析的最佳架构选择仍然不清楚。我们进行了比较分析,为从业者提供了当前最先进的几何深度学习在神经成像中的形状分类的概述。使用生物性别分类作为概念验证任务,我们发现使用FPFH作为节点特征大大提高了GNN的性能和对分布外数据的推广;我们比较了三种替代卷积层的性能;我们加强了数据增强对基于图的学习的重要性。然后,我们确认这些结果适用于临床相关任务,使用阿尔茨海默病的分类。

英文摘要:

Graph neural networks have emerged as a promising approach for the analysis of non-Euclidean data such as meshes. In medical imaging, mesh-like data plays an important role for modelling anatomical structures, and shape classification can be used in computer aided diagnosis and disease detection. However, with a plethora of options, the best architectural choices for medical shape analysis using GNNs remain unclear. We conduct a comparative analysis to provide practitioners with an overview of the current state-of-the-art in geometric deep learning for shape classification in neuroimaging. Using biological sex classification as a proof-of-concept task, we find that using FPFH as node features substantially improves GNN performance and generalisation to out-of-distribution data; we compare the performance of three alternative convolutional layers; and we reinforce the importance of data augmentation for graph based learning. We then confirm these results hold for a clinically relevant task, using the classification of Alzheimer's disease.

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

↑