Препринт — материал ещё не прошёл рецензирование
Один текст для данных пациента оказался не хуже специальных моделей
Кратко
Авторы превратили сведения из электронных медкарт в один обычный текст и дообучили на нём языковую модель. Такой подход проверили на трёх клинических задачах: прогнозе смерти в больнице, отторжения трансплантата и сортировке вызовов скорой помощи. Он не уступил или обошёл специальные модели, а для прогноза отторжения ещё и превзошёл модель, которую уже используют в клинике.
Зачем это знать
Если это подтвердится в других больницах и задачах, врачебные ИИ-системы можно будет строить проще, без сложных схем для слияния разных типов данных. Это может ускорить разработку таких моделей и сделать их легче для внедрения.
Разбор
Авторы взяли довольно простую идею: вместо того чтобы строить отдельные блоки для разных типов данных — например, для заметок врача, давления, анализов и списка болезней — они перевели всё это в один связный текст. Потом они дообучили обычные языковые модели так, чтобы те учились находить связи между кусками этого текста сами, без специальной схемы для «сшивания» разных источников данных.
Зачем это было нужно? Потому что в клинических ИИ-системах такие схемы обычно приходится собирать заново под каждую новую задачу. Это долго, сложно и плохо переносится из одной больницы в другую. Авторы хотели проверить, можно ли обойтись более универсальным способом — и при этом не потерять в качестве прогнозов.
Проверку сделали сразу на трёх очень разных задачах: смерти в стационаре, долгосрочного отторжения трансплантата и сортировки вызовов скорой помощи. Сравнили модели двух типов: одни читают текст целиком и сразу выдают ответ, другие работают как генеративные модели и предсказывают нужный класс по токенам. Важный результат в том, что такой «текстовый» подход не просто держался на уровне специальных мультимодальных моделей, а в некоторых задачах их обходил. Самый заметный пример — прогноз отторжения трансплантата: там новый способ оказался лучше даже клинической модели на основе градиентного бустинга, которую уже используют в практике.
Ещё один полезный нюанс: сильнее всего этот подход выглядел там, где данные шумные и короткие, как в заметках бригады скорой помощи. То есть идея не только про удобство, но и про то, что языковые модели могут неплохо справляться с реальными «грязными» медицинскими записями, если подать им всё в одном формате.
Ключевые цифры
Можно ли доверять
Это препринт на arXiv, то есть работу ещё не прошли независимые рецензенты. При этом авторы проверили идею сразу на трёх задачах и на разных типах данных, а для одной из задач сравнили её даже с клинической моделью из реальной практики — это повышает доверие. Но результаты пока лучше считать сильным ранним сигналом: нужны проверки в других больницах и на других наборах пациентов.
Пересказано ИИ по научной статье. Как это устроено