DropNaE: Alleviating irregularity for large-scale graph representation learning

Loading...
Thumbnail Image
Files
Liu10338526.pdf
Embargoed until 2026-12-06
File version

Accepted Manuscript (AM)

Author(s)
Liu, X
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
2025
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

Collections