Update README.md

README.md

@@ -12,3 +12,17 @@ This is the original 3B BlockFFN checkpoint used in the paper *BlockFFN: Towards
 You can load and use this model simply by using `AutoTokenizer` and `AutoModelForCausalLM`.
 
 Links: [[Paper](https://arxiv.org/pdf/2507.08771)] [[Codes](https://github.com/thunlp/BlockFFN)]
+
+### Citation
+
+If you find our work useful for your research, please kindly cite our paper as follows:
+
+```
+@article{song2025blockffn,
+      title={{BlockFFN}: Towards End-Side Acceleration-Friendly Mixture-of-Experts with Chunk-Level Activation Sparsity}, 
+      author={Chenyang Song and Weilin Zhao and Xu Han and Chaojun Xiao and Yingfa Chen and Yuxuan Li and Zhiyuan Liu and Maosong Sun},
+      journal={arXiv preprint arXiv:2507.08771},
+      year={2025},
+      url={https://arxiv.org/pdf/2507.08771}, 
+}
+```