Update README.md

README.md

@@ -1,6 +1,12 @@
 ---
 license: mit
 ---
+<p align="center">
+    📑 <a href="https://arxiv.org/abs/2503.00808" target="_blank">Paper</a> &nbsp&nbsp | &nbsp&nbsp 🔨 <a href="https://huggingface.co/hkust-nlp/preselect-fasttext-classifier" target="_blank">fastText Classifier</a> &nbsp&nbsp | &nbsp&nbsp 🤗 <a href="https://huggingface.co/datasets/hkust-nlp/PreSelect-100B" target="_blank">Released Dataset</a> &nbsp&nbsp | &nbsp&nbsp 📦 <a href="https://github.com/hkust-nlp/PreSelect" target="_blank">Repo</a>  
+<br>
+</p>
+
+
 ## Model Summary
 This is a fastText-based binary classifier for identifying high-quality data in the pretraining corpus introduced in paper:  [Predictive Data Selection: The Data That Predicts Is the Data That Teaches
 ](). And this is also the classifier we used to build [PreSelect-100B](https://huggingface.co/datasets/hkust-nlp/PreSelect-100B) dataset with a selection threshold of 10%.
@@ -40,3 +46,15 @@ dist_executor.run()
 ## Training
 For more training details, you can refer to the paper and the training code is available on GitHub 
 [PreSelect](https://github.com/hkust-nlp/preselect).
+
+## Citation
+If you find this work helpful, please kindly cite as:
+```
+@article{shum2025predictivedataselectiondata,
+      title={Predictive Data Selection: The Data That Predicts Is the Data That Teaches}, 
+      author={Kashun Shum and Yuzhen Huang and Hongjian Zou and Ding Qi and Yixuan Liao and Xiaoxin Chen and Qian Liu and Junxian He},
+      journal={arXiv preprint arXiv:2503.00808},
+      year={2025},
+      eprint={2503.00808},
+}
+```