В лаборатории, где данные пациентов напоминают россыпь золотого песка, каждый анализ — крупица, а болезнь Альцгеймера — река, уносящая драгоценный металл. Учёные поставили перед нейросетью задачу: отличить, получал ли человек рапамицин, лекарство, замедляющее старение клеток, от обычного состояния. Но песок перемешан с пустой породой: шум в анализах, разброс между людьми, слабый эффект терапии. Чтобы найти самородки, сначала нужно отбросить лишнее.
Модель TreatmentFormer работает как многоступенчатая промывка. Сначала «случайный лес» — ансамбль решающих деревьев — просеивает восемь типов данных: снимки мозга, маркеры воспаления, метаболиты крови, микробиом и другие. Из сотен признаков остаются 27 самых ценных, по три от каждой модальности. Затем отдельные кодировщики превращают каждый тип данных в компактные векторы-эмбеддинги, а контрастное обучение сближает векторы одного пациента и разводит чужие. Наконец, трансформер собирает всё вместе и выдаёт ответ: было лечение или нет.
Результат оказался не идеальным, но значимым. Без фильтрации модель угадывала случайно — около 50%. С фильтрацией точность выросла до 71,25% в среднем по десяти запускам, хотя разброс составлял от 50% до 87,5%. SHAP-анализ показал, что главный вклад вносят показатели общего анализа крови, особенно RDW CV %, и воспалительный маркер MIP-1α. Метаболиты крови дали 14 из 27 признаков, но по отдельности были слабее. Демография почти не важна, а МРТ и микробиом добавляли умеренную пользу.
До этой работы было неясно, можно ли по небольшой группе пациентов уловить слабый след лекарства в шуме. Теперь понятно: ключ не в сложности модели, а в очистке входных данных. Метод сравнивали на одних и тех же данных с фильтрацией и без неё, а важность признаков проверяли двумя независимыми способами. Уже сегодня подход может применяться в клинических испытаниях для отбора пациентов, которым рапамицин с наибольшей вероятностью поможет, и для поиска биомаркеров ответа.
Песчинки снова блестят в лотке, но теперь видно, какие из них золото. Нейросеть не просто угадывает — она показывает, где искать.
Рапамицин был впервые выделен из бактерии, найденной в почве на острове Пасхи (Рапа-Нуи), отсюда и название. Препарат подавляет сигнальный путь mTOR, который регулирует рост клеток и старение, и продлевает жизнь мышам на 10–15%.
Идея предсказывать реакцию организма на лекарство по множеству биологических сигналов перекликается с романом Фрэнка Герберта «Дюна», где врачи используют сложные диагностические системы для оценки состояния человека. Впрочем, в отличие от фантастики, здесь речь о реальных данных крови и мозга.
Работа опубликована не на arXiv. Здесь её страница, PDF и формализация: проверить можно самим, а не только прочитать наш разбор.