Downloads · 30 days
0
oikyoni/drug-screening-llm
drug-screening-llm is a text generation model from oikyoni. Use it when you need the model to write or continue text. The card lists the license as llama3.2.
A QLoRA fine-tune of Meta's Llama 3.2 3B Instruct, specialized for workplace drug screening compliance. Built for HR professionals, occupational health nurses, Medical Review Officers (MROs), and employers navigating…
Downloads · 30 days
0
Access
Public
Updated Sep 12, 2026
Repo size
326 MB
Likes
0
Public
Click a slice to open those files.
.safetensors97.3 MB · 85%
From the Hugging Face model README
A QLoRA fine-tune of Meta's Llama 3.2 3B Instruct, specialized for workplace drug screening compliance. Built for HR professionals, occupational health nurses, Medical Review Officers (MROs), and employers navigating SAMHSA, DOT, FDA, and state-level workplace drug testing regulation.
This repository hosts the V2 adapter (citation- and refusal-augmented). All numbers below are V2 and match the accompanying paper; see Version history for the V1/V2 distinction.
Workplace drug screening operates under a complex, frequently updated regulatory framework spanning SAMHSA mandatory guidelines, DOT 49 CFR Part 40, FDA device regulations, and 50 varying state marijuana laws. Misinterpretation of cutoff thresholds, panel compositions, or jurisdiction-specific rules can result in legal liability, wrongful termination claims, or compliance violations.
Intended users: HR professionals managing workplace drug-free policies; occupational health nurses interpreting screening results; Medical Review Officers; employers in DOT-regulated industries; compliance officers.
Intended use cases: answering regulatory compliance questions (SAMHSA, DOT, FDA); cutoff thresholds by drug class and specimen type; state-specific marijuana testing laws; the two-step screening and confirmation process; MRO and split-specimen procedures.
Out of scope:
Fine-tuned on the Drug Screening Compliance Benchmark Dataset - 710 expert-crafted Q&A pairs grounded exclusively in publicly available regulatory and peer-reviewed sources:
Splits: 639 training / 71 validation. The validation split contains 67 on-topic regulatory questions and 4 off-topic refusal examples. Of the 710 total, 40 are off-topic refusal pairs added during V2 augmentation.
All training data is traceable to primary public sources. No proprietary or confidential data was used. The dataset is released under CC BY-NC 4.0.
| Parameter | Value |
|---|---|
| Base model | meta-llama/Llama-3.2-3B-Instruct |
| Method | QLoRA (4-bit NF4, double quantization, bfloat16 compute) |
| LoRA rank / alpha / dropout | 16 / 32 / 0.05 |
| Target modules | all 7 projections (q, k, v, o, gate, up, down) across 28 layers |
| Trainable parameters | ~24M (97.3 MB adapter) |
| Epochs | 3 (best checkpoint by validation loss) |
| Learning rate | 2e-4 |
| Batch size | 2 per device x 4 gradient accumulation (effective 8) |
| Warmup ratio | 0.05 |
| Max sequence length | 1024 |
| Optimizer / schedule | adamw_torch, linear decay |
| Seed | 42 (framework default; not explicitly varied) |
| Hardware | Kaggle GPU T4 x2 (2 x 16 GB VRAM); model placed across both GPUs via device_map="auto" |
| Framework | HuggingFace PEFT + Transformers + TRL |
| Epoch | Train Loss | Val Loss |
|---|---|---|
| 1 | 2.7949 | 0.6788 |
| 2 | 2.2078 | 0.6422 (best checkpoint) |
| 3 | 1.6542 | 0.6632 |
Evaluated with an LLM-as-judge protocol using openai/gpt-oss-120b (served via Groq)
as an independent judge - a different model family from every candidate. Each dimension
is scored 1-5. Correctness, citations, and on-topic refusal are scored over the 67
on-topic validation questions; off-topic refusal over 10 held-out off-topic prompts.
Candidate generation used temperature 0.3 / top-p 0.9; the judge ran at temperature 0,
one judging pass per item.
| Model | Overall | Correctness | Citations | On-topic refusal | Off-topic refusal |
|---|---|---|---|---|---|
| Frontier (Llama 3.3 70B) | 3.683 | 3.955 | 3.776 | 5.0 | 2.0 |
| DrugScreenLLM V2 (this model) | 3.600 | 2.761 | 1.940 | 5.0 | 4.7 |
| Base Llama 3.2 3B Instruct | 2.726 | 2.284 | 1.821 | 5.0 | 1.8 |
| Qwen 2.5 3B | 2.610 | 2.597 | 1.642 | 5.0 | 1.2 |
The overall score approaches that of a 70B reference model at roughly 1/23 the parameter count, driven substantially by off-topic refusal calibration (2.4 -> 4.7 from V1 to V2). Correctness remains below the frontier model and declined slightly from V1 (2.940 -> 2.761) - a specialization trade-off reported in full in the paper rather than omitted here.
Drug screening decisions with legal or employment consequences must be reviewed by a qualified Medical Review Officer (MRO) and legal counsel. This model is an informational tool only.
This adapter is derived from Meta's Llama 3.2 3B Instruct and is therefore governed by the Llama 3.2 Community License, not an OSI-approved open source license. The accompanying dataset is released under CC BY-NC 4.0 (non-commercial). Review both before use.
@misc{drugscreenllm2026,
title = {DrugScreenLLM: A Domain-Specific Language Model for Workplace Drug Screening Compliance},
author = {Bhattacharjya, Sairath},
year = {2026},
publisher = {HuggingFace},
url = {https://huggingface.co/oikyoni/drug-screening-llm}
}
Developed by Sairath Bhattacharjya, who brings professional experience in federally regulated workplace drug testing compliance.