SSL-AD: Spatiotemporal Self-Supervised Learning for Generalizability and Adaptability Across Alzheimer's Prediction Tasks and Datasets
2509.10453v1
cs.CV, cs.LG
2025-09-16
Авторы:
Emily Kaczmarek, Justin Szeto, Brennan Nichyporuk, Tal Arbel
Резюме на русском
## Контекст
Alzheimer's disease (AD) — это хроническая, прогрессирующая заболевание, характеризующееся постепенным ухудшением памяти и когнитивных функций. Несмотря на развитые глубокие нейросетевые модели для предсказания AD, они сталкиваются с рядом проблем: ограниченная доступность меток данных, слабая община (generalizability) между различными данными, а также невозможность адаптироваться к разному числу сканов и интервалам времени между ними. Эти ограничения подчеркивают необходимость разработки моделей, которые могут обеспечить более широкую общину и гибкость при анализе данных.
## Метод
В предложенной работе разрабатывается модель **SSL-AD**, основанная на **спектрально-пространственной самосупервизорной обучении (SSL)**, которая адаптируется для анализа трехмерных МРТ-сканов головного мозга. Методология основывается на двух ключевых компонентах: **подходе типа темпоральной порядковой прогнозирования** и **контрастивному обучению**. В качестве дополнения к этим подходам введены новые методы, способные справиться с проблемой неодинакового числа входных сканов и интервалов времени между ними. Эта модель позволяет повысить удовлетворяемость (robustness) модели к разным клиническим задачам, включая классификацию диагноза, определение конверсии и прогностическую модель будущих конверсий.
## Результаты
Для тренировки использовались данные из четырех открытых баз данных, включающих 3,161 пациентов. Модель была протестирована на нескольких задачах предсказания AD, включая классификацию, определение конверсии и прогноз будущих конверсий. Результаты показали, что SSL-AD показала лучшую производительность по сравнению с моделями на основе супервизорного обучения в 6 из 7 задач. Особое внимание уделено ее гибкости и робастности при работе с разными входными данными, включая разное количество сканов и разные интервалы времени между ними.
## Значимость
Предложенная модель может быть применена в различных клинических применениях, в том числе в диагностике, мониторинге и прогнозировании AD. Она предлагает важное решение для проблемы нехватки меток, а также обеспечивает высокую общину и гибкость при работе с разными данными. Благодаря этому, SSL-AD может стать полезным инструментом для разработки более точных и эффективных моделей в области предсказания и мониторинга AD.
## Выводы
SSL-AD демонстрирует возможность использования самосупервизорного обучения для улучшения общины и гибкости моделей в области предсказания AD. Она показала значительное улучшение по сравнению с супервизорным обучением и может быть применена в различных клинических сценариях. Будущие ис
Abstract
Alzheimer's disease is a progressive, neurodegenerative disorder that causes
memory loss and cognitive decline. While there has been extensive research in
applying deep learning models to Alzheimer's prediction tasks, these models
remain limited by lack of available labeled data, poor generalization across
datasets, and inflexibility to varying numbers of input scans and time
intervals between scans. In this study, we adapt three state-of-the-art
temporal self-supervised learning (SSL) approaches for 3D brain MRI analysis,
and add novel extensions designed to handle variable-length inputs and learn
robust spatial features. We aggregate four publicly available datasets
comprising 3,161 patients for pre-training, and show the performance of our
model across multiple Alzheimer's prediction tasks including diagnosis
classification, conversion detection, and future conversion prediction.
Importantly, our SSL model implemented with temporal order prediction and
contrastive learning outperforms supervised learning on six out of seven
downstream tasks. It demonstrates adaptability and generalizability across
tasks and number of input images with varying time intervals, highlighting its
capacity for robust performance across clinical applications. We release our
code and model publicly at https://github.com/emilykaczmarek/SSL-AD.
Ссылки и действия
Дополнительные ресурсы: