We demonstrate a deep learning-based framework, called Fourier Imager Network (FIN), which achieves unparalleled generalization in end-to-end phase-recovery and hologram reconstruction. We used Fourier transform modules in FIN architecture, which process the spatial frequencies of the input images in a global receptive field and bring strong regularization and robustness to the hologram reconstruction task. We validated FIN by training it on human lung tissue samples and blindly testing it on human prostate, salivary gland, and Pap smear samples. FIN exhibits superior internal and external generalization compared with existing hologram reconstruction models, also achieving a ~50-fold acceleration in image inference speed.
|