Self-supervised learning of hologram reconstruction using physics consistency

Luzhe Huang,Hanlong Chen,Tairan Liu,Aydogan Ozcan
DOI: https://doi.org/10.1038/s42256-023-00704-7
2023-06-17
Abstract:The past decade has witnessed transformative applications of deep learning in various computational imaging, sensing and microscopy tasks. Due to the supervised learning schemes employed, these methods mostly depend on large-scale, diverse, and labeled training data. The acquisition and preparation of such training image datasets are often laborious and costly, also leading to biased estimation and limited generalization to new sample types. Here, we report a self-supervised learning model, termed GedankenNet, that eliminates the need for labeled or experimental training data, and demonstrate its effectiveness and superior generalization on hologram reconstruction tasks. Without prior knowledge about the sample types to be imaged, the self-supervised learning model was trained using a physics-consistency loss and artificial random images that are synthetically generated without any experiments or resemblance to real-world samples. After its self-supervised training, GedankenNet successfully generalized to experimental holograms of various unseen biological samples, reconstructing the phase and amplitude images of different types of objects using experimentally acquired test holograms. Without access to experimental data or knowledge of real samples of interest or their spatial features, GedankenNet's self-supervised learning achieved complex-valued image reconstructions that are consistent with the Maxwell's equations, and its output inference and object solutions accurately represent the wave propagation in free-space. GedankenNet framework also exhibits resilience to random, unknown perturbations in the physical forward model, including changes in the hologram distances, pixel size and illumination wavelength. This self-supervised learning of image reconstruction tasks creates new opportunities for various inverse problems in holography, microscopy and computational imaging fields.
Computer Vision and Pattern Recognition,Machine Learning,Image and Video Processing,Optics
What problem does this paper attempt to address?
The main problem this paper attempts to address is the dependency of existing deep learning methods on large-scale, diverse, and annotated training data in hologram reconstruction tasks, as well as the limited generalization ability of these methods when faced with unseen sample types. Specifically: 1. **The need for large-scale, diverse, and annotated training data**: Existing supervised learning methods require a large amount of high-quality and diverse training datasets, which typically contain objects from various sources and types, annotated or with experimental images as ground truth labels. Acquiring and preparing such training image datasets is often labor-intensive and costly, and may lead to biased estimates and limited generalization ability. 2. **Limited generalization ability to unseen sample types**: Existing supervised learning models heavily rely on training image datasets obtained through experiments in microscopic imaging and holography tasks. These datasets vary due to differences in optical hardware, sample types, and imaging (sample preparation) protocols, resulting in limited generalization ability of the models when faced with new types of samples. 3. **Lack of explicit connection to physical laws**: Existing deep learning solutions typically lack explicit physical models during the training phase, which limits the compatibility of network inference with the physical laws of light-matter interaction and wave propagation. To overcome these issues, the paper proposes a self-supervised learning model—GedankenNet, which does not require any experimental data or prior knowledge about sample types and spatial features during training. By training with a loss function based on physical consistency, GedankenNet can successfully generalize to experimental holograms of various unseen biological samples, reconstructing phase and amplitude images of different types of objects. Additionally, GedankenNet demonstrates robustness to random unknown perturbations, including variations in hologram distance, pixel size, and illumination wavelength.