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

Communicability-Inspired Positional Encoding (CIPE)

About

Positional encodings (PEs) are essential for Transformers. Yet designing effective PEs for non-Euclidean graphs remains challenging. Such encodings should ideally induce an Attention-Compatible Geometry for self-attention: not merely describing graph structure, but defining a geometry whose inner products reflect meaningful structural relatedness. To realize this geometry, we propose Communicability-Inspired Positional Encoding (CIPE), built from communicability, a measure between pairs of nodes that aggregates contributions from paths of all lengths. By construction, CIPE inner products recover communicability, converting global multi-path connectivity into an attention-ready similarity geometry. For practical Transformer training, we introduce dimensionality alignment, mapping graph-size-dependent CIPE representations to prescribed dimensions while faithfully preserving the induced geometry. Empirically, CIPE improves structure-agnostic Transformers by 35.5% on average across seven benchmarks, outperforming representative PEs; it also consistently improves structure-biased graph Transformers, where competing PEs often yield only marginal benefits. These results position CIPE as a principled framework for attention-compatible graph positional encodings.

Yipeng Zhang, Zhongtian Sun, Pietro Li\`o, Kelin Xia• 2026

Related benchmarks

TaskDatasetResultRank
Molecular property predictionMoleculeNet BBBP (scaffold)
ROC AUC76.6
158
Molecular property predictionMoleculeNet SIDER (scaffold)
ROC-AUC0.849
136
Molecular property predictionMoleculeNet BACE (scaffold)
ROC-AUC85.5
126
Molecular property predictionMoleculeNet MUV (scaffold)
ROC-AUC0.718
107
Molecular property predictionTox21 scaffold split
ROC-AUC79.5
56
Molecular property predictionHIV (scaffold split)
ROC-AUC79.9
56
Molecular property predictionClinTox scaffold split
ROC-AUC0.984
16
Showing 7 of 7 rows

Other info

Follow for update