cs.CVMay 24, 2026

Three-Step Conditional Diffusion 3D Reconstruction for Light-Field Microscopy

Authors: Qihong ZhaoShaokang YanZhimin QiaoJinjia WangBo Xiong

Organizations: 1Yanshan University · 2Peking University

Abstract

Light-field microscopy (LFM) enables single-shot capture of multi-angular information from biological samples, supporting real-time volumetric imaging. However, traditional physics-based algorithms often suffer from limited spatial resolution, severe artifacts, and high computational costs. Existing learning-based methods improve inference efficiency but still face limitations in reconstruction accuracy and generalization capability. To address these challenges, this paper proposes a high-fidelity Three-Step Conditional Diffusion (TCD) 3D reconstruction method for LFM. Although conventional diffusion models have achieved remarkable success in generative modeling, their slow sampling process and the inherent trade-off between quality and efficiency hinder their application in real-time 3D imaging. We redesign the diffusion process through a deterministic three-step sampling strategy coupled with a lightweight conditional U-Net, establishing a new paradigm for fast and accurate volumetric reconstruction. Furthermore, an Inter-Class Detection (ICD) module is incorporated to identify out-of-distribution or anomalous inputs during inference, thereby enhancing model stability and reliability. Extensive experiments and cross-dataset evaluations demonstrate that TCD significantly outperforms state-of-the-art methods in both reconstruction fidelity and generalization, providing an efficient and practical 3D reconstruction solution for light-field microscopy.

Explore similar work

Sep 17, 2026cs.CV

STAR: Structure-aware Test-time Adaptation for diffusion-based light field Reconstruction

Light field (LF) reconstruction from limited and noisy focal stack (FS) measurements is a highly ill-posed inverse problem. Although the LF-to-FS imaging geometry is fixed for a given optical setup, LF spatial-angular structure---including within-view spatial details, cross-view angular dependencies, and disparity across views---varies across scenes. Consequently, a fixed pre-trained prior may not optimally capture the spatial-angular structure of each test LF. We propose Structure-aware Test-time Adaptation for diffusion-based light field Reconstruction (STAR), the first test-time adaptation framework for reconstructing an LF from FS. For each test LF, STAR freezes a pre-trained diffusion prior and fits three lightweight adapters to the observed FS to jointly adapt the three components of the LF's spatial-angular structure. STAR outperforms existing state-of-the-art methods in both two- and three-focal-sheet settings, with shorter inference times than those with test-time parameter updates.
Wontae Choi, Ki Ryum Moon, Jae Young Lee +2
Nov 6, 2025cs.CV

A Linear Fractional Transformation Model and Calibration Method for Light Field Camera

Accurate intrinsic calibration is a crucial yet challenging prerequisite for 3D reconstruction using light field cameras. Existing calibration models typically analyze the main lens and micro lens array (MLA) in a coupled manner, resulting in high complexity and a large number of parameters. In this paper, we propose a linear fractional transformation (LFT) model that introduces a single parameter αα to decouple the imaging processes of the main lens and the MLA. A dedicated matrix Hα\mathbf{H}_α is designed to characterize the MLA projection, enabling the main lens and the MLA to be calibrated independently. The proposed calibration method consists of an analytical least-squares solution for Hα\mathbf{H}_α, followed by joint nonlinear refinement of all intrinsic parameters. Experimental results on both physical datasets and simulated data demonstrate that the proposed method achieves a mean translation error of 2.1%2.1\%, outperforming the state-of-the-art, while maintaining sub-pixel reprojection accuracy. The complete codebase, including a light field simulator based on the proposed model, is openly available to the research community.
Zhong Chen, Changfeng Chen
Jun 30, 2026cs.CV

Wavelet-Optimized Pseudo-3D Accelerated Diffusion Model for Truncated Computed Laminography

Computed Laminography (CL) is a key technology for the nondestructive testing of large plate-shaped objects. However, field-of-view (FOV) limitations inevitably lead to truncation of projected data, an ill-posed inverse problem that causes severe reconstruction artifacts. Existing deep learning methods typically rely on 2D architectures that lack rigorous data consistency constraints. Furthermore, they conventionally confine artifact removal strictly to the FOV, discarding potentially recoverable information outside it. To overcome these limitations, we first introduce a comprehensive CL FOV analysis, categorizing the space into data-complete, data-incomplete, and data-free regions. By extending our reconstruction target to encompass the data-incomplete region, we significantly expand the effective imaging range and enhance scanning efficiency. To achieve this, we propose a novel wavelet-optimized pseudo-3D accelerated diffusion model for CL truncation reconstruction (CL-DM). Our method utilizes a standard 2D diffusion model for slice aggregation, combined with a 3D model-based iterative reconstruction (MBIR) method to ensure strict data consistency. To mitigate inter-slice discontinuities, we introduce wavelet regularization along the z-direction, paired with a translation-invariant (TI) mechanism and a low-frequency preservation strategy. Finally, we introduce a 3D fast sampling architecture, significantly accelerating inference speed. Extensive simulations and real-world experiments demonstrate that CL-DM is superior in effectively eliminating truncation artifacts and restoring high-fidelity, continuous 3D structures.
Genyuan Zhang, Junyao Wang, Chuandong Tan +2