cafebert-ViHSD

cafebert-ViHSD

This model is uitnlp/CafeBERT fine-tuned for ViHSD hate speech detection on ViHSD.

Evaluation protocol

• Dataset size: 33,400 examples.
• Original published fixed splits: 24,048 train / 2,672 development / 6,680 test.
• No rows were moved between the published splits.
• Fine-tuning seeds: [22, 42, 202].
• Training: 3 epoch(s), AdamW.
• Learning rate: 2e-05.
• Weight decay: 0.01.
• Warmup ratio: 0.1.
• Training batch size: 8.
• Maximum sequence length: 256.
• Input mode: raw Vietnamese social-media text.
• The published checkpoint is seed 22.

Results

Metrics are reported as mean ± sample standard deviation over the available completed seeds.

Per-seed results

Label mapping

{
  "0": "CLEAN",
  "1": "OFFENSIVE",
  "2": "HATE"
}

Usage

import torch
from transformers import AutoModelForSequenceClassification, AutoTokenizer

model_id = "BaoNhan/cafebert-ViHSD"
tokenizer = AutoTokenizer.from_pretrained(model_id, use_fast=False)
model = AutoModelForSequenceClassification.from_pretrained(model_id)

text = "Đây là nội dung tiếng Việt cần phân loại."

inputs = tokenizer(
    text,
    return_tensors="pt",
    truncation=True,
    max_length=256,
)

with torch.no_grad():
    probabilities = model(**inputs).logits.softmax(dim=-1)[0]

predicted_id = int(probabilities.argmax())
print(model.config.id2label[predicted_id], probabilities.tolist())

Limitations

ViHSD is class-imbalanced and reflects Vietnamese social-media language from a particular collection period. Performance may not transfer directly to new platforms, dialects, code-switching patterns, irony, or emerging slang. Predictions should not be the sole basis for moderation or punitive decisions.

Dataset citation

@InProceedings{10.1007/978-3-030-79457-6_35,
  author={Luu, Son T. and Nguyen, Kiet Van and Nguyen, Ngan Luu-Thuy},
  title={A Large-Scale Dataset for Hate Speech Detection on Vietnamese Social Media Texts},
  booktitle={Advances and Trends in Artificial Intelligence. Artificial Intelligence Practices},
  year={2021},
  publisher={Springer International Publishing},
  pages={415--426},
  doi={10.1007/978-3-030-79457-6_35}
}