tinylettuce-ettin-17m-en-bioasq

TinyLettuce (Ettin-17M): Efficient Hallucination Detection

<p align="center"> <img src="https://github.com/KRLabsOrg/LettuceDetect/blob/dev/assets/tinytinylettuce.png?raw=true" alt="TinyLettuce" width="400"/> </p>

Model Name: tinylettuce-ettin-17m-en-bioasq

Organization: KRLabsOrg

Github: https://github.com/KRLabsOrg/LettuceDetect

Ettin encoders: https://arxiv.org/pdf/2507.11412

Overview

TinyLettuce is a lightweight token‑classification model that flags unsupported spans in answers given context (span aggregation performed downstream). Built on the 17M Ettin encoder, it targets real‑time CPU inference and low‑cost domain fine‑tuning with synthetic data.

Trained on our synthetic dataset, this 17M variant achieves 90.87% F1 on the held‑out synthetic test set (beating large LLM judges, like GPT-OSS-120), proving the effectiveness of our domain‑specific hallucination data generation pipeline.

Model Details

• Architecture: Ettin encoder (17M) + token‑classification head
• Task: token classification (0 = supported, 1 = hallucinated)
• Input format: [CLS] context [SEP] question [SEP] answer [SEP], up to 4096 tokens
• Language: English; License: MIT

Training Data

• Synthetic generation (train): ~1,500 hallucinated samples (≈3,000 total with non‑hallucinated) from enelpol/rag-mini-bioasq; intensity 0.3.
• Synthetic generation (test): 300 hallucinated samples (≈600 total) held out.

Training Procedure

• Tokenizer: AutoTokenizer; DataCollatorForTokenClassification; label pad −100
• Max length: 4096 tokens; batch size: 8; epochs: 3
• Optimizer: AdamW (lr 1e‑5, weight_decay 0.01)
• Hardware: Single A100 80GB; CPU inference targeted

Results

Synthetic (domain‑specific):

Notes: “Synthetic” metrics reflect generated data; absolute scores depend on post‑processing thresholds and domain.

Usage

First install lettucedetect:

pip install lettucedetect

Then use it:

from lettucedetect.models.inference import HallucinationDetector

detector = HallucinationDetector(
    method="transformer",
    model_path="KRLabsOrg/tinylettuce-ettin-17m-en-v1",
)

spans = detector.predict(
    context=[
        "Ibuprofen is an NSAID that reduces inflammation and pain. The typical adult dose is 400-600mg every 6-8 hours, not exceeding 2400mg daily."
    ],
    question="What is the maximum daily dose of ibuprofen?",
    answer="The maximum daily dose of ibuprofen for adults is 3200mg.",
    output_format="spans",
)
print(spans)
# Output: [{"start": 51, "end": 57, "text": "3200mg"}]

Citing

If you use the model or the tool, please cite the following paper:

@misc{Kovacs:2025,
      title={LettuceDetect: A Hallucination Detection Framework for RAG Applications}, 
      author={Ádám Kovács and Gábor Recski},
      year={2025},
      eprint={2502.17125},
      archivePrefix={arXiv},
      primaryClass={cs.CL},
      url={https://arxiv.org/abs/2502.17125}, 
}