Projection Embedded Diffusion Bridge for CT Reconstruction from Incomplete Data

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Yuang, Jin, Pengfei, Yoon, Siyeop, Tivnan, Matthew, Zhang, Shaoyang, Zhang, Li, Li, Quanzheng, Chen, Zhiqiang, Wu, Dufan
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914115395518464
author Wang, Yuang
Jin, Pengfei
Yoon, Siyeop
Tivnan, Matthew
Zhang, Shaoyang
Zhang, Li
Li, Quanzheng
Chen, Zhiqiang
Wu, Dufan
author_facet Wang, Yuang
Jin, Pengfei
Yoon, Siyeop
Tivnan, Matthew
Zhang, Shaoyang
Zhang, Li
Li, Quanzheng
Chen, Zhiqiang
Wu, Dufan
contents Reconstructing CT images from incomplete projection data remains challenging due to the ill-posed nature of the problem. Diffusion bridge models have recently shown promise in restoring clean images from their corresponding Filtered Back Projection (FBP) reconstructions, but incorporating data consistency into these models remains largely underexplored. Incorporating data consistency can improve reconstruction fidelity by aligning the reconstructed image with the observed projection data, and can enhance detail recovery by integrating structural information contained in the projections. In this work, we propose the Projection Embedded Diffusion Bridge (PEDB). PEDB introduces a novel reverse stochastic differential equation (SDE) to sample from the distribution of clean images conditioned on both the FBP reconstruction and the incomplete projection data. By explicitly conditioning on the projection data in sampling the clean images, PEDB naturally incorporates data consistency. We embed the projection data into the score function of the reverse SDE. Under certain assumptions, we derive a tractable expression for the posterior score. In addition, we introduce a free parameter to control the level of stochasticity in the reverse process. We also design a discretization scheme for the reverse SDE to mitigate discretization error. Extensive experiments demonstrate that PEDB achieves strong performance in CT reconstruction from three types of incomplete data, including sparse-view, limited-angle, and truncated projections. For each of these types, PEDB outperforms evaluated state-of-the-art diffusion bridge models across standard, noisy, and domain-shift evaluations.
format Preprint
id arxiv_https___arxiv_org_abs_2510_22605
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Projection Embedded Diffusion Bridge for CT Reconstruction from Incomplete Data
Wang, Yuang
Jin, Pengfei
Yoon, Siyeop
Tivnan, Matthew
Zhang, Shaoyang
Zhang, Li
Li, Quanzheng
Chen, Zhiqiang
Wu, Dufan
Computer Vision and Pattern Recognition
Medical Physics
Reconstructing CT images from incomplete projection data remains challenging due to the ill-posed nature of the problem. Diffusion bridge models have recently shown promise in restoring clean images from their corresponding Filtered Back Projection (FBP) reconstructions, but incorporating data consistency into these models remains largely underexplored. Incorporating data consistency can improve reconstruction fidelity by aligning the reconstructed image with the observed projection data, and can enhance detail recovery by integrating structural information contained in the projections. In this work, we propose the Projection Embedded Diffusion Bridge (PEDB). PEDB introduces a novel reverse stochastic differential equation (SDE) to sample from the distribution of clean images conditioned on both the FBP reconstruction and the incomplete projection data. By explicitly conditioning on the projection data in sampling the clean images, PEDB naturally incorporates data consistency. We embed the projection data into the score function of the reverse SDE. Under certain assumptions, we derive a tractable expression for the posterior score. In addition, we introduce a free parameter to control the level of stochasticity in the reverse process. We also design a discretization scheme for the reverse SDE to mitigate discretization error. Extensive experiments demonstrate that PEDB achieves strong performance in CT reconstruction from three types of incomplete data, including sparse-view, limited-angle, and truncated projections. For each of these types, PEDB outperforms evaluated state-of-the-art diffusion bridge models across standard, noisy, and domain-shift evaluations.
title Projection Embedded Diffusion Bridge for CT Reconstruction from Incomplete Data
topic Computer Vision and Pattern Recognition
Medical Physics
url https://arxiv.org/abs/2510.22605