Update README.md

README.md

@@ -9,14 +9,14 @@ metrics:
 - f1
 - accuracy
 model-index:
-- name: anonymouspd/SecureBERT-DNRTI
+- name: Cyber-ThreaD/SecureBERT-DNRTI
   results: []
 ---
 
 <!-- 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/SecureBERT-DNRTI
+# Cyber-ThreaD/SecureBERT-DNRTI
 
 This model is a fine-tuned version of [ehsanaghaei/SecureBERT](https://huggingface.co/ehsanaghaei/SecureBERT) on the [DNRTI](https://github.com/SCreaMxp/DNRTI-A-Large-scale-Dataset-for-Named-Entity-Recognition-in-Threat-Intelligence) dataset.
 It achieves the following results on the evaluation set:
@@ -82,3 +82,20 @@ 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}
+}
+
+```