Learning Mechanistic Subtypes of Neurodegeneration with a Physics-Informed Variational Autoencoder Mixture Model

## Контекст Моделирование механизмов прогрессирования нейродегенеративных заболеваний требует методов, которые могли бы локально и гетерогенно отражать динамику, основываясь на неограниченных, высокомерных данных из нейроиммунинга. Интеграция знаний, основанных на физических уравнениях (PDE), с машинным обучением позволяет повысить интерпретируемость и эффективность по сравнению с традиционными численными методами. Однако работы, интегрирующие PDE с машинным обучением, ограничиваются рассмотрением одного PDE, что ограничивает применимость к заболеваниям, где несколько механизмов приводят к разным подгруппам (или подтипам). Данные проблемы усугубляются в моделях, не учитывающих таковую субтипификацию. Мы предлагаем глубокую генерирующую модель, которая может учитывать несколько динамических моделей, управляемых PDE, в рамках фреймворка вариационного автоэнкодера (VAE). Модель позволяет выделять субтипы с интерпретируемыми латентными переменными, такими как распространение (diffusivity) и реакционные скорости, из нейроиммунинга. Мы проверили нашу модель на синтетических данных и показали, что она может раскрывать механизмы прогрессирования заболевания Альцгеймера, основываясь на данных из позитронной эмиссионной томографии (PET). ## Метод Мы предлагаем **Variational Autoencoder Mixture Model (VAEMM)** с встроенными реакционно-диффузионными PDE. Реакционно-диффузионные PDE логически разделяются на подгруппы для каждого субтипа, чтобы отражать специфику динамики этого субтипа. Модель использует метод реконструкции VAE для выявления латентных переменных, которые описывают эти субтипы. Нашу архитектуру модели укрепляет физический смысл, внедренный в регуляризацию. Мы вводим новую функцию потерь, которая объединяет представление VAE с ограничениями, у imposеd by PDE. Эта регуляризация позволяет обеспечить точность в моделировании динамики и найти подтипы с логически согласованными латентными переменными. Мы использовали бенчмарки, созданные из имитационных данных, для проверки нашей модели. ## Результаты Мы проверили нашу модель на синтетических данных, генерируемых из различных предварительных моделей, включая реакционно-диффузионные PDE. Модель показала способность точно выявить субтипы и их латентные переменные, такие как распространение и реакционные скорости. Мы также применили нашу модель к реальным данным из PET-сканеров для прогрессирования Альцгеймера. Мы выявили подтипы, отражающие различные стили динамики заболевания, и показали, что модель может улучшить интерпретируемость томографических данных. Экспери