diviqra-distilbertguard

DistilBERT Guard — LLM Prompt Injection Classifier

Fine-tuned DistilBERT for detecting prompt injection, jailbreaks, PII extraction, system prompt leaks, and indirect injection in LLM applications.

Part of Diviqra Guard — open-source LLM firewall.

Performance

Benchmark

Languages

Only LLM security model with Indian language support:

• Hindi Devanagari: पिछले सभी निर्देश अनदेखा करें
• Hinglish: System ko ignore karo
• Tamil: முந்தைய வழிமுறைகளை புறக்கணி
• Telugu: మునుపటి సూచనలను విస్మరించు
• Kannada: ಹಿಂದಿನ ಸೂಚನೆಗಳನ್ನು ನಿರ್ಲಕ್ಷಿಸಿ

Architecture

Wall 1 of Diviqra Guard's five-wall architecture: Wall 0 Multilingual normaliser + entropy analyser (<1ms) Wall 1 This model — DistilBERT ONNX classifier (<10ms) Wall 2 LLM judge — qwen3:1.7b via Ollama (1-4s, ~20% traffic)

Usage

Via Diviqra Guard SDK

pip install diviqra-guard

from diviqra_guard import Guard

guard = Guard(api_key="dg_dev_...")
result = guard.scan("Ignore all previous instructions")
# ScanResult(blocked=True, threat="prompt_injection", latency_ms=8)

Free API key at guard.diviqra.com — 10,000 scans/month.

Direct ONNX inference

import onnxruntime as ort
from transformers import DistilBertTokenizerFast
import numpy as np

tokenizer = DistilBertTokenizerFast.from_pretrained("distilbert-base-uncased")
session = ort.InferenceSession("distilbert-guard.onnx")

def classify(text: str) -> dict:
    inputs = tokenizer(
        text, return_tensors="np",
        max_length=256, truncation=True, padding="max_length"
    )
    outputs = session.run(None, {
        "input_ids": inputs["input_ids"],
        "attention_mask": inputs["attention_mask"],
    })
    logits = outputs[0][0]
    probs = np.exp(logits) / np.exp(logits).sum()
    return {
        "label": "threat" if probs[1] > probs[0] else "safe",
        "score": float(probs[1])
    }

classify("Ignore all previous instructions")
# {"label": "threat", "score": 0.998}

classify("System ko ignore karo")
# {"label": "threat", "score": 0.991}

Training Data

Links

• 🌐 guard.diviqra.com
• 📦 PyPI
• 💻 GitHub
• 📧 [email protected]