PEFT
Safetensors
English
medical
clinical-nlp
lora
hallucination-detection
adversarial-training
agentic-ai
Instructions to use aydin237/medbrain-squad-auditor with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use aydin237/medbrain-squad-auditor with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("google/medgemma-1.5-4b-it") model = PeftModel.from_pretrained(base_model, "aydin237/medbrain-squad-auditor") - Notebooks
- Google Colab
- Kaggle
MedBrainSquad โ Auditor Agent Adapter
LoRA fine-tuned adapter for MedGemma 1.5 4B, trained via adversarial Supervised Fine-Tuning (SFT) to detect hallucinations, flag clinical inconsistencies, and validate SOAP JSON output from the Scribe agent.
Project
Part of the MedBrainSquad Clinical Safety Net โ a three-tier adversarial multi-agent pipeline for safe, hallucination-resistant clinical documentation.
๐ Full project: github.com/aydiny/medbrain-squad ๐ Scribe Adapter: aydin237/medbrain-squad-scribe
Training Details
- Base model: google/medgemma-1.5-4b-it
- Method: LoRA (Rank 16, Alpha 32)
- Training approach: Adversarial SFT on synthetic hallucinated/corrupted SOAP examples
- Dataset: 500-row adversarial dataset with deliberate hallucinations and clinical errors
- Hardware: Dual NVIDIA T4 (Kaggle)
- Framework: HuggingFace PEFT + TRL
Intended Use
Research and development of adversarial safety layers in clinical NLP pipelines. Demonstrates hallucination detection in regulated healthcare environments (NHS/EU GDPR).
โ ๏ธ Limitations & Disclaimer
- Research prototype โ not validated for clinical deployment
- Base model (MedGemma 1.5 4B) subject to Google HAID Terms
- Auditor itself may miss hallucinations in novel edge cases
- Trained on synthetic data only
- Downloads last month
- 4
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐ Ask for provider support
Model tree for aydin237/medbrain-squad-auditor
Base model
google/medgemma-1.5-4b-it