arXiv preprint ·

A Generalizable Feature Extractor for Alzheimer's-Related Brain MRI Tasks

Reza Rajabli and D. Louis Collins

Published
Source
arXiv preprint, posted before peer review
Original
Read it at arXiv
Topics
Computer VisionQuantitative Methods

In brief

Unlocking Alzheimer's Insights with Fewer Labels

This paper introduces a compact, pretrained 3D CNN model that can be adapted to various Alzheimer's-related tasks with minimal additional parameters. It demonstrates that a model trained for brain age prediction can serve as a robust foundation for diverse neuroimaging applications, even when faced with data constraints.

What they found: The study found that a compact model trained on brain age can be effectively adapted to various Alzheimer's-related tasks with minimal additional parameters. The model achieved an impressive AUC of 0.964 in classifying dementia on the ADNI dataset and maintained strong performance across other tasks, including predicting MCI progression and estimating brain volumes.

AI-written summary. Check anything you rely on against the original paper.

Abstract

When there is not enough labeled data to properly train deep learning models, transfer learning can help. We still do not fully understand how effective it is in neuroimaging, especially for Alzheimer's disease research. It is also not clear if these transferred models can work on new datasets without being retrained for each specific task. We evaluate whether a compact, supervised pretrained model can serve as a reusable foundation model for downstream neuroimaging tasks. We freeze the 7.18 million weights of a 3D CNN previously trained for brain-age prediction, and adapt it to each task using Low-Rank Adaptation (LoRA), requiring only ~1% additional trainable parameters. We evaluate generalizability in six experiments. Adapting the model to classify cognitively normal versus Dementia on ADNI gave an AUC of 0.964 on held-out folds (Experiment #1). Applying that adapted model unchanged to OASIS-3, with no retraining, gave an AUC of 0.871 (Experiment #2). Reusing its output logit together with age and a cognitive score distinguished stable from progressing MCI with an AUC of 0.828 (Experiment #3). Adapting the same backbone to predict amyloid positivity from structural MRI gave an AUC of 0.804 (Experiment #4). Finally, the same approach estimated ICV-normalized hippocampal and white matter hypointensity volumes directly from the T1w image, with R^2 of 0.80 and 0.91 respectively, tasks normally addressed with much larger U-Net networks (Experiments #5 and #6). A compact model supervised on brain age can therefore serve as a reusable backbone, adapting to each task with ~1% additional parameters and transferring to an unseen cohort without any training. Our findings suggest that a carefully trained brain age model can serve as an effective foundation model for Alzheimer's related tasks, even under strict data constraints.

The authors’ abstract, as arXiv lists it. arXiv’s listings, abstracts included, are free for anyone to reuse.

Read it on ReadSearch

Each opens after you create an account, and brings you back to this paper.

Already a reader? Sign in

All new papers in Computer Vision

Get the few papers worth your time in Computer Vision every morning.

ReadSearch reads the new papers in your field each day and brings you the few worth reading, each with its key finding in a sentence.

Papers belong to their authors and publishers; every page here links to the original. All research topics