arrow
Return

Distribution Knowledge Embedding for Graph Pooling

delete2022-01-01
delete3
delete
OA
AI
K
Kaixuan Chen
宋
宋杰 (Jie Song)
S
Shunyu Liu
N
Na Yu
冯
冯尊磊 (Zunlei Feng)
宋
宋明黎 (Mingli Song) *
DOI:10.1109/TKDE.2022.3208063delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Graph-level representation learning is the pivotal step for downstream tasks that operate on the whole graph. The most common approach to this problem is graph pooling, where node features are typically averaged or summed to obtain the graph representations. However, pooling operations like averaging or summing inevitably cause severe information missing, which may severely downgrade the final performance. In this article, we argue what is crucial to graph-level downstream tasks includes not only the topological structure but also the distribution from which nodes are sampled. Therefore, powered by existing Graph Neural Networks (GNN), we propose a new plug-and-play pooling module, termed as Distribution Knowledge Embedding (DKEPool), where graphs are viewed as distributions on top of GNNs and the pooling goal is to summarize the entire distribution information instead of retaining a certain feature vector by simple predefined pooling operations. A DKEPool network de facto disassembles representation learning into two stages, structure learning and distribution learning. Structure learning follows a recursive neighborhood aggregation scheme to update node features where structure information is obtained. Distribution learning, on the other hand, omits node interconnections and focuses more on the distribution depicted by all the nodes. Extensive experiments on graph classification tasks demonstrate that the proposed DKEPool significantly and consistently outperforms the state-of-the-art methods. The code is avaliable at https://github.com/chenchkx/dkepool
Keywords:
Graph neural network
distribution knowledge embedding
graph pooling
graph classification

Journal

IEEE Transactions on Knowledge and Data Engineering cover
IEEE Transactions on Knowledge and Data Engineering
IF:
10.4
Papers:
6.9K
Citations:
3.2W

Organization

Z
zhejiang university
Scholars:
17.7W
Papers: 12.1W
Citations: 152
Cited Papers

Cited Papers

Nonsingular Terminal Sliding Mode Control Based on Adaptive Barrier Function for nth-Order Perturbed Nonlinear Systems
err2021-12-23
err0
errOAAI
errKhalid A. Alattas; Javad Mostafaee; Abdullah K. Alanazi; Saleh Mobayen; Mai The Vu; Anton Zhilenkov; Hala M. Abo-Dief
errShare
errSave
Hierarchical Adaptive Pooling by Capturing High-Order Dependency for Graph Representation Learning
err2023-04-01
err13
errOAAI
errLiu, Ning; Jian, Songlei; Li, Dongsheng; Zhang, Yiming; Lai, Zhiquan; Xu, Hongzuo
errShare
errSave
Somatostatin inhibition of adenylate cyclase activity in different brain areas
err1989-07-01
err0
PREAI
errGennaro Schettini; Tullio Florio; Olimpia Meucci; Elisa Landolfi; Maurizo Grimaldi; Carmelo Ventra; Adriano Marino
errShare
errSave
Factors Associated with Highway Mortality of White-Tailed Deer
err1974-10-01
err0
PREAI
errMichael J. Puglisi; James S. Lindzey; Edward D. Bellis
errShare
errSave
Covariance descriptors on a Gaussian manifold and their application to image set classification
err2020-11-01
err23
PREAI
errChen, Kai-Xuan; Ren, Jie-Yi; Wu, Xiao-Jun; Kittler, Josef
errShare
errSave
Isoflurane reduces the ischemia reperfusion injury surge: A longitudinal study with MRI
err2014-10-01
err0
PREAI
errSaeid Taheri; Anandakumar Shunmugavel; Danielle Clark; Honglian Shi
errShare
errSave
errShare
errSave
researcher View more