CondPSE通过条件调制增强图结构编码,显著提升对复杂图结构的区分能力。
CondPSE: A Polynomial-Filtered Structural Encoder with Conditional Modulation for Graphs
- 用可学习的多项式滤波器组处理节点探针,结合条件调制生成结构响应
- 在合成数据上将分类准确率从42.9%提升至97.3%,超越基线GPSE
- 适合需要精细图结构区分的任务,如图同构检测与拓扑分析
消息传递图神经网络受限于1-WL测试,可能忽略区分非同构图的拓扑结构。位置与结构编码(PSE)引入源自拓扑的信号,如GPSE这类预训练编码器通过随机节点探针学习生成这些信号,随后冻结作为下游模型输入。本文提出CondPSE,它使用可学习的多项式图滤波器组处理标准高斯节点探针,并通过基于跨滤波、局部消息传递和图级信号的FiLM式调制优化结构响应分支。CondPSE在节点级位置/结构目标和图级不变量上进行预训练,随后冻结用于下游任务。在合成结构区分基准上,CondPSE成功区分了1-WL无法分辨的图结构:相比GPSE,CSL准确率由42.9%提升至97.3%,EXP准确率由68.3%提升至99.9%;消融实验表明,多项式滤波器组贡献了主要性能提升。在真实分子属性预测任务中表现有限:采用混合局部消息传递/全局注意力骨干网络时,其性能与GPSE相当,未明显超越;在ZINC数据集上的对比也未显示出一致优劣。本文报告结果并讨论为何强合成结构区分能力并不自动带来下游优势,涉及下游集成方式及结构预训练目标与分子标签间的潜在不匹配。
原文摘要 · Abstract (English)
Message-passing graph neural networks are bounded by the 1-WL test and can miss topological structure that distinguishes non-isomorphic graphs. Positional and structural encodings (PSE) inject such topology-derived signals, and learned PSE encoders such as GPSE pretrain a single encoder to produce these signals from random node probes, which can then be frozen and reused as inputs across downstream graph models. We present CondPSE, a learned PSE encoder that applies a learnable polynomial graph filter bank to standard Gaussian node probes and refines the resulting structural-response branches through FiLM-style modulation conditioned on cross-filter, local message-passing, and graph-level signals. CondPSE is pretrained to reconstruct node-level positional/structural targets and graph-level invariants, and is then frozen for use as a downstream input encoding. On synthetic structural-discrimination benchmarks, CondPSE separates graph structures that 1-WL-bounded message passing cannot: it raises CSL accuracy from 42.9% to 97.3% and EXP accuracy from 68.3% to 99.9% relative to GPSE, and ablations show that the polynomial filter bank accounts for most of this gain. On real molecular property prediction, the picture is more limited. With a hybrid local-message-passing/global-attention backbone, CondPSE performs comparably to GPSE without surpassing it, and a ZINC backbone sweep shows no consistent ordering between the two encoders. We report these results and discuss why strong synthetic structural discrimination does not, on its own, yield a downstream advantage for frozen learned PSE encoders, including the role of downstream integration and possible mismatch between structural pretraining targets and molecular property labels.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。