Patch-Based Diffusion for Data-Efficient, Radiologist-Preferred MRI Reconstruction
2509.21531v1
eess.IV, cs.CV
2025-09-30
Авторы:
Rohan Sanda, Asad Aali, Andrew Johnston, Eduardo Reis, Jonathan Singh, Gordon Wetzstein, Sara Fridovich-Keil
Резюме на русском
## Контекст
Оптимизация методов магнитной резонансной импедитометрии (MRI) является ключевым направлением в современной радиологии, так как длительные времена захвата данных приводят к увеличению затрат, снижению доступности и ухудшению качества снимков из-за движущихся объектов. Одним из перспективных подходов является использование моделей на основе распухания (diffusion probabilistic models), которые могут учитывать данные для построения эффективных предварительных моделей. Однако эти модели часто требуют больших объемов данных, что оказывается дорогостоящим и сложным в реализации. Наряду с этим, патч-базисные модели распухания (patch-based diffusion models) показали свою эффективность в обучении предварительных моделей на основе реальных данных, но до сих пор не были тестированы на практике в клинических условиях. Целью настоящего исследования является расширение метода Patch-Based Diffusion Inverse Solver (PaDIS) для реконструкции MRI сложных многомодульных сигналов, а также сравнение его с текущими лучшими решениями в этой области.
## Метод
Патч-базисное распухание для реконструкции MRI (Patch-Based Diffusion Inverse Solver, PaDIS) является методом обучения с подкреплением, который использует предварительные модели для распределения вероятностей на основе локальных патчей данных. В рамках данного исследования, метод PaDIS был расширен для решения задачи реконструкции MRI сложного, многомодульного сигнала (complex-valued multi-coil MRI). Модели были обучены на небольших выборках данных (25 тысяч к-пространственных изображений), чтобы продемонстрировать их эффективность при минимальных затратах. Для сравнения использовался современный предварительный метод FastMRI-EDM, который использует на вход всю информацию о поле видимости. Метрики качества, такие как PSNR, SSIM и NRMSE, были вычислены для оценки результатов. Для оценки потенциала моделей в клинических условиях проведена клиническая испытательная испытательная испытания с участием трех радиологов.
## Результаты
Проведенные эксперименты показали, что модели PaDIS-MRI, обученные на малых выборках данных, превосходят FastMRI-EDM во всех основных метриках качества изображений (PSNR, SSIM, NRMSE). Особенно заметное преимущество замечено в области пиксельной неопределенности, а также в ситуациях с серьезной недоразметкой к-пространственных данных. Также была проверена кросс-контрастная общность моделей, которая демонстрирует возможность PaDIS-MRI работать в разных условиях. В клиническом испытании с тремя радиологами, PaDIS-MRI были выбраны как более диагностически полезные в 91.7% случаев, что значительно превышает базовые модели.
## Значимость
Результаты показывают, что патч-базисные модели распухания
Abstract
Magnetic resonance imaging (MRI) requires long acquisition times, raising
costs, reducing accessibility, and making scans more susceptible to motion
artifacts. Diffusion probabilistic models that learn data-driven priors can
potentially assist in reducing acquisition time. However, they typically
require large training datasets that can be prohibitively expensive to collect.
Patch-based diffusion models have shown promise in learning effective
data-driven priors over small real-valued datasets, but have not yet
demonstrated clinical value in MRI. We extend the Patch-based Diffusion Inverse
Solver (PaDIS) to complex-valued, multi-coil MRI reconstruction, and compare it
against a state-of-the-art whole-image diffusion baseline (FastMRI-EDM) for 7x
undersampled MRI reconstruction on the FastMRI brain dataset. We show that
PaDIS-MRI models trained on small datasets of as few as 25 k-space images
outperform FastMRI-EDM on image quality metrics (PSNR, SSIM, NRMSE),
pixel-level uncertainty, cross-contrast generalization, and robustness to
severe k-space undersampling. In a blinded study with three radiologists,
PaDIS-MRI reconstructions were chosen as diagnostically superior in 91.7% of
cases, compared to baselines (i) FastMRI-EDM and (ii) classical convex
reconstruction with wavelet sparsity. These findings highlight the potential of
patch-based diffusion priors for high-fidelity MRI reconstruction in
data-scarce clinical settings where diagnostic confidence matters.
Ссылки и действия
Дополнительные ресурсы: