Spherical Brownian Bridge Diffusion Models for Conditional Cortical Thickness Forecasting

## Контекст Форкация и воспроизведение индивидуальных, высокорезольвентных траекторий телесной толщины (CTh) являются ключевыми задачами в генезе и мониторинге новорожденных. Это необходимо для выявления небольших локальных изменений в луковичной структуре мозга, которые могут быть признаками ранних сигналов развития нейродегенеративных заболеваний. Однако эта задача сложна из-за сложной не-Eвклидовой геометрии луковицы мозга, а также из-за необходимости интегрировать многомодальные данные для подготовки точных индивидуальных прогнозов. Настоящая работа адресует эти проблемы, предлагая новую модель Spherical Brownian Bridge Diffusion Model (SBDM), которая обеспечивает достоверное воспроизведение траекторий телесной толщины на уровне вершин регистрируемых луковичных поверхностей. ## Метод Методология SBDM основывается на инновационном подходе, основанном на теории диффузии. Мы предлагаем бидирекциональную схему, которая использует процессы Brownian bridge diffusion для построения траекторий телесной толщины с учетом кондиционированных данных. Сердцем модели лежит уникальный нормализующий модуль, который стабилизирует процесс диффузии и обеспечивает точность прогноза. Мы также предлагаем новую архитектуру CoS-UNet, которая включает в себя сферические конволюции и кросс-аттенцию, чтобы объединить луковичные поверхности и внешние условия в единую модель. Это позволяет модели обрабатывать различные данные, такие как медицинские изображения и другие метрики. ## Результаты Мы проводили эксперименты на двух крупных датасетах: ADNI и OASIS. Наши результаты показывают, что SBDM значительно превосходит предыдущие методы по меременю ошибок прогноза. Например, на ADNI мы получили ошибку в 12% меньше в сравнении с базовым подходом, при этом обеспечивая более высокую точность в прогнозировании траекторий телесной толщины. Мы также продемонстрировали SBDM может генерировать не только правдивые траектории, но и контрфактические (hypothetical) сценарии, обеспечивая новый фреймворк для исследований ранних сигналов нейродегенеративных заболеваний. ## Значимость Модель SBDM имеет высокий потенциал в области медицинских исследований и клинической практики. Она может быть применена для раннего выявления нейродегенеративных заболеваний, а также для мониторинга клинических интервенций. Ее особенность в том, что она может вести прогнозирование на не только стандартных, но и кейсах контрфактических сценариев, что позволяет экспертам более глубоко изучать развитие заболеваний и выявлять новые механизмы. Это может способствовать разработке эффективных терапи