Fill-Mask
Transformers
PyTorch
Safetensors
English
bert
exbert
security
cybersecurity
cyber security
threat hunting
threat intelligence
Instructions to use jackaduma/SecBERT with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use jackaduma/SecBERT with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="jackaduma/SecBERT")# Load model directly from transformers import AutoTokenizer, AutoModelForMaskedLM tokenizer = AutoTokenizer.from_pretrained("jackaduma/SecBERT") model = AutoModelForMaskedLM.from_pretrained("jackaduma/SecBERT", device_map="auto") - Inference
- Notebooks
- Google Colab
- Kaggle
| language: en | |
| thumbnail: https://github.com/jackaduma | |
| tags: | |
| - exbert | |
| - security | |
| - cybersecurity | |
| - cyber security | |
| - threat hunting | |
| - threat intelligence | |
| license: apache-2.0 | |
| datasets: | |
| - APTnotes | |
| - Stucco-Data | |
| - CASIE | |
| # SecBERT | |
| This is the pretrained model presented in [SecBERT: A Pretrained Language Model for Cyber Security Text](https://github.com/jackaduma/SecBERT/), which is a BERT model trained on cyber security text. | |
| The training corpus was papers taken from | |
| * [APTnotes](https://github.com/kbandla/APTnotes) | |
| * [Stucco-Data: Cyber security data sources](https://stucco.github.io/data/) | |
| * [CASIE: Extracting Cybersecurity Event Information from Text](https://ebiquity.umbc.edu/_file_directory_/papers/943.pdf) | |
| * [SemEval-2018 Task 8: Semantic Extraction from CybersecUrity REports using Natural Language Processing (SecureNLP)](https://competitions.codalab.org/competitions/17262). | |
| SecBERT has its own wordpiece vocabulary (secvocab) that's built to best match the training corpus. | |
| We trained [SecBERT](https://huggingface.co/jackaduma/SecBERT) and [SecRoBERTa](https://huggingface.co/jackaduma/SecRoBERTa) versions. | |
| Available models include: | |
| * [`SecBERT`](https://huggingface.co/jackaduma/SecBERT) | |
| * [`SecRoBERTa`](https://huggingface.co/jackaduma/SecRoBERTa) | |
| --- | |
| ## **Fill Mask** | |
| We proposed to build language model which work on cyber security text, as result, it can improve downstream tasks (NER, Text Classification, Semantic Understand, Q&A) in Cyber Security Domain. | |
| First, as below shows Fill-Mask pipeline in [Google Bert](), [AllenAI SciBert](https://github.com/allenai/scibert) and our [SecBERT](https://github.com/jackaduma/SecBERT) . | |
| <!-- <img src="./fill-mask-result.png" width="150%" height="150%"> --> | |
|  | |
| --- | |
| The original repo can be found [here](https://github.com/jackaduma/SecBERT). | |