Papers › Reducing Smoothness with Expressive Memory Enhanced Hierarchical Graph Neural Networks

Reducing Smoothness with Expressive Memory Enhanced Hierarchical Graph Neural Networks

1 Apr 2025arXiv:2504.00349archive 2025-07-28

Thomas Bailie, Yun Sing Koh, S. Karthik Mukkavilli, Varvara Vetrova

Graphical forecasting models learn the structure of time series data via projecting onto a graph, with recent techniques capturing spatial-temporal associations between variables via edge weights. Hierarchical variants offer a distinct advantage by analysing the time series across multiple resolutions, making them particularly effective in tasks like global weather forecasting, where low-resolution variable interactions are significant. A critical challenge in hierarchical models is information loss during forward or backward passes through the hierarchy. We propose the Hierarchical Graph Flow (HiGFlow) network, which introduces a memory buffer variable of dynamic size to store previously seen information across variable resolutions. We theoretically show two key results: HiGFlow reduces smoothness when mapping onto new feature spaces in the hierarchy and non-strictly enhances the utility of message-passing by improving Weisfeiler-Lehman (WL) expressivity. Empirical results demonstrate that HiGFlow outperforms state-of-the-art baselines, including transformer models, by at least an average of 6.1% in MAE and 6.2% in RMSE. Code is available at https://github.com/TB862/ HiGFlow.git.

PaperPDFCode

Code

tb862/higflow officialmentioned in paperpytorch report

Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

Not run by Syntology. Nothing on this page verifies that the listed code works.

Tasks

Time SeriesWeather Forecasting

Results from the paper archive 2025-07-28

No leaderboard rows for this paper in the archive.

Methods

MAE

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections