From 9b3e2cf767441018d11fa0d2c339aedf7d37ee50 Mon Sep 17 00:00:00 2001
From: Josh Gardner <jpgard@users.noreply.github.com>
Date: Thu, 14 Dec 2023 20:09:11 -0600
Subject: [PATCH] Update README.md

---
 README.md | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 2852c6d74f..3818a0e16d 100644
--- a/README.md
+++ b/README.md
@@ -8,7 +8,16 @@
 
 TableShift is a benchmarking library for machine learning with tabular data under distribution shift.
 
-You can read more about TableShift at [tableshift.org](https://tableshift.org/index.html) or read the full paper (published in NeurIPS 2023 Datasets & Benchmarks Track) on [arxiv](https://arxiv.org/abs/2312.07577).
+You can read more about TableShift at [tableshift.org](https://tableshift.org/index.html) or read the full paper (published in NeurIPS 2023 Datasets & Benchmarks Track) on [arxiv](https://arxiv.org/abs/2312.07577). If you use the benchmark in your research, please cite the paper:
+
+```
+@article{gardner2023tableshift,
+  title={Benchmarking Distribution Shift in Tabular Data with TableShift},
+  author={Gardner, Josh and Popovic, Zoran and Schmidt, Ludwig},
+  journal={Advances in Neural Information Processing Systems},
+  year={2023}
+}
+```
 
 If you find an issue, please file a GitHub [issue](https://github.com/mlfoundations/tableshift/issues/new/choose).