Towards Efficient Training of Graph Neural Networks: A Multiscale Approach

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Gal, Eshed, Eliasof, Moshe, Schönlieb, Carola-Bibiane, Kyrchei, Ivan I., Haber, Eldad, Treister, Eran
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908677308416000
author Gal, Eshed
Eliasof, Moshe
Schönlieb, Carola-Bibiane
Kyrchei, Ivan I.
Haber, Eldad
Treister, Eran
author_facet Gal, Eshed
Eliasof, Moshe
Schönlieb, Carola-Bibiane
Kyrchei, Ivan I.
Haber, Eldad
Treister, Eran
contents Graph Neural Networks (GNNs) have become powerful tools for learning from graph-structured data, finding applications across diverse domains. However, as graph sizes and connectivity increase, standard GNN training methods face significant computational and memory challenges, limiting their scalability and efficiency. In this paper, we present a novel framework for efficient multiscale training of GNNs. Our approach leverages hierarchical graph representations and subgraphs, enabling the integration of information across multiple scales and resolutions. By utilizing coarser graph abstractions and subgraphs, each with fewer nodes and edges, we significantly reduce computational overhead during training. Building on this framework, we propose a suite of scalable training strategies, including coarse-to-fine learning, subgraph-to-full-graph transfer, and multiscale gradient computation. We also provide some theoretical analysis of our methods and demonstrate their effectiveness across various datasets and learning tasks. Our results show that multiscale training can substantially accelerate GNN training for large scale problems while maintaining, or even improving, predictive performance.
format Preprint
id arxiv_https___arxiv_org_abs_2503_19666
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Towards Efficient Training of Graph Neural Networks: A Multiscale Approach
Gal, Eshed
Eliasof, Moshe
Schönlieb, Carola-Bibiane
Kyrchei, Ivan I.
Haber, Eldad
Treister, Eran
Machine Learning
Graph Neural Networks (GNNs) have become powerful tools for learning from graph-structured data, finding applications across diverse domains. However, as graph sizes and connectivity increase, standard GNN training methods face significant computational and memory challenges, limiting their scalability and efficiency. In this paper, we present a novel framework for efficient multiscale training of GNNs. Our approach leverages hierarchical graph representations and subgraphs, enabling the integration of information across multiple scales and resolutions. By utilizing coarser graph abstractions and subgraphs, each with fewer nodes and edges, we significantly reduce computational overhead during training. Building on this framework, we propose a suite of scalable training strategies, including coarse-to-fine learning, subgraph-to-full-graph transfer, and multiscale gradient computation. We also provide some theoretical analysis of our methods and demonstrate their effectiveness across various datasets and learning tasks. Our results show that multiscale training can substantially accelerate GNN training for large scale problems while maintaining, or even improving, predictive performance.
title Towards Efficient Training of Graph Neural Networks: A Multiscale Approach
topic Machine Learning
url https://arxiv.org/abs/2503.19666