Few-shot deployment of pretrained MRI transformers in brain imaging tasks

Introduction Transformer-based deep learning has shown great potential in medical imaging, but its real-world applicability remains limited due to the scarcity of annotated data. This study aims to develop a practical framework for the few-shot deployment of pretrained MRI transformers across diverse brain imaging tasks. Methods We employ a Masked Autoencoder (MAE) pretraining strategy on a large-scale, multi-cohort brain MRI dataset comprising over 31 million 2D slices to learn transferable representations. For classification tasks, a frozen MAE encoder with a lightweight linear head (MAE-classify) is used. For segmentation, we propose MAE-FUnet, a hybrid architecture that fuses pretrained MAE embeddings with multi-scale CNN features. Extensive evaluations are conducted on multiple datasets, including NACC, ADNI, OASIS, NFBS, SynthStrip, and MRBrainS18, under controlled few-shot settings. Results The proposed framework achieves state-of-the-art performance in MRI sequence classification, reaching an accuracy of 99.24% with only 6,152 trainable parameters. For segmentation tasks, MAE-FUnet consistently outperforms strong baselines, achieving superior Dice and IoU scores across skull stripping and multi-class anatomical segmentation benchmarks. The model also demonstrates enhanced robustness and stability under data-limited conditions, with lower performance variance compared to competing methods. Discussion These results highlight the effectiveness of pretrained MAE representations for few-shot medical imaging tasks. The proposed framework enables efficient, scalable, and adaptable deployment of transformer-based models in data-constrained clinical environments. The fusion of global transformer embeddings with local CNN features provides a generalizable design paradigm for a wide range of medical imaging applications.

Authors

Institutions

Publication Details

Journal
Frontiers in Artificial Intelligence
Published
2026-04-01
DOI
https://doi.org/10.3389/frai.2026.1771088
Primary Topic
Advanced Neural Network Applications
Type
article
Field-Weighted Citation Impact
0.00

Funders

Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Few-shot deployment of pretrained MRI transformers in brain imaging tasks

Mengyu Li, Chad W. Farris, Xin Zhang
Frontiers in Artificial Intelligence
Advanced Neural Network Applications
article

Few-shot deployment of pretrained MRI transformers in brain imaging tasks

Mengyu Li, Chad W. Farris, Xin Zhang
article en

Abstract

Introduction Transformer-based deep learning has shown great potential in medical imaging, but its real-world applicability remains limited due to the scarcity of annotated data. This study aims to develop a practical framework for the few-shot deployment of pretrained MRI transformers across diverse brain imaging tasks. Methods We employ a Masked Autoencoder (MAE) pretraining strategy on a large-scale, multi-cohort brain MRI dataset comprising over 31 million 2D slices to learn transferable representations. For classification tasks, a frozen MAE encoder with a lightweight linear head (MAE-classify) is used. For segmentation, we propose MAE-FUnet, a hybrid architecture that fuses pretrained MAE embeddings with multi-scale CNN features. Extensive evaluations are conducted on multiple datasets, including NACC, ADNI, OASIS, NFBS, SynthStrip, and MRBrainS18, under controlled few-shot settings. Results The proposed framework achieves state-of-the-art performance in MRI sequence classification, reaching an accuracy of 99.24% with only 6,152 trainable parameters. For segmentation tasks, MAE-FUnet consistently outperforms strong baselines, achieving superior Dice and IoU scores across skull stripping and multi-class anatomical segmentation benchmarks. The model also demonstrates enhanced robustness and stability under data-limited conditions, with lower performance variance compared to competing methods. Discussion These results highlight the effectiveness of pretrained MAE representations for few-shot medical imaging tasks. The proposed framework enables efficient, scalable, and adaptable deployment of transformer-based models in data-constrained clinical environments. The fusion of global transformer embeddings with local CNN features provides a generalizable design paradigm for a wide range of medical imaging applications.

Frontiers in Artificial IntelligenceVol. 9
Boston University (US), Boston Medical Center (US), Rafik Hariri University (LB), Division of Materials Science and Engineering (AU), Analog Photonics (United States) (US)
Eli Lilly and Company, National Institutes of Health, Avid Radiopharmaceuticals, DoD Alzheimer's Disease Neuroimaging Initiative
Openalex Percentile: Top 38%
Advanced Neural Network Applications
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.