Update README.md

README.md

@@ -15,7 +15,7 @@ model-index:
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
 
-# anonymouspd/CyBERT-CyNER
+# Cyber-ThreaD/CyBERT-CyNER
 
 This model is a fine-tuned version of [SynamicTechnologies/CYBERT](https://huggingface.co/SynamicTechnologies/CYBERT) on the [CyNER](https://github.com/aiforsec/CyNER) dataset.
 It achieves the following results on the evaluation set:
@@ -77,3 +77,19 @@ The following hyperparameters were used during training:
 - Pytorch 2.1.0+cu118
 - Datasets 2.14.6
 - Tokenizers 0.14.1
+
+### Citing & Authors
+
+If you use the model kindly cite the following work
+
+```
+@inproceedings{deka2024attacker,
+  title={AttackER: Towards Enhancing Cyber-Attack Attribution with a Named Entity Recognition Dataset},
+  author={Deka, Pritam and Rajapaksha, Sampath and Rani, Ruby and Almutairi, Amirah and Karafili, Erisa},
+  booktitle={International Conference on Web Information Systems Engineering},
+  pages={255--270},
+  year={2024},
+  organization={Springer}
+}
+
+```