Update README.md

README.md

@@ -107,3 +107,19 @@ The following hyperparameters were used during training:
 - Pytorch 2.1.0+cu121
 - Datasets 2.14.6
 - Tokenizers 0.14.1
+
+
+## Citation
+If you find the content of this project helpful, please cite our paper as follows:
+
+```
+@misc{liu2023what,
+      title={What Makes Good Data for Alignment? A Comprehensive Study of Automatic Data Selection in Instruction Tuning}, 
+      author={Wei Liu and Weihao Zeng and Keqing He and Yong Jiang and Junxian He},
+      year={2023},
+      eprint={2312.15685},
+      archivePrefix={arXiv},
+      primaryClass={cs.CL}
+}
+```
+