UM  > Faculty of Business Administration
Residential Collegefalse
Status已發表Published
Module-based graph pooling for graph classification
Deng, Sucheng1; Yang, Geping1; Yang, Yiyang3; Gong, Zhiguo1; Chen, Can2; Chen, Xiang4; Hao, Zhifeng3,5
2024-10-01
Source PublicationPattern Recognition
ISSN0031-3203
Volume154Pages:110606
Abstract

Graph Neural Network (GNN) models are recently proposed to process the graph-structured data for the learning tasks on graphs, e.g., node classification, link prediction, and so on. This work focuses on the graph classification task, aiming to obtain the graph representation and predict the class label for a graph. Existing works proposed applying graph pooling to obtain graph embedding but still suffer from several issues. First, node embeddings are generated according to the topological information of the whole graph, but ignoring the local isomorphic substructures commonly seen in bioinformatics and chemistry. Another limitation arises when aggregating node embeddings. The hard assignment obtained through clustering algorithms, which rely on preset and fixed parameters instead of considering the graph's properties adaptively, restricts the flexibility in handling graphs of varying scales. To address the above problems, a module-based graph pooling framework (MGPool) is proposed in this work. Inspired by the rules of bioinformatics, MGPool assumes that a graph consists of multiple modules (also known as sub structures), which are identified based on the natural organization of the graph rather than the hard allocation of nodes. Benefiting from the hypothesis, MGPool generates node embeddings from graph-view and module-view, which is capable to capture global graph information and local isomorphic information respectively. Then module-level pooling is used to capture the intra-module information, while the inter-module information in terms of the correlation between modules is obtained through graph-level pooling. Finally, an entropy-based weighting mechanism is proposed to adjust the modules’ weights for the graph aggregation. Experiments conducted on bioinformatics benchmark datasets demonstrate the effectiveness of MGPool by outperforming other state-of-the-art graph pooling methods. For social network datasets, MGPool also provides competitive performance. Moreover, the visualization of module entropy weights is given to reveal the interpretability of the model.

KeywordGraph Classification Graph Neural Network Graph Pooling
DOI10.1016/j.patcog.2024.110606
URLView the original
Indexed BySCIE
Language英語English
WOS Research AreaComputer Science ; Engineering
WOS SubjectComputer Science, Artificial Intelligence ; Engineering, Electrical & Electronic
WOS IDWOS:001246321900001
PublisherELSEVIER SCI LTD, 125 London Wall, London EC2Y 5AS, ENGLAND
Scopus ID2-s2.0-85194174715
Fulltext Access
Citation statistics
Document TypeJournal article
CollectionFaculty of Business Administration
DEPARTMENT OF ACCOUNTING AND INFORMATION MANAGEMENT
DEPARTMENT OF COMPUTER AND INFORMATION SCIENCE
Corresponding AuthorGong, Zhiguo
Affiliation1.State Key Laboratory of Internet of Things for Smart City and Department of Computer and Information Science, University of Macau, Macao, Macao
2.University of Macau, Department of Accounting and Information Management, Macao
3.Faculty of Computer, Guangdong University of Technology, Guangdong, China
4.Sun Yat-Sen University, School of Electronics and Information Technology, Guangdong, China
5.Department of Mathematics, College of Science, Shantou University, Guangdong, China
First Author AffilicationUniversity of Macau
Corresponding Author AffilicationUniversity of Macau
Recommended Citation
GB/T 7714
Deng, Sucheng,Yang, Geping,Yang, Yiyang,et al. Module-based graph pooling for graph classification[J]. Pattern Recognition, 2024, 154, 110606.
APA Deng, Sucheng., Yang, Geping., Yang, Yiyang., Gong, Zhiguo., Chen, Can., Chen, Xiang., & Hao, Zhifeng (2024). Module-based graph pooling for graph classification. Pattern Recognition, 154, 110606.
MLA Deng, Sucheng,et al."Module-based graph pooling for graph classification".Pattern Recognition 154(2024):110606.
Files in This Item:
There are no files associated with this item.
Related Services
Recommend this item
Bookmark
Usage statistics
Export to Endnote
Google Scholar
Similar articles in Google Scholar
[Deng, Sucheng]'s Articles
[Yang, Geping]'s Articles
[Yang, Yiyang]'s Articles
Baidu academic
Similar articles in Baidu academic
[Deng, Sucheng]'s Articles
[Yang, Geping]'s Articles
[Yang, Yiyang]'s Articles
Bing Scholar
Similar articles in Bing Scholar
[Deng, Sucheng]'s Articles
[Yang, Geping]'s Articles
[Yang, Yiyang]'s Articles
Terms of Use
No data!
Social Bookmark/Share
All comments (0)
No comment.
 

Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.