DropNaE: Alleviating irregularity for large-scale graph representation learning
Files
File version
Accepted Manuscript (AM)
Author(s)
Xiong, X
Yan, M
Xue, R
Pan, S
Pei, S
Deng, L
Ye, X
Fan, D
Griffith University Author(s)
Primary Supervisor
Other Supervisors
Editor(s)
Date
Size
File type(s)
Location
Abstract
Large-scale graphs are prevalent in various real-world scenarios and can be effectively processed using Graph Neural Networks (GNNs) on GPUs to derive meaningful representations. However, the inherent irregularity found in real-world graphs poses challenges for leveraging the single-instruction multiple-data execution mode of GPUs, leading to inefficiencies in GNN training. In this paper, we try to alleviate this irregularity at its origin—the irregular graph data itself. To this end, we propose DropNaE to alleviate the irregularity in large-scale graphs by conditionally dropping nodes and edges before GNN training. Specifically, we first present a metric to quantify the neighbor heterophily of all nodes in a graph. Then, we propose DropNaE containing two variants to transform the irregular degree distribution of the large-scale graph to a uniform one, based on the proposed metric. Experiments show that DropNaE is highly compatible and can be integrated into popular GNNs to promote both training efficiency and accuracy of used GNNs. DropNaE is offline performed and requires no online computing resources, benefiting the state-of-the-art GNNs in the present and future to a significant extent.
Journal Title
Neural Networks
Conference Title
Book Title
Edition
Volume
183
Issue
Thesis Type
Degree Program
School
Publisher link
Patent number
Funder(s)
Grant identifier(s)
Rights Statement
Rights Statement
This accepted manuscript is distributed under the Creative Commons Attribution-NonCommercial-NoDerivatives 4.0 International license (https://creativecommons.org/licenses/by-nc-nd/4.0/).
Item Access Status
Note
Access the data
Related item(s)
Subject
Artificial intelligence
Machine learning
Statistics
Persistent link to this record
Citation
Liu, X; Xiong, X; Yan, M; Xue, R; Pan, S; Pei, S; Deng, L; Ye, X; Fan, D, DropNaE: Alleviating irregularity for large-scale graph representation learning, Neural Networks, 2025, 183, pp. 106930