CoolFace
Modelpublic

r2911/lm_p1

sourceHugging Facemitupdated 10mo agoView on Hugging Face
0likes9downloads
Model Card

๐Ÿง  AI Text Detector โ€“ DeBERTa v3 Large (Fine-tuned on Human vs AI Text)

This model is fine-tuned on a labeled dataset for AI-generated vs. Human-written text detection.


โš™๏ธ Model Details

  • โ€”Base Model: microsoft/deberta-v3-large
  • โ€”Fine-tuned by: @abhinav
  • โ€”Epochs: 4
  • โ€”Learning Rate: 2e-05
  • โ€”Batch Size: 8
  • โ€”GPU: 80 GB A100
  • โ€”Optimizer: AdamW (Fused)
  • โ€”Scheduler: Cosine
  • โ€”Mixed Precision: FP16
  • โ€”Gradient Checkpointing: Enabled

๐Ÿ“Š Evaluation Results (Test Set)

MetricScore
Accuracy0.993
Human (0) โ€“ Precision1.000
Human (0) โ€“ Recall0.986
AI (1) โ€“ Precision0.988
AI (1) โ€“ Recall1.000

๐Ÿงฎ Confusion Matrix

[image]


๐Ÿš€ Example Inference

python
from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch

tokenizer = AutoTokenizer.from_pretrained("abhi099k/ai-text-detector-deberta-v3-large-h1")
model = AutoModelForSequenceClassification.from_pretrained("abhi099k/ai-text-detector-deberta-v3-large-h1")

text = "This text was likely written by an AI model."
inputs = tokenizer(text, return_tensors="pt")
outputs = model(**inputs)
probs = torch.nn.functional.softmax(outputs.logits, dim=1)
print(probs)