用量子行走机制解决图神经网络过平滑和异质图性能差的问题
Continuous-Time Quantum Walks based Graph Neural Network

- 基于连续时间量子行走设计单位性传播,保留高频信号
- 三模块融合提升异质/同质图表现,避免特征衰减
- 理论支撑选时策略,适合复杂图结构任务
图神经网络在图数据上广泛应用,但普遍存在两大缺陷:一是在同质假设下消息传递呈低通滤波,导致异质图性能差;二是堆叠层数使节点特征趋于常数,引发过平滑。现有方法多分治处理,少数联合方案依赖经验启发,且常牺牲模型表达能力。本文提出基于连续时间量子行走(CTQW)的GNN(CTQW-GNN),利用其传播算子 $e^{- ext{i}Ht}$ 的两个特性:一是酉性且特征值在单位圆上,不衰减任何频率分量,克服低通偏差;二是酉性保持特征范数,抑制狄利克雷能量随深度指数衰减,缓解过平滑。模型包含三个互补聚合模块:基于CTQW的聚合保留中高频信号,适配异质图;CTQW-注意力聚合基于量子振幅构建多跳邻域图并施加注意力,捕获远距离同质节点;低通聚合(LF)采用标准低通GAT分支,维持同质图性能。此外,我们提供谱隙分析解释能量保持机制,并给出类似李布-罗宾逊的界,为行走时间 $t$ 提供理论选择依据。
原文摘要 · Abstract (English)
Graph Neural Networks (GNNs) are widely used on graph-structured data, but most suffer from two key weaknesses. First, message passing behaves as a low-pass filter under the homophily assumption, leading to poor performance on heterophilic graphs. Second, stacking layers drives node features toward constants, causing over-smoothing. Existing methods usually address these issues separately, while the few joint solutions rely largely on empirical heuristics, and many over-smoothing remedies sacrifice model expressiveness. We propose \textbf{CTQW-GNN}, a GNN based on Continuous-Time Quantum Walks (CTQW), to address both issues with theoretical justification. Its design exploits two properties of the CTQW propagator $e^{-\mathrm{i}Ht}$. First, it is unitary and has eigenvalues on the unit circle, so no frequency component is damped, counteracting the low-pass bias. Second, unitarity preserves feature norms and prevents the Dirichlet energy from decaying exponentially with depth, thereby mitigating over-smoothing. CTQW-GNN combines three complementary aggregation modules. \textit{CTQW-based Aggregation} evolves node features through the unitary propagator, preserving mid- and high-frequency signals for heterophilic graphs while preventing Dirichlet-energy collapse. \textit{CTQW-Attention Aggregation} constructs a multi-hop neighbor graph from CTQW amplitudes and applies attention over it, enabling access to distant homophilic nodes missed by single-hop aggregation. \textit{LF Aggregation} uses a standard low-pass GAT branch to retain strong performance on homophilic graphs, where pure CTQW aggregation can be suboptimal. We further provide a spectral-gap analysis explaining energy preservation and a Lieb--Robinson-type bound that gives a principled rule for selecting the walk time $t$.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。