Model-Guided Iterative Diffusion Sampling for NLOS Reconstruction

Xiongfei Su,Yu Hong,Jun-tian Ye,Feihu Xu,Xin Yuan
DOI: https://doi.org/10.1109/jstqe.2024.3350971
IF: 4.9
2024-01-01
IEEE Journal of Selected Topics in Quantum Electronics
Abstract:We consider the reconstruction problem of non-line-of-sight (NLOS) imaging. The proposed diffusion-based method starts with back-projection from NLOS measurement and iteratively estimates the noise using a trained light-weight U-Net and model-guided iterative gradient descent. Unlike previous large networks, our diffusion-based method relies on a series of iterative refinement steps, each of which is trained with a regression loss. This benefits our iterative approach to capture richer distributions. Furthermore, rather than estimating the posterior mean value, our method generates samples from the target posterior and allows only one trained model being adaptive to different settings, such as diverse sampling time resolution, various spatial resolution and multiple channels of color scenes. Simulation and real-data experiments verify that our proposed method achieves better reconstruction results both in quality and quantity than existing methods.
engineering, electrical & electronic,optics,physics, applied,quantum science & technology
What problem does this paper attempt to address?