arrow
Return

How Useful Is Graph Pooling for Node-Level Tasks?

delete2026-01-01
delete0
PRE
AI
Y
Yijun Duan *
X
Xin Liu
S
Steven Lynden
A
Akiyoshi Matono
Q
Qiang Ma
DOI:10.1007/978-3-032-06066-2_6delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
As an essential component of graph neural networks, graph pooling is indispensable for graph-level tasks such as graph classification and generation. However, certain node-level tasks inherently require graph pooling, particularly multiple instance learning (MIL) on graphs, a weakly supervised learning paradigm where only set-level labels are available for training node-level predictors. Existing embedding-based pooling aggregates node embeddings to obtain a holistic graph-level representation, neglecting direct inference of node labels. To address this limitation, we propose instance-based pooling, which maps node embeddings to node probabilities before generating graph representations. We prove that embedding-based pooling methods can be seamlessly transformed into instance-based ones without losing permutation invariance or expressiveness, while the latter offers better interpretability. Extensive experiments on diverse benchmark datasets validate the effectiveness of our proposed method, providing key insights into the selection of pooling methods for different machine learning tasks on graphs.
Keywords:
Graph Pooling
Graph Multiple Instance Learning
Instance-level Learning
Weakly supervised learning

Journal

M
MACHINE LEARNING AND KNOWLEDGE DISCOVERY IN DATABASES. RESEARCH TRACK, ECML PKDD 2025, PT III
IF:
0
Papers:
30
Citations:
0

Organization

K
kyoto institute of technology
Scholars:
491
Papers: 217
Citations: 0
Cited Papers

Cited Papers

Understanding Pooling in Graph Neural Networks
err2024-02-01
err38
errOAAI
errGrattarola, Daniele; Zambon, Daniele; Bianchi, Filippo Maria; Alippi, Cesare
errShare
errSave
Multiple instance learning: A survey of problem characteristics and applications
err2018-05-01
err434
errOAAI
errCarbonneau, Marc-Andre; Cheplygina, Veronika; Granger, Eric; Gagnon, Ghyslain
errShare
errSave
Deep Learning on Graphs
err
IF0
err2021-09-02
err0
errOAAI
errYao Ma; Jiliang Tang
errShare
errSave
Locality Sensitive Deep Learning for Detection and Classification of Nuclei in Routine Colon Cancer Histology Images
err2016-05-01
err898
errOAAI
errSirinukunwattana, Korsuk; Raza, Shan E. Ahmed; Tsang, Yee-Wah; Snead, David R. J.; Cree, Ian A.; Rajpoot, Nasir M.
errShare
errSave
Dual Cost-sensitive Graph Convolutional Network
err2022-07-18
err0
PREAI
errDuan,Yijun; Liu,Xin; Jatowt,Adam; Yu,Hai-tao; Lynden,Steven; Kim,Kyoung-Sook; Matono,Akiyoshi
errShare
errSave
Inexact Graph Representation Learning
err2024-06-30
err0
PREAI
errDuan,Yijun; Liu,Xin; Jatowt,Adam; Yu,Hai-Tao; Lynden,Steven; Matono,Akiyoshi
errShare
errSave
Revisiting multiple instance neural networks
err2018-02-01
err325
errOAAI
errWang, Xinggang; Yan, Yongluan; Tang, Peng; Bai, Xiang; Liu, Wenyu
errShare
errSave
errShare
errSave
researcher View more