G-MIMO: Empowering GNNs with Diverse Sub-Networks for Graph Classification

Qipeng Zhu,Jie Chen,Junping Zhang,Jian Pu
DOI: https://doi.org/10.1109/icme57554.2024.10687616
2024-01-01
Abstract:Graph neural networks (GNNs) demonstrate impressive performance in graph classification, albeit exhibiting challenges in terms of generalizability and robustness. Despite its proven efficacy in enhancing the robustness and generalizability of GNNs, ensemble learning encounters practical limitations due to its extensive computational and memory requirements. This paper presents a novel low-cost ensemble learning method for graph classification, which utilizes a Graph Multi-Input and Multi-Output framework (G-MIMO) to allow multiple sub-networks within a single GNN simultaneously treated. Extensive experiments demonstrate that G-MIMO effectively enhances GNN performance across multiple graph classification tasks without introducing much computational overhead. G-MIMO outperforms existing graph augmentations and ensemble approaches, delivering a 3.33% and 4.25% increase in average accuracy over standard GCN and GIN models, respectively. The source code has been released at https://github.com/smurf-1119/GMIMO.
What problem does this paper attempt to address?