Update README.md

README.md

@@ -31,4 +31,20 @@ import torch
 # Load transcoders into a ReplacementModel
 model = ReplacementModel.from_pretrained("meta-llama/Llama-3.1-8B-Instruct", "facebook/crv-8b-instruct-transcoders", dtype=torch.bfloat16)
 ```
-Once you have loaded the model, you can perform attribution or intervention as shown in [this demo](https://github.com/safety-research/circuit-tracer/blob/main/demos/llama_demo.ipynb).
+Once you have loaded the model, you can perform attribution or intervention as shown in [this demo](https://github.com/safety-research/circuit-tracer/blob/main/demos/llama_demo.ipynb).
+
+## Citation
+
+If you use this model, please cite our paper:
+
+```bibtex
+@article{zhao2025verifying,
+      title={Verifying Chain-of-Thought Reasoning via Its Computational Graph},
+      author={Zheng Zhao and Yeskendir Koishekenov and Xianjun Yang and Naila Murray and Nicola Cancedda},
+      year={2025},
+      eprint={2510.09312},
+      archivePrefix={arXiv},
+      primaryClass={cs.CL},
+      url={https://arxiv.org/abs/2510.09312},
+}
+```