DepressLLM: Interpretable domain-adapted language model for depression detection from real-world narratives

## Контекст Современные прогрессы в области бо LARGE LANGUAGE MODELS (LLMs) опенным образом расширили возможности применения искусственного интеллекта в различных сферах. Однако в сфере психиатрических заболеваний, таких как депрессия, существуют значительные сложности. Основной проблемой является нехватка больших, высококачественных и тщательно аннотированных данных, необходимых для обучения моделей. Это приводит к ограниченной точности в предсказании депрессии и затруднению ранней диагностики. Наличие таких данных является ключевым фактором для повышения точности моделей и их применения в психиатрической практике. ## Метод DepressLLM является первым интерпретируемым доменно-адаптированным языковым моделем, разработанным для детекции депрессии из реального жизненного текста. Он обучен на разработанной коресе DepressNarrativeCorpus, содержащей 3 699 автобиографических рассказов, которые отражают различные аспекты счастья и тревоги. Для обеспечения интерпретируемости решения использован модуль Score-guided Token Probability Summation (SToPS), который обеспечивает дополнительную видимость в процессе предсказания. Модель была тщательно тестирована на обученных данных, а также подвергнута валидации на независимых датасетах, включая журналы дневников и клинические записи. ## Результаты В ходе экспериментов DepressLLM показал высокую точность в предсказании депрессии, с AUC-значением 0.789. Благодаря SToPS-модулю, модель достигла улучшения в классификации и повысила уровень доверия к своим предсказаниям. На высокодоверительных мисклассификациях, проверенных вручную психиатрами, выявлены ключевые ограничения модели и данных, которые могут быть использованы для дальнейшего повышения точности. ## Значимость Разработанная модель может быть применена в различных областях, включая клиническую практику, мобильные приложения и анализ журналов эмоциональных состояний. Она предоставляет интерпретируемые результаты, обеспечивая доверие к предсказаниям и улучшая надежность в диагностике депрессии. Это демонстрирует потенциал интерпретируемых моделей AI в ранней диагностике и улучшении здоровья пациентов. ## Выводы DepressLLM представляет собой прорыв в области психиатрического AI за счет создания интерпретируемой модели, которая улучшает точность диагностики депрессии. Будущие исследования будут фокусироваться на повышении точности и расширении области применения модели, включая психотерапевтические приложения и анализ других психических состояний.