YOSO: single-frame Gerchberg-Saxton phase retrieval with AI-based data augmentation for in-line holography

TL;DR

YOSO is a deep learning framework that enhances single-shot in-line holography by generating multi-height datasets for improved phase retrieval, enabling fast, generalizable, and physics-consistent hologram processing.

Contribution

The paper introduces YOSO, a novel AI-based method that combines deep learning with traditional algorithms for efficient, accurate, and flexible hologram reconstruction from a single image.

Findings

YOSO achieves rapid training under two hours on standard hardware.

It generalizes well across different systems and sample types.

YOSO enables effective 3D object reconstruction and numerical refocusing.

Abstract

We present YOSO (You Only Shot Once), a single-frame phase retrieval framework for digital in-line holographic microscopy (DIHM) in which supervised deep learning is used to numerically generate an additional hologram corresponding to different defocus distance, creating a so-called multi-height dataset, which is then conventionally processed with a well-established Gerchberg-Saxton (GS) algorithm. YOSO is trained on computer-generated data derived from natural images, enabling strong generalization. The selected multi-scale ResNet architecture enables rapid training in under two hours on a mid-range workstation, which is done only once, enabling efficient inference thereafter. We further show that YOSO network can process inputs of varying spatial dimensions, allowing training on small inputs and direct inference on full-sized holograms while bypassing patch-and-stitch procedure. A further advantage of YOSO is its physics-consistent hologram padding, which replaces conventional zero or edge-value padding with a physically grounded approach compatible with the GS framework. The YOSO framework is tested on various systems (lens-based and lensless DIHM) and diverse samples: a resolution test target, adherent and suspended biological cells, and a mouse brain slice. The results show that YOSO is compatible with 3D objects and correctly recovers defocused object wave features, enabling holographic postprocessing such as numerical refocusing. The results of this work are available publicly as software for end-to-end implementation.