Abstract:The problem of phase retrieval (PR) involves recovering an unknown image from limited amplitude measurement data and is a challenge nonlinear inverse problem in computational imaging and image processing. However, many of the PR methods are based on black-box network models that lack interpretability and plug-and-play (PnP) frameworks that are computationally complex and require careful parameter tuning. To address this, we have developed PRISTA-Net, a deep unfolding network (DUN) based on the first-order iterative shrinkage thresholding algorithm (ISTA). This network utilizes a learnable nonlinear transformation to address the proximal-point mapping sub-problem associated with the sparse priors, and an attention mechanism to focus on phase information containing image edges, textures, and structures. Additionally, the fast Fourier transform (FFT) is used to learn global features to enhance local information, and the designed logarithmic-based loss function leads to significant improvements when the noise level is low. All parameters in the proposed PRISTA-Net framework, including the nonlinear transformation, threshold parameters, and step size, are learned end-to-end instead of being manually set. This method combines the interpretability of traditional methods with the fast inference ability of deep learning and is able to handle noise at each iteration during the unfolding stage, thus improving recovery quality. Experiments on Coded Diffraction Patterns (CDPs) measurements demonstrate that our approach outperforms the existing state-of-the-art methods in terms of qualitative and quantitative evaluations. Our source codes are available at \emph{<a class="link-external link-https" href="https://github.com/liuaxou/PRISTA-Net" rel="external noopener nofollow">this https URL</a>}.

Robust Contrast-Transfer-function Phase Retrieval Via Flexible Deep Learning Networks

Optimal Phase Mask Design for Coherent Modulation Imaging by Deep Learning

Automatic Phase Recognition Method Based on Convolutional Neural Network

Phase Retrieval for X-ray Differential Phase Contrast Radiography with Knowledge Transfer Learning from Virtual Differential Absorption Model.

Single-frame Noisy Interferogram Phase Retrieval Using an End-to-end Deep Learning Network with Physical Information Constraints

Quantitative Pure-Phase Object Reconstruction under Single-Shot Fourier Measurement Via Deep Learning

Phase Retrieval from a Single Diffraction Intensity Pattern by Generating the Support Constraint Using Deep Learning

Spatio-Temporal 3-D Residual Networks for Simultaneous Detection and Depth Estimation of CFRP Subsurface Defects in Lock-In Thermography

Three-dimensional Coherent X-ray Diffraction Imaging via Deep Convolutional Neural Networks

Phase Retrieval at All Defocus Distances

Phase Retrieval Via Nonlocal Complex-Domain Sparsity.

Model-driven Phase Retrieval Network for Single-Shot X-Ray Talbot-Lau Interferometer Imaging.

Real-time 3D Nanoscale Coherent Imaging via Physics-aware Deep Learning

NeuPh: Scalable and Generalizable Neural Phase Retrieval with Local Conditional Neural Fields

Enhancing the X-Ray Differential Phase Contrast Image Quality With Deep Learning Technique

Physics-informed Deep Neural Network Reconstruction Framework for Propagation-Based X Ray Phase-Contrast Computed Tomography with Sparse-View Projections

Learning from Projection to Reconstruction: A Deep Learning Reconstruction Framework for Sparse-View Phase Contrast Computed Tomography Via Dual-Domain Enhancement

Phase Retrieval Based on Convolutional Neural Network

Using Diffraction Deep Neural Networks for Indirect Phase Recovery Based on Zernike Polynomials

PRISTA-Net: Deep Iterative Shrinkage Thresholding Network for Coded Diffraction Patterns Phase Retrieval