English

What Structural Inductive Bias Helps Transformers Reason Over Knowledge Graphs? A Study with Tabula RASA

Machine Learning 2026-05-12 v3 Artificial Intelligence

Abstract

What structural inductive bias helps transformers reason over knowledge graphs? Through controlled ablations of a minimal transformer modification with four independently removable components (sparse adjacency masking, edge-type biases, query scaling, value gating), we isolate which structural signals drive multi-hop reasoning. Our finding is sharp: sparse adjacency masking alone accounts for the dominant share of improvement over unmasked transformers (+72.5pp on 3-hop MetaQA, +45.5pp on WebQSP, +53.9pp on CWQ), while learned relation parameters add only modest refinement and can actively hurt without structural guidance. A zero-shot experiment provides architecturally independent corroboration: masking-based attention degrades 4.0x less than relation-specific weights when edge types are held out. The useful inductive bias for multi-hop KGQA is predominantly topological, not relational.

Keywords

Cite

@article{arxiv.2602.02834,
  title  = {What Structural Inductive Bias Helps Transformers Reason Over Knowledge Graphs? A Study with Tabula RASA},
  author = {Jonas Petersen and Camilla Mazzoleni and Gian-Alessandro Lombardi and Federico Martelli and Riccardo Maggioni},
  journal= {arXiv preprint arXiv:2602.02834},
  year   = {2026}
}

Comments

8 pages + appendix, 9 figures

R2 v1 2026-07-01T09:33:04.452Z