Skip to content
Preprint

TopoFormer: Topology Meets Attention for Graph Learning

Jul 2026 · 0 citations · 104 references
Computer Science Mathematics

TL;DR

This work introduces Topoformer, a lightweight and scalable framework for graph representation learning that encodes topological structure into attention-friendly sequences by decomposing a graph into a short, ordered sequence of topological tokens by slicing over node or edge filtrations.

Abstract

We introduce Topoformer, a lightweight and scalable framework for graph representation learning that encodes topological structure into attention-friendly sequences. At the core of our method is Topo-Scan, a novel module that decomposes a graph into a short, ordered sequence of topological tokens by slicing over node or edge filtrations. These sequences capture multi-scale structural patterns, from local motifs to global organization, and are processed by a Transformer to produce expressive graph-level embeddings. Unlike traditional persistent homology pipelines, Topo-Scan is parallelizable, avoids costly diagram computations, and integrates seamlessly with standard deep learning architectures. We provide theoretical guarantees on the stability of our topological encodings and demonstrate state-of-the-art performance across graph classification and molecular property prediction benchmarks. Our results show that Topoformer matches or exceeds strong GNN and topology-based baselines while offering predictable and efficient compute. This work opens a new path for parallelizable and unifying approaches to graph representation learning that integrate topological inductive biases into attention frameworks.

View source

Similar papers

Preprint Aug 2026

Differentiable Lifting for Topological Neural Networks

This work proposes $\partial$lift (DiffLift), a general framework for learning graph liftings to hypergraphs and cellular- and simplicial complexes in an end-to-end fashion and shows that $\partial$lift outperforms existing lifting methods on multiple benchmarks for graph and node classification across different TNN architectures.

J. L. Franco, Gabriel Duarte, Alexander Nikitin et al. · 2 citations
Preprint Aug 2026

GraphK: Variable-Size Graph Generation with Efficient Edge Construction

Graph generation models have advanced significantly with deep learning, yet they remain limited in scalability, flexibility, and ability to model underlying structures. We present GraphK, a novel encoder-sampler-decoder framework for graph generation that overcomes these challenges through structural flexibility and computational efficiency. Unlike autoregressive approaches constrained by vocabulary size (i.e. number of nodes in graph generation), GraphK allows for both upscaling (generating graphs with more nodes than the input) and downscaling, providing a flexible control over output graph size. By learning permutation-invariant latent representations and sampling new node embeddings via maximum likelihood estimation, GraphK generalizes across graph sizes and structures. For edge generation, we employ edge prediction with a KDTree-based top-k neighbor search in the latent space, reducing computational cost. Based on the manifold smoothness assumption, our method effectively captures graph properties. Experiments on synthetic and real-world datasets show that GraphK outperforms existing methods, accurately learns graph structures, and generates synthetic graphs without explicit definitions.

Resul Tugay, Eren Olug, Elif Ak et al. · 0 citations
Open access Jul 2026

Global graph features unveiled by unsupervised deep learning

Graph Autoencoder Uncovering Descriptive Information (GAUDI), an unsupervised graph deep learning framework designed to capture both local details and global structure, is introduced, providing new insights into emergent phenomena across diverse scientific domains.

Mirja Granfors, Jesús Pineda, Blanca Zufiria-Gerbolés et al. · 0 citations
Book Open access Aug 2026

H4G: Unlocking Faithful Inference for Zero-Shot Graph Learning in Hyperbolic Space

Text-attributed graphs are widely used across domains, offering rich opportunities for zero-shot learning via graph-text alignment. However, existing methods struggle with tasks requiring fine-grained pattern recognition, particularly on heterophilic graphs. Through empirical and theoretical analysis, we identify an over-abstraction problem: current approaches operate at excessively large hyperbolic radii, compressing multi-scale structural information into uniform high-level abstractions. This abstraction-induced information loss obscures critical local patterns essential for accurate predictions. By analyzing embeddings in hyperbolic space, we demonstrate that optimal graph learning requires faithful preservation of fine-grained structural details, better retained by representations positioned closer to the origin. To address this, we propose H4G, a framework that systematically reduces embedding radii using learnable block-diagonal scaling matrices and Möbius matrix multiplication. This approach restores access to fine-grained patterns while maintaining global receptive ability with minimal computational overhead. Experiments show H4G achieves state-of-the-art zero-shot performance with 12.8% improvement on heterophilic graphs and 8.4% on homophilic graphs, confirming that radius reduction enables faithful multi-scale representation for advancing zero-shot graph learning.

Heng Zhang, Jin Huang · 0 citations
Book Open access Aug 2026

Topology-Aware Neural Collapse for Generalized Category Discovery on Graphs

Learning a stable yet highly discriminative representation space that can simultaneously recognize known categories and discover novel ones from limited labeled data is fundamental to Generalized Category Discovery (GCD) on graphs. Recently, Neural Collapse (NC) theory has emerged as a powerful geometric principle for GCD, yielding maximally separated and well-structured class representations by encouraging feature embeddings to converge toward Simplex Equiangular Tight Frame (Simplex ETF) prototypes. However, when extending this paradigm to graph-structured data, a critical challenge arises. Graph representations are inherently shaped by topological dependencies, where neighborhood-based message passing enforces local smoothness among connected nodes. This topology-induced smoothing conflicts with the strict geometric convergence required by Simplex ETF, making neural collapse difficult to realize on graphs. To address this issue, we propose TopoNC, a topology-aware neural collapse framework for Graph GCD. Specifically, it fixes Simplex ETF prototypes as global geometric targets and introduces a Dual-Stream Encoder that decouples topology smoothing from feature-discriminative learning, adaptively balancing the two streams via a gating mechanism. In addition, we further design a Topology-Conditioned Pseudo-Labeling strategy that integrates Sinkhorn-based global balancing, Old-Class Top-K Admission Masking, and Neighborhood-Consensus Screening, to reliably guide feature collapse. Extensive experiments on several benchmark datasets have demonstrated that TopoNC consistently outperforms existing methods, highlighting the importance of topology-aware neural collapse for Graph GCD.

Xu Xi, Zhong Zhang, Hongliang Wang et al. · 0 citations
Book Open access Jul 2026

One-for-All Community Search on Unseen Graphs

A Spectral-Aware Feature Alignment module to unify feature dimensionality and align cross-domain semantics in a community-aware manner and a Graph Diffusion Tokenized Transformer that constructs hybrid token sequences from local and global structural contexts for Transformer encoding, and applies diffusion-based refinement to mitigate distribution shifts on unseen graphs.

Mo Li, Zhaosong Zhao, Linlin Ding et al. · 0 citations