Usage

To use the model for detecting machine-generated text:

python

from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch

tokenizer = AutoTokenizer.from_pretrained("GeorgeDrayson/modernbert-raid-mage")
model = AutoModelForSequenceClassification.from_pretrained("GeorgeDrayson/modernbert-raid-mage")

text = "Your input text here."
inputs = tokenizer(text, return_tensors="pt")
outputs = model(**inputs)
probabilities = torch.nn.functional.softmax(outputs.logits, dim=-1)
print(f"Probability of machine-generated text: {probabilities[0][1].item():.4f}")

Citation

If you use this model or find the research helpful, please cite:

bibtex

@article{drayson2025machine,
  title={Machine-generated text detection prevents language model collapse},
  author={Drayson, George and Yilmaz, Emine and Lampos, Vasileios},
  journal={arXiv preprint arXiv:2502.15654},
  year={2025}
}

GeorgeDrayson/modernbert-ai-detection-raid-mage

Machine-generated text detection prevents language model collapse

Usage

Citation

GeorgeDrayson/
modernbert-ai-detection-raid-mage