Update README.md

README.md

@@ -118,8 +118,14 @@ response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
 
 ## Citation
 
-If you find our work helpful, feel free to give us a cite.
-
-```
-// todo
+```bibtex
+@misc{aggarwal2025robustlearningdiversecode,
+      title={Robust Learning of Diverse Code Edits}, 
+      author={Tushar Aggarwal and Swayam Singh and Abhijeet Awasthi and Aditya Kanade and Nagarajan Natarajan},
+      year={2025},
+      eprint={2503.03656},
+      archivePrefix={arXiv},
+      primaryClass={cs.SE},
+      url={https://arxiv.org/abs/2503.03656}, 
+}
 ```