Speedrunning Relational Foundation Model Pretraining
Organizations: Kunumi Institute, Belo Horizonte, Minas Gerais, Brazil · Universidade Federal de Minas Gerais, Belo Horizonte, Minas Gerais, Brazil
Abstract
Progress on foundation model pretraining is gated by who can afford a trial, and relational pretraining is no exception. Recipes are developed inside individual labs, reported under private protocols, and do not accumulate. Speedrun leaderboards remove that barrier by decentralizing the search rather than the training run. The problem is fixed, the stack is open, entries are ranked by wall-clock time to a quality target, and one GPU is enough to hold the record. We introduce modded-nanoRT, the first such leaderboard for Relational Transformers. Which database to hold out decides whether the benchmark can resolve improvements at all, so we fix it by calibration, ranking six candidates by how sharply a 45-minute learning curve separates improvement from seed noise. We open the leaderboard with typed trilinear attention, a relational operator that conditions attention scores and values on the foreign key traversed, together with Muon and SIGReg. The recipe reaches the target in 305.63 seconds on one NVIDIA H200, 8.11x faster than the reference implementation, and wins 13 of 15 pairs outside the development database. We will release the benchmark, its frozen artifacts, and the record runs. Our code is available at https://github.com/kunumi/nanoRT.
Figures & tables
| Pretraining tasks | RelBench v1 binary classification tasks, minus rel-Avito and rel-Event |
| Mixture | temperature sampling, |
| Held-out database | rel-Avito |
| Metric | pooled AUROC on rel-Avito |
| Target | |
| Reference model | RT, 12 blocks, width 256, 8 heads |
| Trainable parameters | 19.3M |
| Component | Paired contrast | AUROC (pp) | Wins |
| Muon | on AdamW RT | 3/3 | |
| with typed attention | 3/3 | ||
| Typed attention | on Muon SIGReg | 2/3 | |
| on AdamW | 1/3 | ||
| SIGReg | before typed attention | 0/3 | |
| with typed attention | 3/3 |
| Held-out DB | AdamW RT | Final | (pp) | Wins |
| rel-Amazon | 58.45 | 58.17 | 2/3 | |
| rel-Avito | 80.21 | 83.15 | 2/3 | |
| rel-F1 | 79.55 | 84.58 | 3/3 | |
| rel-HM | 62.77 | 65.14 | 3/3 | |
| rel-Stack | 75.07 | 76.96 | 3/3 | |
| rel-Trial | 51.50 | 52.38 | 2/3 |
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
| Relation vector | Best AUROC | Steps 100–900 | Reached target |
| Semantic FK (record recipe) | 5/7 | ||
| Shuffled FK | 5/7 | ||
| Random per type/layer | 7/7 | ||
| Random per edge | 3/7 |
Relational Foundation Models
Relational foundation models learn reusable representations and predictors for tasks spanning linked tables and databases.
See all 17 papers