Efficient Single-Exposure Holographic Imaging via a Lightweight Distilled Strategy

Digital holography can capture and reconstruct 3D object information, making it valuable for biomedical imaging and materials science. However, traditional holographic reconstruction methods require the use of phase shift operation in the time or space domain combined with complex computational processes, which, to some extent, limits the range of application areas. The integration of deep learning (DL) advancements with physics-informed methodologies has opened new avenues for tackling this challenge. However, most of the existing DL-based holographic reconstruction methods have high model complexity. In this study, we first design a lightweight model with fewer parameters through the synergy of deep separable convolution and Swish activation function and then employ it as a teacher to distill a smaller student model. By reducing the number of network layers and utilizing knowledge distillation to improve the performance of a simple model, high-quality holographic reconstruction is achieved with only one hologram, greatly reducing the number of parameters in the network model. This distilled lightweight method cuts computational expenses dramatically, with its parameter count representing just 5.4% of the conventional Unet-based method, thereby facilitating efficient holographic reconstruction in settings with limited resources.