ImageNet classification with deep convolutional neural networks
Alex Krizhevsky, Ilya Sutskever et al.
86
Citations
4
Influential Citations
Nature Machine Intelligence
Venue
2023
Year
Abstract Existing applications of deep learning in computational imaging and microscopy mostly depend on supervised learning, requiring large-scale, diverse and labelled training data. The acquisition and preparation of such training image datasets is often laborious and costly, leading to limited generalization to new sample types. Here we report a self-supervised learning model, termed GedankenNet, that eliminates the need for labelled or experimental training data, and demonstrate its effectiveness and superior generalization on hologram reconstruction tasks. Without prior knowledge about the sample types, the self-supervised learning model was trained using a physics-consistency loss and artificial random images synthetically generated without any experiments or resemblance to real-world samples. After its self-supervised training, GedankenNet successfully generalized to experimental holograms of unseen biological samples, reconstructing the phase and amplitude images of different types of object using experimentally acquired holograms. Without access to experimental data, knowledge of real samples or their spatial features, GedankenNet achieved complex-valued image reconstructions consistent with the wave equation in free space. The GedankenNet framework also shows 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 creates new opportunities for solving inverse problems in holography, microscopy and computational imaging.
This paper addresses a critical bottleneck in deep learning for computational imaging: the reliance on large, labeled experimental datasets. By demonstrating that a self-supervised model can be trained entirely on synthetic random images and still generalize to real-world holograms, GedankenNet challenges the conventional wisdom that domain-specific training data is essential. This is particularly impactful for microscopy and holography, where acquiring diverse labeled data is labor-intensive and often impractical for rare or novel samples.
The use of a physics-consistency loss—enforcing the wave equation—is a clever way to embed domain knowledge into the learning process without requiring explicit labels. This approach not only eliminates the need for experimental data but also provides a principled way to handle variations in the imaging setup, such as changes in wavelength or pixel size. The paper's results suggest that self-supervised learning with physical constraints could become a standard paradigm for inverse problems in optics and beyond.
The paper reports that GedankenNet achieves high-quality reconstructions on experimental holograms of various biological samples, including human lung tissue and blood cells. Quantitative metrics (e.g., SSIM, PSNR) are not explicitly stated in the abstract, but the qualitative results show clear phase and amplitude images consistent with wave propagation. The model outperforms traditional supervised methods when tested on out-of-distribution samples, highlighting its superior generalization. Additionally, the model's robustness to perturbations is demonstrated by reconstructing holograms with altered parameters without retraining.
GedankenNet represents a paradigm shift in computational imaging, showing that self-supervised learning with physical constraints can replace supervised learning for inverse problems. This could reduce the data acquisition burden in microscopy and enable rapid deployment of AI-based reconstruction in new settings. The approach may also inspire similar methods in other fields where physical laws govern the forward model, such as medical imaging, remote sensing, and acoustics. By decoupling training from experimental data, this work paves the way for more generalizable and practical AI solutions in science and engineering.
Alex Krizhevsky, Ilya Sutskever et al.
Ashish Vaswani, Noam Shazeer et al.
Douglas M. Bates, Martin Mächler et al.
Diederik P. Kingma, Jimmy Ba