arXiv:2601.22317cs.LG2026-01

基于对称性与物理规律,用图注意力网络补全网络中缺失流量数据。

FlowSymm: Physics Aware, Symmetry Preserving Graph Attention for Network Flow Completion

  • 利用对称性操作和图注意力机制学习保留观测流量的修复方向。
  • 在三个真实数据集上,均方根误差、平均绝对误差和相关性指标优于现有方法。
  • 适合交通、电力、共享出行等需满足守恒定律的流量补全场景。

恢复网络边上的缺失流量并严格遵守局部守恒定律,是交通、能源和移动系统中的基础逆问题。本文提出FlowSymm,一种新架构:首先将观测数据锚定于最小范数的无散度完成解;随后计算保持观测流量不变的所有容许群作用的正交基,参数化有效解空间(具阿贝尔群结构);再通过堆叠GATv2层编码图结构与边特征,生成缺失边的嵌入,并聚合为每基的注意力权重;该注意力引导过程选择一组保持观测值不变的物理感知群作用;最后通过隐式双层优化求解轻量级Tikhonov正则化,梯度经楚列斯基分解传播。在交通、电力、共享单车三个真实流量基准上,FlowSymm在RMSE、MAE和相关性指标上均超越当前最优方法。

原文摘要 · Abstract (English)

Recovering missing flows on the edges of a network, while exactly respecting local conservation laws, is a fundamental inverse problem that arises in many systems such as transportation, energy, and mobility. We introduce FlowSymm, a novel architecture that combines (i) a group-action on divergence-free flows, (ii) a graph-attention encoder to learn feature-conditioned weights over these symmetry-preserving actions, and (iii) a lightweight Tikhonov refinement solved via implicit bilevel optimization. The method first anchors the given observation on a minimum-norm divergence-free completion. We then compute an orthonormal basis for all admissible group actions that leave the observed flows invariant and parameterize the valid solution subspace, which shows an Abelian group structure under vector addition. A stack of GATv2 layers then encodes the graph and its edge features into per-edge embeddings, which are pooled over the missing edges and produce per-basis attention weights. This attention-guided process selects a set of physics-aware group actions that preserve the observed flows. Finally, a scalar Tikhonov penalty refines the missing entries via a convex least-squares solver, with gradients propagated implicitly through Cholesky factorization. Across three real-world flow benchmarks (traffic, power, bike), FlowSymm outperforms state-of-the-art baselines in RMSE, MAE and correlation metrics.

流量补全图神经网络对称性物理信息

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。