用三步扩散模型提升光场显微镜3D重建速度与精度
Three-Step Conditional Diffusion 3D Reconstruction for Light-Field Microscopy

- 三步确定性采样+轻量条件U-Net,加速重建过程
- 在多个数据集上优于当前最佳方法,重建质量更高
- 新增异常检测模块,提升模型推理稳定性
光场显微镜(LFM)可实现单次拍摄获取生物样本的多角度信息,支持实时体成像。然而,传统基于物理的方法常受限于空间分辨率低、伪影严重及计算成本高。现有基于学习的方法虽提升了推理效率,但在重建精度和泛化能力上仍有不足。为此,本文提出一种高保真三步条件扩散(TCD)3D重建方法。尽管扩散模型在生成建模中表现卓越,但其采样慢且质量与效率存在权衡,限制了在实时3D成像中的应用。我们通过确定性的三步采样策略与轻量级条件U-Net重构扩散过程,建立快速准确的体成像新范式。此外,引入跨类别检测(ICD)模块,在推理阶段识别分布外或异常输入,增强模型稳定性与可靠性。大量实验与跨数据集评估表明,TCD在重建保真度与泛化能力上显著优于现有先进方法,为光场显微镜提供高效实用的3D重建解决方案。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。