Added paper details

README.md

@@ -24,3 +24,24 @@ statement goal.
 
 ## Training
 - PyTorch Lightning
+
+## 📄 Related Paper & Citation
+
+For more details on the methodology, data augmentation strategy, and dynamic sampling used for Lean theorem proving, please refer to our paper:
+
+**Enhancing Neural Theorem Proving through Data Augmentation and Dynamic Sampling Method**  
+Rahul Vishwakarma, Subhankar Mishra  
+*arXiv:2312.14188 (cs.AI, cs.LG, cs.LO)*
+
+🔗 **Paper:** https://arxiv.org/abs/2312.14188
+
+If you use this model or build upon this work, please cite:
+
+```bibtex
+@article{vishwakarma2023enhancing,
+  title={Enhancing Neural Theorem Proving through Data Augmentation and Dynamic Sampling Method},
+  author={Vishwakarma, Rahul and Mishra, Subhankar},
+  journal={arXiv preprint arXiv:2312.14188},
+  year={2023},
+  primaryClass={cs.AI}
+}