Text Classification
Transformers
ONNX
Safetensors
multilingual
xlm-roberta
privacy
pii-detection
text-embeddings-inference
Instructions to use Roblox/roblox-pii-classifier-v2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Roblox/roblox-pii-classifier-v2 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="Roblox/roblox-pii-classifier-v2")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("Roblox/roblox-pii-classifier-v2") model = AutoModelForSequenceClassification.from_pretrained("Roblox/roblox-pii-classifier-v2", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -11,9 +11,7 @@ base_model: FacebookAI/xlm-roberta-large
|
|
| 11 |
datasets:
|
| 12 |
- Roblox/roblox-pii-safety-for-chat-benchmark
|
| 13 |
- ai4privacy/pii-masking-openpii-1.5m
|
| 14 |
-
- ai4privacy/pii-masking-health-phi-400k
|
| 15 |
- nvidia/Nemotron-PII
|
| 16 |
-
- PKU-Alignment/BeaverTails
|
| 17 |
---
|
| 18 |
|
| 19 |
# Roblox PII Classifier v2.0 — Model Card
|
|
@@ -29,7 +27,7 @@ Unlike its predecessor and other NER models, **v2 evaluates text within its surr
|
|
| 29 |
* **RAG-Augmented LLM Labeling:** Primarily trained on data labeled by LLMs leveraging Retrieval-Augmented Generation (RAG) on a curated corpus of labeled samples, outperforming human expert baselines in both precision and recall on complex PII cases.
|
| 30 |
* **Massive Multilingual Expansion:** Broadened synthetic data generation and fine-tuning capabilities to expand language coverage from **17 to 189 languages**.
|
| 31 |
* **Automated Red-Teaming & Clustering:** Applied automated adversarial red-teaming and cluster visualization to systematically surface, isolate, and patch model vulnerabilities.
|
| 32 |
-
* **Significant Performance Leap:** Overall F1 score increased from **63.41% to 90.52%**
|
| 33 |
|
| 34 |
## Model Architecture & Classification Scope
|
| 35 |
|
|
@@ -66,13 +64,12 @@ emitted in this fixed index order:
|
|
| 66 |
|
| 67 |
The table below illustrates best f1 performance across internal and open-source evaluation benchmarks (full conversational data used when possible):
|
| 68 |
|
| 69 |
-
| Dataset / Benchmark | Roblox PII v2 | *Roblox PII v1 | *OpenAI Privacy Filter | *GLiNER2 | Qwen3Guard | LlamaGuard v3 1B | LlamaGuard v3 8B | LlamaGuard v4 12B | NemoGuard 8B | *Piiranha NER | Shieldstral
|
| 70 |
| :--- | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
|
| 71 |
| **roblox-pii-safety-for-chat** | **88.78%** | 64.74% | 58.10% | 54.42% | 66.15% | 27.80% | 56.06% | 54.30% | 56.35% | 58.34% | 57.66 |
|
| 72 |
| **Roblox Internal Evaluation** | **90.52%** | 63.41% | 20.98% | 28.21% | 15.62% | 8.59% | 12.90% | 21.92% | 14.77% | 20.98% | 31.87 |
|
| 73 |
| **Nemotron-PII** | **99.22%** | 70.07% | 65.22% | 62.92% | 37.46% | 53.32% | 46.07% | 35.05% | 56.16% | 68.52% | 56.69 |
|
| 74 |
| **PII Masking OpenPII 1.5M** | **99.76%** | 86.79% | 84.23% | 83.07% | 61.94% | 65.10% | 64.23% | 56.16% | 54.01% | 86.35% | 85.33 |
|
| 75 |
-
| **PII Masking 400k** | **97.64%** | 97.44% | 72.66% | 72.28% | 48.83% | 52.41% | 49.64% | 42.48% | 49.82% | 97.09% | 67.13 |
|
| 76 |
|
| 77 |
\* These models all were supplied with only the target-user text as other speaker texts are meant to be supplemental contextual information. roblox-pii-classifier-v2 performs best when supplied with conversational context, however it still outperforms roblox-pii-classifier when supplied only with target-user text.
|
| 78 |
|
|
|
|
| 11 |
datasets:
|
| 12 |
- Roblox/roblox-pii-safety-for-chat-benchmark
|
| 13 |
- ai4privacy/pii-masking-openpii-1.5m
|
|
|
|
| 14 |
- nvidia/Nemotron-PII
|
|
|
|
| 15 |
---
|
| 16 |
|
| 17 |
# Roblox PII Classifier v2.0 — Model Card
|
|
|
|
| 27 |
* **RAG-Augmented LLM Labeling:** Primarily trained on data labeled by LLMs leveraging Retrieval-Augmented Generation (RAG) on a curated corpus of labeled samples, outperforming human expert baselines in both precision and recall on complex PII cases.
|
| 28 |
* **Massive Multilingual Expansion:** Broadened synthetic data generation and fine-tuning capabilities to expand language coverage from **17 to 189 languages**.
|
| 29 |
* **Automated Red-Teaming & Clustering:** Applied automated adversarial red-teaming and cluster visualization to systematically surface, isolate, and patch model vulnerabilities.
|
| 30 |
+
* **Significant Performance Leap:** Overall F1 score increased from **63.41% to 90.52%** on our refreshed internal evaluation set, with a f1 score increase from **64.74% to 88.78%** on the recently released roblox-pii-safety-for-chat-benchmark.
|
| 31 |
|
| 32 |
## Model Architecture & Classification Scope
|
| 33 |
|
|
|
|
| 64 |
|
| 65 |
The table below illustrates best f1 performance across internal and open-source evaluation benchmarks (full conversational data used when possible):
|
| 66 |
|
| 67 |
+
| Dataset / Benchmark | Roblox PII v2 | *Roblox PII v1 | *OpenAI Privacy Filter | *GLiNER2 | Qwen3Guard Gen 8B | LlamaGuard v3 1B | LlamaGuard v3 8B | LlamaGuard v4 12B | NemoGuard 8B | *Piiranha NER | Shieldstral
|
| 68 |
| :--- | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
|
| 69 |
| **roblox-pii-safety-for-chat** | **88.78%** | 64.74% | 58.10% | 54.42% | 66.15% | 27.80% | 56.06% | 54.30% | 56.35% | 58.34% | 57.66 |
|
| 70 |
| **Roblox Internal Evaluation** | **90.52%** | 63.41% | 20.98% | 28.21% | 15.62% | 8.59% | 12.90% | 21.92% | 14.77% | 20.98% | 31.87 |
|
| 71 |
| **Nemotron-PII** | **99.22%** | 70.07% | 65.22% | 62.92% | 37.46% | 53.32% | 46.07% | 35.05% | 56.16% | 68.52% | 56.69 |
|
| 72 |
| **PII Masking OpenPII 1.5M** | **99.76%** | 86.79% | 84.23% | 83.07% | 61.94% | 65.10% | 64.23% | 56.16% | 54.01% | 86.35% | 85.33 |
|
|
|
|
| 73 |
|
| 74 |
\* These models all were supplied with only the target-user text as other speaker texts are meant to be supplemental contextual information. roblox-pii-classifier-v2 performs best when supplied with conversational context, however it still outperforms roblox-pii-classifier when supplied only with target-user text.
|
| 75 |
|