Классификатор диагнозов алопеции (ruBERT)
Модель классифицирует текст медицинского приёма трихолога на 4 класса:
| Класс | МКБ | Описание |
|---|---|---|
| АГА | L64.x | Андрогенетическая алопеция |
| Телогеновая | L65.x | Диффузное выпадение |
| Себорея | L21.x | Себорейный дерматит |
| Гнёздная | L63.x | Очаговая алопеция |
Архитектура
- Текстовая ветвь: DeepPavlov/rubert-base-cased → [CLS] → Linear(768→256) → GELU
- Структурная ветвь: числовые признаки (возраст, пол, keyword-флаги) → Linear(N→64) → GELU
- Классификатор: concat(320) → Linear(320→4)
Метрики (test set)
- F1 macro: ~0.82
- Val F1 лучшая эпоха: ~0.83
Датасет
Трихологическая клиника УГМК, ~3500 первичных приёмов, 2018–2026.
- Downloads last month
- 4