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.

Ссылки и действия