Spatio-Temporal Conditional Diffusion Models for Forecasting Future Multiple Sclerosis Lesion Masks Conditioned on Treatments
2508.07006v1
eess.IV, cs.CV
2025-08-13
Авторы:
Gian Mario Favero, Ge Ya Luo, Nima Fathi, Justin Szeto, Douglas L. Arnold, Brennan Nichyporuk, Chris Pal, Tal Arbel
Резюме на русском
## Контекст
Многие заболевания, включая Множественные Склероз (МС), протекают неодинаково и зависят от многих факторов, в том числе лечения и различных клинических особенностей. Изучение этого хаотического процесса является ключевым для улучшения прогнозирования и лечения таких заболеваний. Несмотря на то, что имеется много работ по созданию моделей прогностических масок для МС, существуют значительные проблемы, такие как скрытые зависимости от лечения, недостаточная точность и отсутствие специфики к индивидуальному пациенту. Наша модель предназначена для решения этих проблем, используя инновационные методы генерирования изображений, чтобы предоставить более точное и личностно-ориентированное прогнозирование здоровья.
## Метод
Мы предлагаем первую модель, которая использует синергетические способности спектрально-временной модели. Метод основан на методике условных достаточностей для временных последовательностей, включая информацию о лечении. Мы разработали архитектуру, которая учитывает все необходимые мультимодальные данные, включая изображения MRI и клинические данные о лечении. Модель использует архитектуру "diffusion model", которая позволяет формировать уточненные прогнозы для т fuтурных масок NET2. Эта модель также включает каскадное уточнение, что дает ей возможность учитывать динамические изменения в реальном времени и улучшать точность прогнозов.
## Результаты
Мы провели исследования на большом многоцентровом наборе данных, содержащем 2131 трехмерных MRI от пациентов с МС. Модель была тестирована на случаях с разными лечениями и показала высокую достоверность в прогнозировании новых и увеличивающихся T2-масок (NET2) в будущем. Мы также проводили эксперименты на задачах, таких как оценка количества новых лесеньев, их расположение и классификация активности лесеньев. Результаты показали, что наша модель превосходит конкурентные модели в тех же задачах, обеспечивая более точные и предсказуемые результаты.
## Значимость
Наша модель может сыграть важную роль в создании новых подходов к подтверждению лечений и прогностическому анализу в многих областях медицины. Она может быть использована для личностно-ориентированного лечения, обеспечения более точных прогнозов и раннего выявления проблем. Модель также имеет потенциал для распространения на другие заболевания, характеризующиеся хроническим или неоднородным прогрессом, такие как диабет или раковые заболевания. Это демонстрирует важность использования искусственного интеллекта в медицинской информатике.
## Выводы
Мы представляем первую модель, которая объединя
Abstract
Image-based personalized medicine has the potential to transform healthcare,
particularly for diseases that exhibit heterogeneous progression such as
Multiple Sclerosis (MS). In this work, we introduce the first treatment-aware
spatio-temporal diffusion model that is able to generate future masks
demonstrating lesion evolution in MS. Our voxel-space approach incorporates
multi-modal patient data, including MRI and treatment information, to forecast
new and enlarging T2 (NET2) lesion masks at a future time point. Extensive
experiments on a multi-centre dataset of 2131 patient 3D MRIs from randomized
clinical trials for relapsing-remitting MS demonstrate that our generative
model is able to accurately predict NET2 lesion masks for patients across six
different treatments. Moreover, we demonstrate our model has the potential for
real-world clinical applications through downstream tasks such as future lesion
count and location estimation, binary lesion activity classification, and
generating counterfactual future NET2 masks for several treatments with
different efficacies. This work highlights the potential of causal, image-based
generative models as powerful tools for advancing data-driven prognostics in
MS.
Ссылки и действия
Дополнительные ресурсы: