Improving 3D Imaging with Pre-Trained Perpendicular 2D Diffusion Models

Image credit: Suhyeon Lee

Abstract

Diffusion models have become a popular approach for image generation and reconstruction due to their numerous advantages. However, most diffusion-based inverse problem-solving methods only deal with 2D images, and even recently published 3D methods do not fully exploit the 3D distribution prior. To address this, we propose a novel approach using two perpendicular pre-trained 2D diffusion models to solve the 3D inverse problem. By modeling the 3D data distribution as a product of 2D distributions sliced in different directions, our method effectively addresses the curse of dimensionality. Our experimental results demonstrate that our method is highly effective for 3D medical image reconstruction tasks, including MRI Z-axis super-resolution, compressed sensing MRI, and sparse-view CT. Our method can generate high-quality voxel volumes suitable for medical applications.

Publication
In IEEE/CVF International Conference on Computer Vision (ICCV), 2023
Hyungjin Chung
Hyungjin Chung
Ph.D. student - Generative Models & Inverse Problems

My research interests include, but is not restricted to developing efficient, modular deep generative models (diffusion models), and solving real-world inverse problems (MRI, tomography, microscopy, phase retrieval, etc.) with deep generative priors.

Related