Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

X-LogSMask: Expand Transformer for Graph-Structured Data

About

Transformers have become general-purpose architectures, but their all-to-all self-attention is poorly matched to graph data, whose interactions are sparse, structured and multi-scale. Existing Graph Transformers address this mismatch through structural encodings, hybrid message-passing modules or learned attention constraints, often introducing additional complexity and limited interpretability. Here we introduce X-LogSMask, an explainable multi-head logarithmic structural mask that injects symmetrically normalized graph topology directly into attention logits. The logarithmic transform converts structural connectivity into a topology-aware gating signal, suppressing unsupported node interactions while preserving feature-dependent attention. By assigning different powers of the normalized adjacency matrix to different attention heads, X-LogSMask gives each head a defined structural radius and supports multi-hop information propagation within a single layer. We further show that a standard Transformer encoder can be interpreted as one-step message passing on a complete graph, motivating X-LogSMask as a topology-constrained alternative to unrestricted self-attention. Across 20 node-, edge- and graph-level benchmarks, Transformers equipped with X-LogSMask achieve state-of-the-art performance on 13 datasets and remain competitive in a lightweight one-layer configuration. These results show that simple, interpretable structural masks can make self-attention an effective graph-learning operator without changing the Transformer architecture. The code is available at https://github.com/LiLeyan-0120/X-LogSMask.

Leyan Li, Rennong Yang, Zhenxing Zhang, Liping Hu• 2026

Related benchmarks

TaskDatasetResultRank
Graph ClassificationPROTEINS
Accuracy80.63
1383
Graph ClassificationMUTAG
Accuracy88.89
1229
Graph ClassificationNCI1
Accuracy82.24
707
Graph ClassificationCOLLAB
Accuracy80.8
532
Graph ClassificationIMDB-B
Accuracy77
455
Graph ClassificationD&D
Accuracy81.2
179
Graph ClassificationMolHIV
ROC AUC78.91
110
Node ClassificationwikiCS
Accuracy (WikiCS)80.36
101
Node ClassificationPhysics
Accuracy97.68
91
Node ClassificationComputers
Accuracy (%)92.01
74
Showing 10 of 19 rows

Other info

Follow for update