Add pipeline tag and link to original Relational Transformer paper

README.md

@@ -1,6 +1,12 @@
 ---
-license: mit
+datasets:
+- kvignesh1420/plurel
 library_name: pytorch
+license: mit
+pipeline_tag: other
+metrics:
+- roc_auc
+- r_squared
 tags:
 - relational-data
 - tabular
@@ -9,30 +15,34 @@ tags:
 - relational-transformer
 - relbench
 - synthetic-data
-datasets:
-- kvignesh1420/plurel
-metrics:
-- roc_auc
-- r_squared
 ---
 
 # Relational Transformer — PluRel Checkpoints
 
-Relational Transformer (RT) model checkpoints pretrained on synthetic relational databases generated by [PluRel](https://huggingface.co/datasets/kvignesh1420/plurel), as introduced in:
+Relational Transformer (RT) model checkpoints pretrained on synthetic relational databases generated by [PluRel](https://huggingface.co/datasets/kvignesh1420/plurel). 
+
+Relational Transformer is a foundation model architecture for relational data that enables zero-shot transfer across heterogeneous schemas and tasks. It was introduced in:
+
+> **Relational Transformer: Toward Zero-Shot Foundation Models for Relational Data**  
+> Rishabh Ranjan, Valter Hudovernik, Mark Znidar, Charilaos Kanatsoulis, Roshan Upendra, Mahmoud Mohammadi, Joe Meyer, Tom Palczewski, Carlos Guestrin, Jure Leskovec — [arXiv:2510.06377](https://arxiv.org/abs/2510.06377) (ICLR 2026)
+
+The checkpoints provided in this repository were trained using the methodology described in:
 
-> **PluRel: Synthetic Data unlocks Scaling Laws for Relational Foundation Models**
-> Kothapalli, Ranjan, Hudovernik, Dwivedi, Hoffart, Guestrin, Leskovec — arXiv:2602.04029 (2026)
+> **PluRel: Synthetic Data unlocks Scaling Laws for Relational Foundation Models**  
+> Kothapalli, Ranjan, Hudovernik, Dwivedi, Hoffart, Guestrin, Leskovec — [arXiv:2602.04029](https://arxiv.org/abs/2602.04029) (2026)
 
-[![arXiv](https://img.shields.io/badge/arXiv-2602.04029-b31b1b?style=flat&logo=arxiv)](https://arxiv.org/abs/2602.04029)
-[![Project Page](https://img.shields.io/badge/Project-Page-blue?style=flat&logo=github)](https://snap-stanford.github.io/plurel/)
-[![GitHub](https://img.shields.io/badge/Code-GitHub-black?style=flat&logo=github)](https://github.com/snap-stanford/plurel)
+[![arXiv (RT)](https://img.shields.io/badge/arXiv-2510.06377-b31b1b?style=flat&logo=arxiv)](https://arxiv.org/abs/2510.06377)
+[![GitHub (RT)](https://img.shields.io/badge/Code-RT_GitHub-black?style=flat&logo=github)](https://github.com/snap-stanford/relational-transformer)
+[![arXiv (PluRel)](https://img.shields.io/badge/arXiv-2602.04029-b31b1b?style=flat&logo=arxiv)](https://arxiv.org/abs/2602.04029)
+[![Project Page (PluRel)](https://img.shields.io/badge/Project-Page-blue?style=flat&logo=github)](https://snap-stanford.github.io/plurel/)
+[![GitHub (PluRel)](https://img.shields.io/badge/Code-PluRel_GitHub-black?style=flat&logo=github)](https://github.com/snap-stanford/plurel)
 [![Dataset](https://img.shields.io/badge/Dataset-HuggingFace-yellow?style=flat&logo=huggingface)](https://huggingface.co/datasets/kvignesh1420/plurel)
 
 ---
 
 ## Model Architecture
 
-The Relational Transformer operates on multi-tabular relational databases, treating rows across linked tables as a sequence via BFS-ordered context sampling.
+The Relational Transformer operates on multi-tabular relational databases, treating rows across linked tables as a sequence via BFS-ordered context sampling. It utilizes a Relational Attention mechanism over columns, rows, and primary-foreign key links.
 
 | Hyperparameter | Value |
 |----------------|-------|
@@ -44,7 +54,7 @@ The Relational Transformer operates on multi-tabular relational databases, treat
 | Text encoder | `all-MiniLM-L12-v2` (d_text = 384) |
 | Max BFS width | 128 |
 
-The architecture and training loop build on the [Relational Transformer (ICLR 2026)](https://arxiv.org/abs/2510.06377) codebase.
+The architecture and training loop build on the [Relational Transformer](https://github.com/snap-stanford/relational-transformer) codebase.
 
 ---
 
@@ -53,4 +63,5 @@ The architecture and training loop build on the [Relational Transformer (ICLR 20
 ```bash
 huggingface-cli download kvignesh1420/relational-transformer-plurel \
     --repo-type model \
-    --local-dir ~/scratch/rt_hf_ckpts
+    --local-dir ~/scratch/rt_hf_ckpts
+```