首页|基于物理模型驱动无监督学习的无透镜成像质量增强方法

基于物理模型驱动无监督学习的无透镜成像质量增强方法

扫码查看
数字全息术是一种能够记录和重建物场完整波前信息的光学成像技术.同轴全息术以其更大的空间带宽积和简单的成像系统等优势,得到了广泛关注和应用.然而,传统的同轴全息术在物场重建时会受到零级像和孪生像的干扰,影响对物场复振幅的观察和测量.传统的相位复原算法在重建时会产生较大误差,尤其是在高分辨率、大视场等成像需求下,重建质量会受到影响.并且基于数据驱动的神经网络方法受限于数据集采集和质量.针对这些问题,本文提出了基于振幅约束的双输入物理模型驱动的神经网络(DPNNA)的同轴全息像重建方法.该方法以两幅不同距离下记录的全息图作为输入,结合振幅约束并采用同轴全息成像模型和神经网络相结合的无监督方式,重建物体的相位信息或振幅信息.并且在无需大量训练数据的前提下,对输入的同轴全息图质量要求低,重建结果背景噪声弱.同时,在不同衍射间距和噪声水平的条件下,DPNNA在峰值信噪比和结构相似性等指数方面均优于其他方法,有效实现了相位或振幅的恢复,有较好的鲁棒性和稳定性.
Enhancement Method of Image Quality for Lensless Imaging Based on Physics-Driven Unsupervised Learning
Objective Digital holography is an optical imaging technique that records and reconstructs the complete wavefront information of an object field.In-line holography,known for its broader spatial bandwidth and simpler imaging system,has gained widespread attention and application.However,traditional in-line holography suffers from interference by zero-order and twin images during object field reconstruction,affecting the observation and measurement of the object field's complex amplitude.In addition,traditional phase recovery algorithms can produce significant errors,especially under high-resolution and large field-of-view imaging requirements,where reconstruction quality can be compromised.Inaccurate initial estimates or incomplete prior information may cause the algorithm to fail to converge or produce incorrect results due to the strong reliance of phase retrieval methods on initial estimates and prior information.Data-driven neural network methods are limited by dataset collection and quality and lack interpretability.Physics-driven neural networks combine neural networks with models that adhere to physical laws,mitigating the drawbacks of data-driven neural networks.However,physically motivated neural networks require certain constraints on data acquisition when only a single hologram is provided as input.Consequently,the final reconstruction results are susceptible to the influence of secondary interference fringes,leading to uneven backgrounds in the recovered results.In response to these issues,a method for reconstructing holographic images using a dual-input physics-driven neural network with amplitude constraint(DPNNA)is proposed.Methods Firstly,the DPNNA method is constructed.Two holograms captured at diffraction distances d1 and d2 are used as inputs to the neural network.A physical model is employed to compute the estimated hologram and phase corresponding to diffraction distance d1 based on the neural network's estimated phase.Subsequently,combining the estimated phase at d1 with the true amplitude,the physical model generates the estimated hologram corresponding to diffraction distance d2.The loss function is computed using the estimated and true holograms,and network parameters are optimized to achieve phase or amplitude imaging.Then,in-line holograms of phase and amplitude resolution targets,as well as phase objects with irregular phase variations,are simulated to validate the feasibility of DPNNA method.The peak signal-to-noise ratio(PSNR)and structural similarity(SSIM)among the reconstruction results of traditional Gerchberg-Saxton(G-S)method,the single-input physics-driven neural network method PhysenNet,and the proposed DPNNA method,and the ground truth are calculated.Finally,a lensless imaging optical setup is constructed to record in-line holograms containing information on different objects.The DPNNA method is employed for reconstruction and compared with other in-line holographic reconstruction methods.Results and Discussions The DPNNA method is utilized to simulate the reconstruction of pure phase objects with different phase distributions and is then compared with the reconstruction results obtained using the G-S algorithm and PhysenNet(Figs.5-7).Overall,when compared to the traditional G-S method and PhysenNet,the proposed DPNNA method demonstrated significant advantages in the high-quality recovery of object amplitude and phase information.This is particularly notable in lensless imaging,where it achieves the highest evaluation metrics.Subsequently,numerical reconstruction of holographic images is performed using in-line holograms of phase and amplitude resolution targets recorded in experiments(Figs.8 and 10).The results show that using the DPNNA method for holographic image reconstruction far outperforms other algorithms,with the weakest background noise in the reconstructed images.The DPNNA method not only effectively removes twin images but also demonstrates the robustness for different diffraction distances and noise levels(Tables 3 and 4).Conclusions We propose a DPNNA method.By combining the dual-input physics-driven neural network with the G-S reconstruction algorithm,accurate phase or amplitude reconstruction can be achieved without requiring a large amount of training data,ensuring stability and accuracy.Compared to the traditional G-S algorithm,the DPNNA method performs better in handling edge information and exhibits weaker background noise.The DPNNA method demonstrates good reconstruction accuracy for both amplitude-type and various phase-type objects,with stronger generalization ability and interpretability compared to data-driven neural network methods.The proposed method provides a low-cost,high-precision solution for phase imaging,overcoming dependencies on initial estimates and prior information,as well as issues such as high-cost dataset construction and uneven backgrounds.Combining a dual-input physics-driven neural network with amplitude constraint leads to more accurate in-line holographic imaging.This advancement holds significant application value in the field of computational imaging.

in-line holographyphysics-driven neural networkunsupervised learningamplitude constraint

左嘉乐、张蒙蒙、唐雎、张佳伟、任振波、邸江磊、赵建林

展开 >

光场调控与信息感知工业和信息化部重点实验室,陕西省光信息技术重点实验室,西北工业大学物理科学与技术学院,陕西西安 710129

西北工业大学深圳研究院,广东深圳 518063

广东工业大学信息工程学院,先进光子技术研究院,广东省信息光子技术重点实验室,广东广州 510006

同轴全息术 物理模型驱动神经网络 无监督学习 振幅约束

2024

光学学报
中国光学学会 中国科学院上海光学精密机械研究所

光学学报

CSTPCD北大核心
影响因子:1.931
ISSN:0253-2239
年,卷(期):2024.44(16)