Upload README.md with huggingface_hub

README.md

@@ -181,12 +181,18 @@ This model predicts **specific (child) CWE categories** where possible. For exam
 - **Description-based**: Uses only the text description, not CVSS scores, CPE, or other metadata.
 - **Single-label**: Predicts one CWE per CVE, though some vulnerabilities may involve multiple weakness types.
 
+## Paper
+
+📄 **[Fine-tuning RoBERTa for CVE-to-CWE Classification: A 125M Parameter Model Competitive with LLMs](https://arxiv.org/abs/2603.14911)**
+
 ## Citation
 
 ```bibtex
-@model{cve_to_cwe_classifier_2025,
-  title={CWE Classifier (RoBERTa-base)},
-  year={2025},
+@article{mosievskiy2026cwe,
+  title={Fine-tuning RoBERTa for CVE-to-CWE Classification: A 125M Parameter Model Competitive with LLMs},
+  author={Mosievskiy, Nikita},
+  journal={arXiv preprint arXiv:2603.14911},
+  year={2026}
   url={https://huggingface.co/xamxte/cwe-classifier-roberta-base}
 }
 ```