Generalizing the Gerchberg–Saxton algorithm for retrieving complex optical transmission matrices

The Gerchberg–Saxton (GS) algorithm, which retrieves phase information from the measured intensities on two related planes (the source plane and the target plane), has been widely adopted in a variety of applications when holographic methods are challenging to be implemented. In this work, we showed that the GS algorithm can be generalized to retrieve the unknown propagating function that connects these two planes. As a proof-of-concept, we employed the generalized GS (GGS) algorithm to retrieve the optical transmission matrix (TM) of a complex medium through the measured intensity distributions on the target plane. Numerical studies indicate that the GGS algorithm can efficiently retrieve the optical TM while maintaining accuracy. With the same training data set, the computational time cost by the GGS algorithm is orders of magnitude less than that consumed by other non-holographic methods reported in the literature. Besides numerical investigations, we also experimentally demonstrated retrieving the optical TMs of a stack of ground glasses and a 1-m-long multimode fiber using the GGS algorithm. The accuracy of the retrieved TM was evaluated by synthesizing high-quality single foci and multiple foci on the target plane through these complex media. These results indicate that the GGS algorithm can handle a large TM with high efficiency, showing great promise in a variety of applications in optics.