Improving ARDS Diagnosis Through Context-Aware Concept Bottleneck Models

## Контекст Систематическое исследование заболеваний через автоматизированные методы является ключевым подходом для понимания их характеристик и развития эффективных методов лечения. Одним из сложных, но важных задачей является диагностика Акшенаиаса Ресшиора Дисторсион Синдрома (ARDS) — серьезного, тяжелого синдрома, характеризующегося острым дефицитом кислорода и легких. Несмотря на развитие медицинских технологий, диагностика ARDS до сих пор остается сложной, требующей внимания и профессионального опыта. Одним из основных проблемах в этой области является ограниченная интерпретируемость алгоритмов, которая свидетельствует о необходимости внести дополнительные улучшения. Одним из подходов, который набирает популярность, является применение концептуальных бункеров моделей (Concept Bottleneck Models, CBMs), которые могут объяснять свои решения с помощью человекочитаемых понятий. Однако, существуют сложности с их использованием в реальных задачах, так как их ограниченность может привести к ошибкам в определении критических факторов. Мы предлагаем новую модель, которая использует контекстно-зависимые понятия из клинических записей для оптимизации CBMs. Это позволяет улучшить точность диагностики ARDS и повысить уровень доверия к результатам. ## Метод Мы проводим исследование, используя модель Concept Bottleneck Models, которая работает на основе классификации понятий, связанных с ARDS. Модель обучается на данных, где каждая строка — это описание клинической картины заболевания. Затем, чтобы улучшить эту модель, мы вводим дополнительные понятия, полученные с помощью Ланды Ларги Модели (LLM), которые просматривают и обрабатывают клинические записи. Эта модель использует не только стандартные метки, но и дополнительные контекстные сведения, которые могут быть недоступны в традиционных данных. Мы также применяем методы обучения с подкреплением (Reinforcement Learning) для точного определения и выделения ключевых понятий, связанных с ARDS. ## Результаты Мы проводим эксперименты на большом наборе клинических данных, включающих записи более чем 10 000 пациентов. Модель, включающая контекстные понятия, показала 10% более высокую точность по сравнению с базовой моделью CBM. Этот результат достигнут благодаря обнаружению более сложных и подробных понятий, которые могут быть отсутствовали в первоначальных данных. Кроме того, мы обнаружили, что использование LLM уменьшает риск использования спурайвных связей (spurious shortcuts), что влечет за собой повышение надежности и точности диагностики ARDS. ## Значимость Улучшенная модель может применяться в различ