Update README.md

README.md

@@ -100,4 +100,18 @@ outputs = llm.generate(["Explain speculative decoding in simple terms."], sampli
 
 This model was trained using outputs from [meta-llama/Llama-3.1-8B-Instruct](https://huggingface.co/meta-llama/Llama-3.1-8B-Instruct). Use of this model is additionally subject to the [Llama 3.1 Community License Agreement](https://github.com/meta-llama/llama-models/blob/main/models/llama3_1/LICENSE).
 
-Llama 3.1 is licensed under the Llama 3.1 Community License, Copyright © Meta Platforms, Inc. All Rights Reserved.
+Llama 3.1 is licensed under the Llama 3.1 Community License, Copyright © Meta Platforms, Inc. All Rights Reserved.
+
+## Citation
+
+```
+@misc{samarin2026lklosses,
+  title     = {LK Losses: Direct Acceptance Rate Optimization for Speculative Decoding},
+  author    = {Alexander Samarin and Sergei Krutikov and Anton Shevtsov and Sergei Skvortsov and Filipp Fisin and Alexander Golubev},
+  year      = {2026},
+  eprint    = {2602.23881},
+  archivePrefix = {arXiv},
+  primaryClass  = {cs.LG},
+  url       = {https://arxiv.org/abs/2602.23881}
+}
+```