Guardado en:
Detalles Bibliográficos
Autores principales: Zhu, Cheng, Zhu, Jiayi, Wu, Xi, Zhang, Lijuan, Yang, Shuqi, Liang, Ping, Chen, Honghan, Tan, Ying
Formato: Preprint
Publicado: 2023
Materias:
Acceso en línea:https://arxiv.org/abs/2308.11909
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913247321391104
author Zhu, Cheng
Zhu, Jiayi
Wu, Xi
Zhang, Lijuan
Yang, Shuqi
Liang, Ping
Chen, Honghan
Tan, Ying
author_facet Zhu, Cheng
Zhu, Jiayi
Wu, Xi
Zhang, Lijuan
Yang, Shuqi
Liang, Ping
Chen, Honghan
Tan, Ying
contents Graph Convolutional Networks (GCNs) can capture non-Euclidean spatial dependence between different brain regions. The graph pooling operator, a crucial element of GCNs, enhances the representation learning capability and facilitates the acquisition of abnormal brain maps. However, most existing research designs graph pooling operators solely from the perspective of nodes while disregarding the original edge features. This confines graph pooling application scenarios and diminishes its ability to capture critical substructures. In this paper, we propose a novel edge-aware hard clustering graph pool (EHCPool), which is tailored to dominant edge features and redefines the clustering process. EHCPool initially introduced the 'Edge-to-Node' score criterion which utilized edge information to evaluate the significance of nodes. An innovative Iteration n-top strategy was then developed, guided by edge scores, to adaptively learn sparse hard clustering assignments for graphs. Additionally, a N-E Aggregation strategy is designed to aggregate node and edge features in each independent subgraph. Extensive experiments on the multi-site public datasets demonstrate the superiority and robustness of the proposed model. More notably, EHCPool has the potential to probe different types of dysfunctional brain networks from a data-driven perspective. Method code: https://github.com/swfen/EHCPool
format Preprint
id arxiv_https___arxiv_org_abs_2308_11909
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Edge-aware Hard Clustering Graph Pooling for Brain Imaging
Zhu, Cheng
Zhu, Jiayi
Wu, Xi
Zhang, Lijuan
Yang, Shuqi
Liang, Ping
Chen, Honghan
Tan, Ying
Computer Vision and Pattern Recognition
Graphics
Graph Convolutional Networks (GCNs) can capture non-Euclidean spatial dependence between different brain regions. The graph pooling operator, a crucial element of GCNs, enhances the representation learning capability and facilitates the acquisition of abnormal brain maps. However, most existing research designs graph pooling operators solely from the perspective of nodes while disregarding the original edge features. This confines graph pooling application scenarios and diminishes its ability to capture critical substructures. In this paper, we propose a novel edge-aware hard clustering graph pool (EHCPool), which is tailored to dominant edge features and redefines the clustering process. EHCPool initially introduced the 'Edge-to-Node' score criterion which utilized edge information to evaluate the significance of nodes. An innovative Iteration n-top strategy was then developed, guided by edge scores, to adaptively learn sparse hard clustering assignments for graphs. Additionally, a N-E Aggregation strategy is designed to aggregate node and edge features in each independent subgraph. Extensive experiments on the multi-site public datasets demonstrate the superiority and robustness of the proposed model. More notably, EHCPool has the potential to probe different types of dysfunctional brain networks from a data-driven perspective. Method code: https://github.com/swfen/EHCPool
title Edge-aware Hard Clustering Graph Pooling for Brain Imaging
topic Computer Vision and Pattern Recognition
Graphics
url https://arxiv.org/abs/2308.11909