---
title: 'MillGNN: Multi-Scale Lead-Lag Forecasting'
url: https://www.emergentmind.com/topics/millgnn
type: topic
---

# MillGNN: Multi-Scale Lead-Lag Forecasting

Searching arXiv for MillGNN and closely related forecasting papers.
MillGNN is a graph neural network-based method for multivariate time series forecasting that learns multiple grouping scale lead-lag dependencies. It is designed for settings in which hierarchical propagation produces delayed effects across variables and groups of variables, and it combines statistically grounded lead-lag detection with structured multi-scale message passing. Formally, given historical observations $\boldsymbol{X}\in\mathbb{R}^{N\times L}$ for $N$ variables over input length $L$, MillGNN learns a forecasting function $\hat{\boldsymbol{X}}=f(\boldsymbol{X};\boldsymbol{\Theta})$ with $\hat{\boldsymbol{X}}\in\mathbb{R}^{N\times H}$ for prediction horizon $H$ [2509.03852].

## 1. Problem setting and conceptual basis

MillGNN addresses multivariate time series forecasting under explicit lead-lag structure. In this setting, a directed dependency means that segments of a leading variate affect a lagging variate after a time delay $\tau$; for a pair $(i,j)$, the effect from $x_i(t)$ to $x_j(t+\tau)$ means that $i$ leads $j$ at lag $\tau$. The method is motivated by the observation that existing methods often overlook lead-lag dependencies at multiple grouping scales, and therefore fail to capture hierarchical lead-lag effects in complex systems [2509.03852].

The central modeling assumption is that complex systems can exhibit both fine-scale and coarse-scale propagation. At fine scales, individual variate interactions are sensitive and dynamic; at coarse scales, fluctuations are smoothed and lead-lag relations become more stable. MillGNN therefore models both variate-wise and group-wise lead-lag dependencies. This suggests a division of labor across scales: fine scales capture localized and rapidly changing dependencies, whereas coarser scales summarize longer-range and more stable propagation patterns.

The underlying graph view is time-resolved and causal. Nodes are defined as $V=\{v_{i,n}\mid i\in[N], n\in[L]\}$, where time steps are nodes, and a directed edge $\epsilon_{ij,nm}$ encodes influence from $v_{i,n}$ to $v_{j,m}$ with lag $\tau=m-n$, subject to causality. The weighted adjacency $\boldsymbol{A}\in\mathbb{R}^{N\times N\times L\times L}$ is upper triangular per pair $(i,j)$ because
$$
\boldsymbol{A}_{ij,nm}=0 \quad \text{for } m<n.
$$
This causal structure is retained after scale-specific aggregation and patchification [2509.03852].

## 2. Multi-scale grouping and scale-specific lead-lag graph learning

A defining feature of MillGNN is its multi-scale grouping mechanism. Variates are grouped into $S$ scales based on a binarized Dynamic Time Warping similarity graph. If $\boldsymbol{D}^0\in\{0,1\}^{N\times N}$ denotes the initial binary similarity graph from the training set, then the assignment matrix at scale $s$ is
$$
\boldsymbol{S}^s=\operatorname{Clustering}(N^s,\boldsymbol{D}^{s-1}), \quad \boldsymbol{S}^s\in\{0,1\}^{N^{s-1}\times N^s}.
$$
The similarity structure and signals at scale $s$ are then
$$
\boldsymbol{D}^s=(\boldsymbol{S}^s)^\top\boldsymbol{D}^{s-1}\boldsymbol{S}^s,\qquad
\boldsymbol{X}^s=(\boldsymbol{S}^s)^\top\boldsymbol{X}^{s-1}.
$$
Patchification is scale-dependent:
$$
\bar{\boldsymbol{X}}^s\in\mathbb{R}^{N^s\times P^s\times p^s}, \quad P^s=L/p^s.
$$
Larger patch lengths $p^s$ are used at coarser scales to capture longer delays [2509.03852].

The scale-specific lead-lag graph learning module, SiLL-GL, begins with FFT-based cross-correlation. Using Wiener–Khinchin, MillGNN computes
$$
\{\boldsymbol{R}_{ij}(\tau)\}_{\tau=0}^{L-1}
=
\frac{1}{L}\,\mathcal{F}^{-1}\big(\mathcal{F}(\boldsymbol{X}_j)\odot\overline{\mathcal{F}(\boldsymbol{X}_i)}\big),
$$
where $\mathcal{F}$ is FFT, $\mathcal{F}^{-1}$ is inverse FFT, $\odot$ is element-wise product, and the overline denotes complex conjugation. In practice, cross-correlations are computed on patch-averaged signals per scale, and the Top-$K$ lag values are selected for each candidate pair [2509.03852].

Candidate pairs at scale $s$ are restricted to groups that share the same parent at scale $s+1$. For each such pair $(i,j)$, the lag set is
$$
\boldsymbol{\tau}_{ij}=\operatorname{TopK}\{R_{ij}(\tau)\}_{\tau=0}^{P^s-1}, \quad |\boldsymbol{\tau}_{ij}|=K.
$$
The initial binary adjacency $\boldsymbol{C}^s$ over patches $n,m\in[P^s]$ is defined by
$$
\boldsymbol{C}_{ij,nm}^s=
\begin{cases}
1, & m-n\in\boldsymbol{\tau}_{ij}\ \text{and}\ n<P^s-\max(\boldsymbol{\tau}_{ij}),\\
0, & \text{otherwise}.
\end{cases}
$$
Because of causality, $\boldsymbol{C}^s$ is upper triangular [2509.03852].

MillGNN then augments this statistically initialized graph with a decay-aware attention mechanism. Group embeddings for excitatory and inhibitory characteristics are introduced as $\boldsymbol{E}_{\mathrm{Ni}}^s,\boldsymbol{E}_{\mathrm{Ne}}^s\in\mathbb{R}^{N^s\times d_e}$, together with time embeddings $\boldsymbol{E}_{\mathrm{P}}^s\in\mathbb{R}^{P^s\times d_e}$. Broadcasting addition yields
$$
\boldsymbol{E}_{\mathrm{i}}^s=\boldsymbol{E}_{\mathrm{Ni}}^s\oplus\boldsymbol{E}_{\mathrm{P}}^s,\quad
\boldsymbol{E}_{\mathrm{e}}^s=\boldsymbol{E}_{\mathrm{Ne}}^s\oplus\boldsymbol{E}_{\mathrm{P}}^s,
$$
with $\boldsymbol{E}_{\mathrm{i}}^s,\boldsymbol{E}_{\mathrm{e}}^s\in\mathbb{R}^{N^s\times P^s\times d_e}$. Rate matrices are computed through SoftMax over patch interactions:
$$
\boldsymbol{\beta}_{\mathrm{e}}^s=\operatorname{SoftMax}\big(\boldsymbol{E}_{\mathrm{e}}^s(\boldsymbol{E}_{\mathrm{e}}^s)^\top\big),\quad
\boldsymbol{\beta}_{\mathrm{i}}^s=\operatorname{SoftMax}\big(\boldsymbol{E}_{\mathrm{i}}^s(\boldsymbol{E}_{\mathrm{i}}^s)^\top\big).
$$

Real-time input attention is derived from patchwise features:
$$
\boldsymbol{Q}^s=\operatorname{Linear}(\bar{\boldsymbol{X}}^s),\quad
\boldsymbol{K}^s=\operatorname{Linear}(\bar{\boldsymbol{X}}^s),\quad
\boldsymbol{\alpha}^s=\operatorname{SoftMax}\Big(\frac{\boldsymbol{Q}^s(\boldsymbol{K}^s)^\top}{\sqrt{d_a}}\Big).
$$
A lag tensor $\boldsymbol{\Delta}^s$ stores absolute patch differences. The dynamic decaying weight tensor is then
$$
\boldsymbol{\Lambda}^s=
\operatorname{ReLU}\Big(
e^{-\boldsymbol{\beta}_{\mathrm{e}}^s\odot\boldsymbol{\Delta}^s}
-
(1-\boldsymbol{\alpha}^s)\odot e^{-\boldsymbol{\beta}_{\mathrm{i}}^s\odot\boldsymbol{\Delta}^s}
\Big),
$$
and the scale-specific lead-lag adjacency becomes
$$
\boldsymbol{A}^s=\boldsymbol{C}^s\odot\boldsymbol{\Lambda}^s.
$$
The intended interpretation is explicit: larger $\boldsymbol{\beta}_{\mathrm{e}}^s$ strengthen decay, while larger $\boldsymbol{\alpha}^s$ and $\boldsymbol{\beta}_{\mathrm{i}}^s$ counteract decay, modeling periodic revival [2509.03852].

## 3. Hierarchical lead-lag message passing

MillGNN’s second core module is hierarchical lead-lag message passing, HiLL-MP. At each scale, patchified signals are embedded as
$$
\boldsymbol{H}^s=\operatorname{MLP}(\bar{\boldsymbol{X}}^s)\in\mathbb{R}^{N^s\times P^s\times d}.
$$
Hierarchical aggregation is then performed back to the variate level. The aggregate message is
$$
\boldsymbol{M}
=
\underbrace{\sigma\big(\boldsymbol{A}^0\boldsymbol{H}^0\boldsymbol{\theta}^0\big)}_{\text{Intra-scale}}
+
\underbrace{\sum_{s=1}^{S-1}\sigma\Big(\frac{1}{|v_s|}\,\big(\prod_{l=1}^{s}\boldsymbol{S}^l\big)\,\bigotimes\big(\boldsymbol{A}^s\boldsymbol{H}^s\boldsymbol{\theta}^s\big)\Big)}_{\text{Inter-scale}}.
$$
Here, $\sigma$ is an activation function; $\theta^s$ are linear maps; $\prod_{l=1}^{s}\boldsymbol{S}^l\in\{0,1\}^{N\times N^s}$ maps scale-$s$ groups to variates; $|v_s|$ is the number of variates under each scale-$s$ group; and $\otimes$ is a duplication operator that efficiently broadcasts group messages to assigned variates [2509.03852].

This structure separates two propagation modes. Intra-scale propagation acts among similar variates or groups within the same scale. Inter-scale propagation moves information progressively from coarser groups down to finer ones through duplication along the learned hierarchy. The paper characterizes this design as balancing comprehensiveness and efficiency, since it transmits lead-lag effects across multiple grouping scales without exhaustive pairwise multiplication [2509.03852].

After aggregation, variate-level and group-level features are updated. For the variate level,
$$
\boldsymbol{H}'=\sigma\big((\boldsymbol{H}\,||\,\boldsymbol{M})\cdot\boldsymbol{\theta}_{\mathrm{update}}\big).
$$
For groups,
$$
\boldsymbol{H}^{s\prime}
=
\sigma\Big(
\big(\boldsymbol{H}^s\,||\,\operatorname{Pool}\big(\boldsymbol{H}'\cdot(\prod_{l=1}^{s}\boldsymbol{S}^l)\big)\big)\cdot\boldsymbol{\theta}_{\mathrm{update}}^s
\Big).
$$
Multiple HiLL-MP layers are stacked with skip connections, producing the final variate-level output $\tilde{\boldsymbol{H}}\in\mathbb{R}^{N\times P^0\times d}$ [2509.03852].

The forecasting head follows PatchTST. MillGNN flattens patches and hidden channels and linearly projects to the forecast horizon:
$$
\hat{\boldsymbol{X}}=\operatorname{Linear}\big(\operatorname{Flatten}(\tilde{\boldsymbol{H}})\big).
$$
Training uses Mean Squared Error. Evaluation uses MSE and MAE for long-term tasks and MAE/MAPE/RMSE for short-term traffic and air-quality tasks [2509.03852].

## 4. Computational properties and algorithmic pipeline

MillGNN is designed to preserve tractability despite explicit lag modeling. FFT-based cross-correlation reduces per-pair complexity at scale $s$ from $O((P^s)^2)$ to $O(P^s\log P^s)$ after patch pooling, and cross-correlation is computed only for candidate pairs. The total message passing complexity per forward pass is stated as $O(SNL)$, based on the facts that total nodes satisfy $V\le S\times N\times L$, edges per scale are linear in $N\times L$ given bounded lag window and group size $k$, graph message passing scales as $O(V+E)=O(SNL)$, and inter-scale duplication is $O(1)$ per group-to-variates broadcast. Adjacency tensors per scale require $O(N^s\times N^s\times (P^s)^2)$ space; practical memory is controlled by candidate-pair filtering, patch sizes $p^s$, and small $S$, typically $1$–$3$ in practice [2509.03852].

The training and inference workflow has four main stages. First, data preprocessing normalizes series, computes DTW similarities on the training set, binarizes them to form $\boldsymbol{D}^0$, clusters to build the assignment matrices $\{\boldsymbol{S}^s\}_{s=1}^{S-1}$, aggregates signals per scale, and patchifies with scale-specific patch length $p^s$. Second, SiLL-GL identifies candidate group pairs, computes FFT-based cross-correlation, selects Top-$K$ lags, builds $\boldsymbol{C}^s$, computes the excitatory and inhibitory embeddings and real-time attention, forms the lag tensor, and constructs $\boldsymbol{A}^s=\boldsymbol{C}^s\odot\boldsymbol{\Lambda}^s$. Third, HiLL-MP computes $\boldsymbol{H}^s=\operatorname{MLP}(\bar{\boldsymbol{X}}^s)$, aggregates intra-scale and inter-scale messages, updates variate-level and group-level features, stacks layers with skip connections, and retains $\tilde{\boldsymbol{H}}$. Fourth, the forecast head computes $\hat{\boldsymbol{X}}=\operatorname{Linear}(\operatorname{Flatten}(\tilde{\boldsymbol{H}}))$ and optimizes MSE with Adam, using early stopping via validation; inference repeats the same stages and outputs $\hat{\boldsymbol{X}}$ [2509.03852].

Implementation details reported for the experiments include PyTorch, a single NVIDIA A100 80GB GPU, Adam optimization, initial learning rates in $\{10^{-2},10^{-3},5\times10^{-4},10^{-4}\}$, batch size $32$, $10$ epochs, and early stopping tolerance $3$. Hyperparameters are searched with NNI using a Bayesian tuner with $20$ trials and a curve fitting assessor for early termination. Search spaces are dataset dependent: for small-$N$ datasets, scales in $\{1,2\}$ and groups at scale $1$ in $\{1,2,3\}$; for large-$N$ datasets, scales in $\{1,2,3\}$, groups at scale $1$ in $\{5,10,15,20,25\}$, and groups at scale $2$ in $\{1,2,3\}$. Patch length satisfies $p^s\in\{2,4,8,12,24\}$ and Top-$K$ lags satisfy $K\in\{4,6,8,10,15,20\}$ [2509.03852].

## 5. Empirical performance and ablation results

MillGNN is evaluated on 11 datasets. The long-term forecasting benchmarks are ETTh1, ETTh2, ETTm1, ETTm2, Weather, Electricity, Exchange, and Traffic, with input length $L=96$ unless noted and horizons $H\in\{96,192,336,720\}$. The short-term benchmarks are PEMS04, PEMS08, and China-AQI, with $H=12$ for PEMS and $H=24$ for China-AQI. The baselines comprise 16 state-of-the-art methods: Autoformer, FEDformer, PatchTST, Stationary, DLinear, SCINet, TimesNet, TimeMixer, MTGNN, Crossformer, CrossGNN, MSGNet, iTransformer, FilterTS, LightMTS/LIFT, and VCformer [2509.03852].

Across the long-term benchmarks, MillGNN reports the following average results: ETTm1, MSE $0.373$ and MAE $0.390$; ETTm2, MSE $0.272$ and MAE $0.319$; ETTh1, MSE $0.431$ and MAE $0.428$; ETTh2, MSE $0.374$, described as close to best while TimeMixer attains $0.364$; Electricity, MSE $0.170$ and MAE $0.262$; Exchange, MSE $0.344$ and MAE $0.393$; Traffic, MSE $0.423$ and MAE $0.276$; and Weather, MSE $0.240$ with MAE approximately $0.275$ near best. On ETTm1 at horizon $H=96$, MillGNN attains MSE $0.315$ and MAE $0.352$; at $H=720$, MSE $0.442$ and MAE $0.429$. The reported “Count” line assigns MillGNN $26$ best scores for both MSE and MAE [2509.03852].

On short-term forecasting, MillGNN reports the best performance across all three metrics on all three datasets: PEMS04 with MAE $18.93$, MAPE $12.24$, RMSE $30.72$; PEMS08 with MAE $14.86$, MAPE $9.54$, RMSE $23.46$; and China-AQI with MAE $19.05$, MAPE $31.51$, RMSE $31.55$ [2509.03852].

The ablation results isolate the contribution of the major components. Removing multi-scale modeling yields ETTm1 MSE $0.379$ instead of $0.373$, Electricity MSE $0.176$ instead of $0.170$, and China-AQI MAE $19.61$ instead of $19.05$. Removing the initial graphs based on cross-correlation yields ETTm1 $0.388$, Electricity $0.182$, and China-AQI $19.88$. Removing decay-aware weights yields ETTm1 $0.394$, Electricity $0.183$, and China-AQI $20.02$. Replacing HiLL-MP with a vanilla GNN yields ETTm1 $0.385$, Electricity $0.179$, and China-AQI $19.53$. The full design consistently yields the lowest errors, and the decay-aware weights are reported as particularly impactful on China-AQI [2509.03852].

Robustness analyses further qualify the performance claims. On ETTh1 and ETTh2, MillGNN excels when train-test lag distributions are stable, as on ETTh1, but performance degrades when lag distributions drift, as observed on ETTh2. Hyperparameter analysis on China-AQI identifies an optimal number of scales around $S\approx 2$, Top-$K$ lags around $K\approx 5$, and decay embedding dimension around $d_e\approx 7$; too small $S$ or $K$ underfit, while too large $S$ or $K$ risk overfitting. Among inter-scale strategies, duplication is reported as both accurate and efficient relative to learnable assignment or bi-directional passing, the latter being described as overmixing and slowing training [2509.03852].

## 6. Interpretability, limitations, and relation to adjacent literature

MillGNN is presented as combining statistical interpretability with learned dynamic weighting. Interpretability begins with the initial edges, which arise from FFT-based cross-correlation, a well-understood statistic for time-lagged co-movements. It continues through the decay-aware weights, whose explicit lag-dependent form
$$
\boldsymbol{\Lambda}^s(\tau)
=
\operatorname{ReLU}\big(e^{-\beta_e(\tau)}-(1-\alpha(\tau))e^{-\beta_i(\tau)}\big)
$$
encodes decay over longer lags, revival under strong attention, and suppression under high inhibitory rates. Case studies on China-AQI report learned variate-to-group assignments, including cities grouped by similar AQI dynamics, and lead-lag edges matching observed delays; one example states that wind direction over Shanghai explains lagged AQI peaks in Ningbo, and that MillGNN detects these lags purely from AQI without exogenous wind inputs while outperforming the listed state-of-the-art methods [2509.03852].

The reported limitations are correspondingly specific. MillGNN depends on the presence of meaningful lead-lag structure; if lead-lag relations are weak or dominated by synchronous interactions, gains may be limited. Performance can degrade under regime shifts and lag distribution drift. Multi-scale graphs and FFT-based cross-correlation add overhead relative to purely channel-independent models, although HiLL-MP is intended to mitigate this. Grouping quality also matters: spectral clustering is described as accurate but costly, whereas K-means offers better efficiency with a modest accuracy trade-off [2509.03852].

In relation to other forecasting architectures, MillGNN is distinguished from graph-based models such as MTGNN, CrossGNN, and MSGNet by explicitly modeling lead-lag edges across patches and scales, using decay-aware dynamic weights informed by real-time inputs and lag magnitude, and employing structured hierarchical message passing through coarse-to-fine duplication. Relative to Transformer-based patching models such as PatchTST, iTransformer, and Crossformer, it retains patching but adds multi-scale lead-lag graphs with explicit lag-aware decay and hierarchical propagation. Relative to decomposition-based approaches such as DLinear, Autoformer, FEDformer, and TimesNet, it is characterized by explicit lag modeling rather than predominantly synchronous trend, seasonal, or frequency decomposition. Relative to single-scale lead-lag methods such as LightMTS/LIFT and VCformer, it advances by discovering and using lead-lag dependencies at multiple grouping scales and by learning dynamic decay-aware weights [2509.03852].

The name “MillGNN” also has a narrower terminological boundary than might be assumed from secondary discussions around graph neural networks for mixed-integer linear programs. The paper "On Representing Mixed-Integer Linear Programs by Graph Neural Networks" studies MILP graph encodings, WL-bounded expressivity, foldable versus unfoldable instances, and random-feature symmetry breaking, but it does not introduce a single named model called “MillGNN” [2210.10759]. A plausible implication is that the established use of “MillGNN” in the arXiv literature refers to the multivariate time series forecasting model rather than to the MILP-representation theory framework.

## 7. Practical use cases and methodological significance

The practical guidance reported for MillGNN centers on systems with hierarchical propagation and evident delays, including transportation networks, power grids, and environmental diffusion, especially when datasets contain many variables and known group structures such as spatial regions or functional subsystems. Recommended starting values from the hyperparameter study are $S=2$ and $K=5$, with larger $S$ suggested for large $N$ when clear group hierarchies are present. Patch lengths $p^s$ should reflect sampling rates and expected lag ranges, with longer patches at coarser scales for longer delays. Spectral clustering is preferred when compute permits, while K-means is suggested for efficiency. Domain priors, such as geography or network topology, may be incorporated into $\boldsymbol{D}^0$ to guide grouping. Efficiency recommendations include restricting candidate pairs to siblings under the same parent at scale $s+1$, keeping $d_e$ and $d_a$ modest, and avoiding bi-directional inter-scale passing [2509.03852].

Methodologically, MillGNN occupies a specific position in the forecasting literature. Its distinctive claim is not simply that graph neural networks improve multivariate time series prediction, but that lead-lag structure should be represented as a causal, multi-scale, lag-aware graph whose edges are initialized by cross-correlation and refined by dynamic, decay-aware weights. The empirical results indicate that this combination is especially effective on datasets with spatially structured propagation, including Traffic, PEMS, and China-AQI. This suggests that the method is particularly well matched to domains in which delayed transmission across related subsystems is an organizing feature of the data-generating process rather than a minor deviation from synchronous dependence [2509.03852].

Source: https://www.emergentmind.com/topics/millgnn