Physics-Refined Spatiotemporal Forecasting on Open-Boundary Hydrologic Graphs

cs.LG Haoyang Jiang, Zhengui Wang, Shenghan Gao, Y. Joseph Zhang, Xingquan Zhu, Yi He · Oct 1, 2026
Local to this browser
What it does
Open-boundary hydrologic graphs such as river networks and coastal meshes exchange fluxes with unobserved exteriors, making long-horizon forecasting unstable because boundary forcing is unavailable at prediction time. This work proposes...
Why it matters
This work proposes PRSOBG, which learns ghost-node proxies to approximate missing external forcing and applies local and global physics refiners to suppress error amplification during autoregressive rollout. Evaluated on the Danube river...
Main concern
The paper presents a well-motivated framework for an important problem and delivers strong empirical gains on two distinct hydrologic topologies, yet it overstates novelty and leaves key physical claims on real data essentially unverified....
Community signal
0
0 up · 0 down
Sign in to vote with arrows
AI Review AI reviewed
Plain-language introduction

Open-boundary hydrologic graphs such as river networks and coastal meshes exchange fluxes with unobserved exteriors, making long-horizon forecasting unstable because boundary forcing is unavailable at prediction time. This work proposes PRSOBG, which learns ghost-node proxies to approximate missing external forcing and applies local and global physics refiners to suppress error amplification during autoregressive rollout. Evaluated on the Danube river system and Chesapeake Bay, the framework cuts RMSE by up to 51% over a vanilla STGNN backbone with only about 8% training-time overhead.

Critical review
Verdict
Bottom line

The paper presents a well-motivated framework for an important problem and delivers strong empirical gains on two distinct hydrologic topologies, yet it overstates novelty and leaves key physical claims on real data essentially unverified. The two-tier "what-how" decomposition is conceptually clean, but the authors' assertion that this is the first such formalization is questionable given prior boundary-aware surrogates, and the learned ghost proxies are validated only on a ten-node toy system.

“To the best of our knowledge, this is the first study formalizing and decomposing the spatiotemporal forecasting on open-boundary graphs as a two-tier what-how challenge.”
paper · Section I, Contributions
What holds up

The empirical evidence for the core design is strong and consistent across topologies. On the Danube directed graph, adding ghost proxies alone reduces RMSE by 20.63% at $T=6$ and 18.37% at $T=30$, and the full PRSOBG achieves up to 33.95% improvement. The convexified semi-implicit refiner consistently outperforms both the explicit variant and prior physics-informed baselines (PDE-Refiner, PhysicsCorrect) with only modest computational overhead (1.08x train time relative to the backbone), suggesting that anchoring corrections to boundary-refined predictions while imposing physics-structured regularization is a robust recipe for long-horizon rollout.

“On River System, Ghost-Only reduces RMSE from 0.3499 to 0.2777 at T=6 ... This indicates that explicitly representing missing source-node boundary forcing helps reduce persistent boundary-side error injection under both short and long rollout.”
paper · Section IV, RQ2
“Ours-Convexified achieves RMSE values of 0.2311, 0.3079, and 0.4268 at T=6, T=16, and T=30, yielding 33.95%, 25.75%, and 27.95% improvements over the STGNN backbone.”
paper · Section IV, RQ3
Main concerns

Several limitations temper the paper's claims. First, the ghost proxies are validated only on a synthetic linearized 1D shallow-water chain with ten nodes, and the authors correctly concede that "these results do not establish strict physical identifiability"; on real data, the ghosts are unsupervised and their physical meaning is assumed rather than demonstrated. Second, the global physics refiners invoke highly simplified priors—an advective prior for rivers and a linearized barotropic-wave prior for the coastal mesh—that can mismatch real dynamics, and the paper offers no quantification of this mismatch. Third, sensitivity and extreme-flow analyses are confined to the smaller river dataset at short horizons, leaving robustness on the large Chesapeake mesh under rare events unaddressed.

“These results do not establish strict physical identifiability, but they provide direct and falsifiable evidence that the learned ghost is more than a generic smoothing variable.”
paper · Section IV, RQ4
“when simplified dynamics priors mismatch real hydrologic dynamics”
paper · Section I, last paragraph before Contributions
Evidence and comparison

The reported comparisons broadly support PRSOBG's superiority over nine baselines. A particularly valuable diagnostic is that a temporal-only GRU outperforms the STGNN backbone on the Danube, with the authors arguing that "adding spatial message passing alone does not resolve the missing open-boundary information and can amplify boundary-induced errors." However, this also raises the possibility that the backbone itself is a weak baseline. The physics-informed competitors (FAIR, PDE-Refiner, PhysicsCorrect) are adapted from different problem settings; the paper notes configurations follow original papers "when applicable," but the exact adaptations and hyperparameter searches are not detailed, making fair assessment difficult.

“Since both models use the same GRU temporal backbone, the inferior performance of STGNN indicates that adding spatial message passing alone does not resolve the missing open-boundary information and can amplify boundary-induced errors during rollout.”
paper · Section IV, RQ1
“For all competitors, we follow the original papers for model configurations when applicable”
paper · Section IV, Implementation
Reproducibility

The authors release code, which is positive, but several details needed for independent reproduction are missing: train/validation/test splits are not stated, learning rates and batch sizes are omitted, and the preprocessing pipeline from raw data to the final graphs is undescribed. The convexified refiner uses only two PCG iterations by default, yet there is no analysis of solver convergence or residual norms, so practitioners cannot judge whether the approximate solution is accurate enough for other meshes. Additionally, the default ghost MLP is described as a "2-layer" architecture in Section IV but listed as "Depth 3" in Table VI, a minor inconsistency that should be clarified.

“Our code is available at https://github.com/HaoyangJiang-WM/OpenBoundary”
paper · Abstract
“we solve the resulting SPD system using a differentiable unrolled PCG layer with two iterations by default”
paper · Section IV, Implementation
Abstract

Spatiotemporal forecasting on hydrologic graphs is especially prone to instability in open-boundary systems, where the forecast domain exchanges fluxes with an unobserved exterior. In such systems, boundary nodes receive external forcing, e.g., upstream inflows in rivers or tidal signals in coastal regions, that is typically unavailable at prediction time. The absence of this information can compound errors as forecasts unfold in an autoregressive fashion, leading to inferior long-horizon performance. This paper dissects this instability issue by exploring two questions. 1) What boundary forcing enters the forecast domain when information beyond the boundary is missing? 2) How should this forcing propagate through the domain without incurring error amplification under autoregressive rollout? To address both, we propose a new computing framework comprising two key components. First, to compensate for the boundary forcing, our framework learns ghost node proxies from the boundary and interior nodes, striving to approximate unobserved external inputs. Second, to control error accumulation from these learned proxies, we leverage two physics refiners. In particular, one refiner enforces local consistency by aligning ghost proxies with their two-hop neighbors (i.e., boundary nodes and their immediate interiors). The other refiner enhances global stability by correcting the model forecasts through a physics-guided graph neural operator, reducing long-horizon numerical drift. Two real-world hydrologic graphs are employed for empirical evaluation. Comparative results show that our proposal enjoys higher prediction accuracy and long-horizon stability over both learning-based and physics-informed model competitors.

Challenge the Review

Pick a starting point or write your own. Challenges run in the background, so you can keep reading while the AI investigates.

No challenges yet. Disagree with the review? Ask the AI to revisit a specific claim.