论文标题
大脑图嵌入和分类的深度超图U-NET
Deep Hypergraph U-Net for Brain Graph Embedding and Classification
论文作者
论文摘要
-背景。网络神经科学研究大脑是由网络(或连接组)代表的复杂系统,为脑形态和功能提供了更深入的见解,从而可以鉴定出非典型的大脑连通性改变,可以用作神经系统疾病的诊断标记。 - 存在方法。将数据样本(例如,大脑网络)映射到低维空间中的图形嵌入方法已被广泛用于探索用于分类或预测任务的样本之间的关系。但是,这些作品中的大多数是基于对样本之间的成对关系建模,无法捕获其高阶关系。 - 新方法。在本文中,受到几何深度学习的新生场所的启发,我们提出了HyperGraph U-NET(Hunet),这是一个新型的数据嵌入框架,利用了HyperGraph结构来学习数据样本的低维嵌入,同时捕获其高阶关系。具体而言,我们通过改善局部特征聚合并保留数据中存在的高阶关系,将U-NET体系结构自然在图上进行概括为超图。 -结果。我们测试了分别针对自闭症患者和痴呆症患者的小尺度和大规模异质脑连接数据集的方法。 -结论。我们的饥饿表现优于最先进的几何图和超图数据嵌入技术,分类精度的增长率为4-14%,表明可伸缩性和概括性。 Hunet代码可在https://github.com/basiralab/hunet上找到。
-Background. Network neuroscience examines the brain as a complex system represented by a network (or connectome), providing deeper insights into the brain morphology and function, allowing the identification of atypical brain connectivity alterations, which can be used as diagnostic markers of neurological disorders. -Existing Methods. Graph embedding methods which map data samples (e.g., brain networks) into a low dimensional space have been widely used to explore the relationship between samples for classification or prediction tasks. However, the majority of these works are based on modeling the pair-wise relationships between samples, failing to capture their higher-order relationships. -New Method. In this paper, inspired by the nascent field of geometric deep learning, we propose Hypergraph U-Net (HUNet), a novel data embedding framework leveraging the hypergraph structure to learn low-dimensional embeddings of data samples while capturing their high-order relationships. Specifically, we generalize the U-Net architecture, naturally operating on graphs, to hypergraphs by improving local feature aggregation and preserving the high-order relationships present in the data. -Results. We tested our method on small-scale and large-scale heterogeneous brain connectomic datasets including morphological and functional brain networks of autistic and demented patients, respectively. -Conclusion. Our HUNet outperformed state-of-the-art geometric graph and hypergraph data embedding techniques with a gain of 4-14% in classification accuracy, demonstrating both scalability and generalizability. HUNet code is available at https://github.com/basiralab/HUNet.