Three-Step Conditional Diffusion 3D Reconstruction for Light-Field Microscopy
About
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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| 3D Reconstruction | LFM 3D Reconstruction Dataset average of 5 scenes (test) | PSNR41.72 | 6 | |
| 3D Volumetric Reconstruction | Tubulin | PSNR38.41 | 6 | |
| 3D Volumetric Reconstruction | Vessel | PSNR36.04 | 6 | |
| 3D Volumetric Reconstruction | Bcell | PSNR47.54 | 6 | |
| 3D Volumetric Reconstruction | Mito | PSNR45.77 | 6 | |
| 3D Volumetric Reconstruction | Dendrite | PSNR38.85 | 6 |