arXiv:2608.01160cs.LGcs.SI2026-08被引 2

让神经网络自动发现图中高阶结构,提升性能

Differentiable Lifting for Topological Neural Networks

  • 用可微分方法自动学习图的高阶结构(超图、单纯复形)
  • 在多个图分类任务上比传统方法最高提升45%
  • 适用于各类拓扑神经网络,能端到端训练

拓扑神经网络(TNNs)通过利用图中的高阶结构(如环和团)来增强消息传递网络的表达能力。然而,这些结构通常依赖无监督的图提升操作预先确定,这一选择对下游任务性能影响显著。为解决此问题,我们提出∂lift(DiffLift),一种可微分的通用框架,用于端到端学习将图提升为超图及细胞与单纯复形。该方法利用学习到的顶点级潜在表示,识别并参数化候选高阶单元的包含分布,从而实现可扩展建模,并可无缝集成到任意TNN中。实验表明,∂lift在多个图与节点分类基准上优于现有提升方法,相较于静态提升方式(包括基于连接性和特征的)最高提升达45%。

原文摘要 · Abstract (English)

Topological neural networks (TNNs) enable leveraging high-order structures on graphs (e.g., cycles and cliques) to boost the expressive power of message-passing neural networks. In turn, however, these structures are typically identified a priori through an unsupervised graph lifting operation. Notwithstanding, this choice is crucial and may have a drastic impact on a TNN's performance on downstream tasks. To circumvent this issue, we propose $\partial$lift (DiffLift), a general framework for learning graph liftings to hypergraphs and cellular- and simplicial complexes in an end-to-end fashion. In particular, our approach leverages learned vertex-level latent representations to identify and parameterize distributions over candidate higher-order cells for inclusion. This results in a scalable model which can be readily integrated into any TNN. Our experiments show that $\partial$lift outperforms existing lifting methods on multiple benchmarks for graph and node classification across different TNN architectures. Notably, our approach leads to gains of up to 45% over static liftings, including both connectivity- and feature-based ones.

拓扑神经网络图学习可微分提升

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