Yamin Arefeen, Sidharth Kumar, Steven Warach, Hamidreza Saber, Jonathan I. Tamir
7 min
Abstract
Purpose: To develop a data-efficient strategy for accelerated MRI reconstruction with Diffusion Probabilistic Generative Models (DPMs) that enables faster scan times in clinical stroke MRI when only limited fully-sampled data samples are available. Methods: Our simple training strategy, inspired by the foundation model paradigm, first trains a DPM on a large, diverse collection of publicly available brain MRI data in fastMRI and then fine-tunes on a small dataset from the target application using carefully selected learning rates and fine-tuning durations. The approach is evaluated on controlled fastMRI experiments and on clinical stroke MRI data with a blinded clinical reader study. Results: DPMs pre-trained on approximately 4000 subjects with non-FLAIR contrasts and fine-tuned on FLAIR data from only 20 target subjects achieve reconstruction performance comparable to models trained with substantially more target-domain FLAIR data across multiple acceleration factors. Experiments reveal that moderate fine-tuning with a reduced learning rate yields improved performance, while insufficient or excessive fine-tuning degrades reconstruction quality. When applied to clinical stroke MRI, a blinded reader study involving two neuroradiologists indicates that images reconstructed using the proposed approach from $2 \times$ accelerated data are non-inferior to standard-of-care in terms of image quality and structural delineation. Conclusion: Large-scale pre-training combined with targeted fine-tuning enables DPM-based MRI reconstruction in data-constrained, accelerated clinical stroke MRI. The proposed approach substantially reduces the need for large application-specific datasets while maintaining clinically acceptable image quality, supporting the use of foundation-inspired diffusion models for accelerated MRI in targeted applications.
Alex: Okay, so it's like mixing a recipe from memory with the few ingredients you have on hand, adjusting the balance to avoid a bad dish. But how does the pre-training on general brains transfer so well to stroke-specific ones without messing up?
Sam: The pre-training gives the model a strong starting point on brain anatomy across contrasts like T1 and T2 scans from 4000 people. For stroke FLAIR images—which highlight fluid buildup in damaged tissue by dimming watery areas like spinal fluid—they fine-tune all the model's weights using just 20 subjects, but with a learning rate 10 times lower and only 2% as many training steps as the initial phase. This gentle tweak adapts it to stroke details without overwriting the broad knowledge or overfitting to too few examples—the paper shows this matches performance trained on 344 FLAIR cases.
Alex: That lower rate and short time makes sense—like proofreading a long essay with quick passes instead of rewriting everything. Does this hold up in real stroke cases, not just lab data?
Sam: Yes, they pre-trained on fastMRI brains including some FLAIR, then fine-tuned on 25 local stroke patients' data for contrasts like DWI and SWI. Tested on 5 more, it reconstructed 2x faster scans with quality radiologists deemed non-inferior to standard ones in a blinded review—meaning no worse for spotting anatomy or lesions, despite 17 times less stroke data. DPMs shine here because they're flexible to different scanner setups, unlike rigid methods needing matching data.
Alex: Huh. So the real win is turning scarce ER data into reliable speed without quality dips. But in that blinded review with radiologists, how exactly did they measure if the faster images were truly up to par for spotting stroke damage?
Sam: Two experienced neuroradiologists—one with 11 years, the other 30—looked at image sets from 80 stroke patients, each set mixing standard and sped-up versions without labels or hints about which was which. They scored things like clarity of brain layers, noise levels, sharpness, and overall usefulness on a scale from one needing a re-scan to five being outstanding, judging whole sets together since doctors see multiple image types at once. Stats like paired Wilcoxon signed-rank tests showed no meaningful quality drop, and Cohen's kappa measured solid agreement between them—meaning the faster AI versions held up for real diagnosis. Scores stayed high overall, around 4 or better out of 5.
Alex: Okay, so random blind judging by pros confirms it's not worse, backed by stats matching scores across versions. What about scans done fast from the start, not just simulated shortcuts?
Sam: They tested that too, scanning a healthy volunteer with hospital stroke settings for FLAIR and SWI sequences—one at normal speed, one prospectively faster by skipping data points upfront. The AI, pre-trained broadly then fine-tuned on their stroke cases, reconstructed the fast ones plus extra-retrospectively sped-up standards, beating a wavelet baseline method on error measures. This shows the approach works on real undersampled data, not just lab tricks.
Alex: Huh—that bridges lab to clinic even more solidly. Were there bigger catches, like how real-world scans or timing might limit this?
Sam: A main one is that most tests used retrospective undersampling—starting with full data and pretending it's sparse—rather than true fast scans from the outset. Prospective tests were just on a healthy volunteer for FLAIR and SWI, where the model beat a basic wavelet method but needs stroke patient proof. Reconstruction times are also long per slice compared to standard tools, slowing ER use until sped up. And while quality held, future checks should see if it changes actual diagnosis decisions.
Alex: Right, so strong lab-to-clinic transfer with scarce data, but real deployment needs prospective stroke data and faster processing to cut those ER delays fully. Pulling this together, it suggests a meaningful step toward routine faster stroke MRI.
Sam: Exactly. It addresses the ER delay from long scans and motion, letting MRI rival CT's speed for better stroke detection. The evidence from radiologists confirms it's clinically viable, balancing adaptation without overfitting.
Alex: That's our look at accelerating stroke MRI through smart pre-training and fine-tuning. Thanks for joining ResearchPod.