ClinicalAbstain-Text2SQL

Репозиторий артефактов проекта BN5212 ClinicalAbstain-Text2SQL-LLM: LoRA-адаптеры для Text-to-SQL, обученные отвечать SQL или воздерживаться от ответа. Адаптеры сделаны для Llama-3-8B-Instruct, Qwen3-8B, Qwen3-4B-Instruct-2507 и Mistral-7B-Instruct-v0.3 в двух вариантах — abstention-aware и SQL-only.

🔤 Ключевые особенности

• 🔤 LoRA для четырёх базовых моделей: Llama-3-8B-Instruct, Qwen3-8B, Qwen3-4B-Instruct-2507 и Mistral-7B-Instruct-v0.3.
• ➕ Два варианта обучения: abstention-aware на answerable + unanswerable данных и SQL-only только на answerable данных.
• 🟰 В каталогах лежат adapter_model.safetensors, adapter_config.json, файлы токенизатора, train.log и eval_results.json.
• 🔜 Обучение только на синтетическом OMOP Text-to-SQL бенчмарке (Legrand et al.); данные MIMIC-IV и MIMIC-IV-Note не включены.
• ➖ Это только адаптеры: их нужно загружать поверх соответствующей базовой модели с отдельной лицензией.

➕ Пример

huggingface-cli download Nitin2004/clinicalabstainsql-artifacts --include "repro/runs/lora_seed42/*" --local-dir .

Подробнее

Это только LoRA-адаптеры: их нужно загружать поверх соответствующей базовой модели (Meta Llama-3, Qwen3, Mistral) с отдельными лицензиями.
Обучение — только на синтетическом OMOP Text-to-SQL бенчмарке; данные MIMIC-IV / MIMIC-IV-Note и производные файлы не включены.
В репозитории есть чекпоинты (например, 150/300/309 для lora_seed42) и eval_results.json, но готовых цифр бенчмарков в карточке нет.
Для PostgreSQL-данных синтетического бенчмарка приложен pgdata_synthetic.tar.gz.