Upload README.md with huggingface_hub

README.md

@@ -0,0 +1,72 @@
+---
+license: mit
+language:
+- en
+tags:
+- cybersecurity
+- vulnerability
+- mitre-attck
+- text-classification
+- fine-tuned
+base_model: ehsanaghaei/SecureBERT
+---
+
+# SecureBERT — MITRE ATT&CK Classifier
+
+Part of the **CVE-LMTune** model suite — language models fine-tuned for multi-taxonomy vulnerability classification.
+
+## Paper
+
+> Franco Terranova, Sana Rekbi, Abdelkader Lahmadi, Isabelle Chrisment. *Multi-Taxonomy Vulnerability Classification with Hierarchically Finetuned Language Models.* DIMVA '26.
+
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+[![Zenodo](https://img.shields.io/badge/Zenodo-Data%20%26%20Models-lightblue)](https://zenodo.org/records/17368476)
+
+## Task
+
+**MITRE ATT&CK technique classification from CVE descriptions**
+
+## Performance
+
+See paper for details
+
+## Model Structure
+
+flat — standard `AutoModelForSequenceClassification`
+
+## Quick Start
+
+```python
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+import torch
+
+tokenizer = AutoTokenizer.from_pretrained("Sana9/securebert-mitre-attack")
+model = AutoModelForSequenceClassification.from_pretrained("Sana9/securebert-mitre-attack")
+model.eval()
+
+text = "Buffer overflow vulnerability in OpenSSL allows remote attackers to execute arbitrary code."
+inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=512)
+
+with torch.no_grad():
+    logits = model(**inputs).logits
+    probs = torch.sigmoid(logits)   # multi-label → sigmoid
+```
+
+> **Note for hierarchical models:** This repo contains multiple sub-folders (master + slave models).
+> Load each sub-folder separately using `from_pretrained("Sana9/securebert-mitre-attack/master")` etc.
+
+## Citation
+
+```bibtex
+@inproceedings{terranova2026cvelmtune,
+  title     = {Multi-Taxonomy Vulnerability Classification with Hierarchically Finetuned Language Models},
+  author    = {Terranova, Franco and Rekbi, Sana and Lahmadi, Abdelkader and Chrisment, Isabelle},
+  booktitle = {Proceedings of DIMVA '26},
+  year      = {2026}
+}
+```
+
+## Related Resources
+
+- 🤗 [Full model suite](https://huggingface.co/Sana9)
+- 📦 [Zenodo data & logs](https://zenodo.org/records/17368476)