Instructions to use SkillJev/SkillJev-2B with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use SkillJev/SkillJev-2B with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="SkillJev/SkillJev-2B")# pip install -U transformers accelerate # Load model directly from transformers import AutoProcessor, AutoModelForSequenceClassification processor = AutoProcessor.from_pretrained("SkillJev/SkillJev-2B") model = AutoModelForSequenceClassification.from_pretrained("SkillJev/SkillJev-2B", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Credit SkillJev team and Team Lead
Browse files
README.md
CHANGED
|
@@ -10,36 +10,54 @@ datasets:
|
|
| 10 |
- ProtectSkills/MaliciousSkillBench
|
| 11 |
---
|
| 12 |
|
|
|
|
| 13 |
# SkillJev-Direct-2B
|
| 14 |
|
|
|
|
| 15 |
SkillJev-Direct-2B scores malicious AI-agent skills from `SKILL.md`, using [OpenJev 2B](https://huggingface.co/AlexWortega/openjev/tree/a20448012c213128955ca0c693e7c943865cab77/qwen3.5-2b-nli-v5), LoRA and a trained classification head. It returns a calibrated score without generating text.
|
| 16 |
|
| 17 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
|
| 19 |
**Research preview:** training and frozen evaluation are complete; weights are not yet published here.
|
| 20 |
|
|
|
|
| 21 |
## Results
|
| 22 |
|
|
|
|
| 23 |
[MaliciousSkillBench](https://huggingface.co/datasets/ProtectSkills/MaliciousSkillBench/tree/d4b42ce5766a6e0359c987cf59c1007cb3795a90), official **Source-Balanced Random** protocol: 6,812 train / 973 validation / 1,950 test.
|
| 24 |
|
|
|
|
| 25 |
| Method | Test recall | Benign false-positive rate |
|
| 26 |
|---|---:|---:|
|
| 27 |
| **SkillJev-Direct-2B** | **99.13%** | **5.57%** |
|
| 28 |
| Word-SVM | 99.40% | 40.98% |
|
| 29 |
| Untuned OpenJev-2B | 99.47% | 99.78% |
|
| 30 |
|
|
|
|
| 31 |
Thresholds were frozen on validation, targeting 99% recall. Direct has **159 fewer false positives and four additional misses** than Word-SVM. One seed; sources are shared across partitions.
|
| 32 |
|
|
|
|
| 33 |
## Model input
|
| 34 |
|
|
|
|
| 35 |
Premise: `SKILL.md`. Fixed hypothesis:
|
| 36 |
|
|
|
|
| 37 |
> This AI agent skill is malicious.
|
| 38 |
|
|
|
|
| 39 |
Run `direct-r002`: one epoch, 2.19M trainable parameters, 8,192-token limit. Long inputs retain the Markdown prefix and complete hypothesis. Calibration uses validation only.
|
| 40 |
|
|
|
|
| 41 |
## Scope and availability
|
| 42 |
|
|
|
|
| 43 |
Markdown-only detection does not certify a complete package as safe; performance on other distributions is unverified.
|
| 44 |
|
|
|
|
| 45 |
Adapter/head release terms are being reviewed. OpenJev is labelled MIT; third-party training texts retain their upstream terms. No skill payloads or TrendAI data are hosted here.
|
|
|
|
|
|
| 10 |
- ProtectSkills/MaliciousSkillBench
|
| 11 |
---
|
| 12 |
|
| 13 |
+
|
| 14 |
# SkillJev-Direct-2B
|
| 15 |
|
| 16 |
+
|
| 17 |
SkillJev-Direct-2B scores malicious AI-agent skills from `SKILL.md`, using [OpenJev 2B](https://huggingface.co/AlexWortega/openjev/tree/a20448012c213128955ca0c693e7c943865cab77/qwen3.5-2b-nli-v5), LoRA and a trained classification head. It returns a calibrated score without generating text.
|
| 18 |
|
| 19 |
+
|
| 20 |
+
**Team Lead:** Jerrold Huang
|
| 21 |
+
**Team members:** Junwei Quan, HaoXuan Huang, Shiyi Yang
|
| 22 |
+
University of Toronto MScAC AI team, in collaboration with TrendAI.
|
| 23 |
+
|
| 24 |
|
| 25 |
**Research preview:** training and frozen evaluation are complete; weights are not yet published here.
|
| 26 |
|
| 27 |
+
|
| 28 |
## Results
|
| 29 |
|
| 30 |
+
|
| 31 |
[MaliciousSkillBench](https://huggingface.co/datasets/ProtectSkills/MaliciousSkillBench/tree/d4b42ce5766a6e0359c987cf59c1007cb3795a90), official **Source-Balanced Random** protocol: 6,812 train / 973 validation / 1,950 test.
|
| 32 |
|
| 33 |
+
|
| 34 |
| Method | Test recall | Benign false-positive rate |
|
| 35 |
|---|---:|---:|
|
| 36 |
| **SkillJev-Direct-2B** | **99.13%** | **5.57%** |
|
| 37 |
| Word-SVM | 99.40% | 40.98% |
|
| 38 |
| Untuned OpenJev-2B | 99.47% | 99.78% |
|
| 39 |
|
| 40 |
+
|
| 41 |
Thresholds were frozen on validation, targeting 99% recall. Direct has **159 fewer false positives and four additional misses** than Word-SVM. One seed; sources are shared across partitions.
|
| 42 |
|
| 43 |
+
|
| 44 |
## Model input
|
| 45 |
|
| 46 |
+
|
| 47 |
Premise: `SKILL.md`. Fixed hypothesis:
|
| 48 |
|
| 49 |
+
|
| 50 |
> This AI agent skill is malicious.
|
| 51 |
|
| 52 |
+
|
| 53 |
Run `direct-r002`: one epoch, 2.19M trainable parameters, 8,192-token limit. Long inputs retain the Markdown prefix and complete hypothesis. Calibration uses validation only.
|
| 54 |
|
| 55 |
+
|
| 56 |
## Scope and availability
|
| 57 |
|
| 58 |
+
|
| 59 |
Markdown-only detection does not certify a complete package as safe; performance on other distributions is unverified.
|
| 60 |
|
| 61 |
+
|
| 62 |
Adapter/head release terms are being reviewed. OpenJev is labelled MIT; third-party training texts retain their upstream terms. No skill payloads or TrendAI data are hosted here.
|
| 63 |
+
|