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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Molecular property prediction | MoleculeNet BBBP (scaffold) | ROC AUC76.6 | 158 | |
| Molecular property prediction | MoleculeNet SIDER (scaffold) | ROC-AUC0.849 | 136 | |
| Molecular property prediction | MoleculeNet BACE (scaffold) | ROC-AUC85.5 | 126 | |
| Molecular property prediction | MoleculeNet MUV (scaffold) | ROC-AUC0.718 | 107 | |
| Molecular property prediction | Tox21 scaffold split | ROC-AUC79.5 | 56 | |
| Molecular property prediction | HIV (scaffold split) | ROC-AUC79.9 | 56 | |
| Molecular property prediction | ClinTox scaffold split | ROC-AUC0.984 | 16 |