Phase retrieval explained

Phase retrieval is the process of algorithmically finding solutions to the phase problem. Given a complex signal

F(k)

, of amplitude

|F(k)|

, and phase

\psi(k)

:

F(k)=|F(k)|ei

infty
=\int
-infty

f(x)e-dx

where x is an M-dimensional spatial coordinate and k is an M-dimensional spatial frequency coordinate. Phase retrieval consists of finding the phase that satisfies a set of constraints for a measured amplitude. Important applications of phase retrieval include X-ray crystallography, transmission electron microscopy and coherent diffractive imaging, for which

M=2

.[1] Uniqueness theorems for both 1-D and 2-D cases of the phase retrieval problem, including the phaseless 1-D inverse scattering problem, were proven by Klibanov and his collaborators (see References).

Problem formulation

Here we consider 1-D discrete Fourier transform (DFT) phase retrieval problem. The DFT of a complex signal

f[n]

is given by
N-1
F[k]=\sum
n=0

f[n]

-j2\pi
kn
N
,
e

=|F[k]|ej\psi[k]k=0,1,\ldots,N-1

,

and the oversampled DFT of

x

is given by
N-1
F[k]=\sum
n=0

f[n]

-j2\pi
kn
M
,
e

k=0,1,\ldots,M-1

,

where

M>N

.

Since the DFT operator is bijective, this is equivalent to recovering the phase

\psi[k]

. It is common recovering a signal from its autocorrelation sequence instead of its Fourier magnitude. That is, denote by

\hat{f}

the vector

f

after padding with

N-1

zeros. The autocorrelation sequence of

\hat{f}

is then defined as

g[m]=\sumi=max

}^ \hat_ \overline, \quad m=-(N-1), \ldots, N-1 ,

and the DFT of

g[m]

, denoted by

G[k]

, satisfies

G[k]=|F[k]|2

.

Methods

Error reduction algorithm

The error reduction is a generalization of the Gerchberg–Saxton algorithm. It solves for

f(x)

from measurements of

|F(u)|

by iterating a four-step process. For the

k

th iteration the steps are as follows:

Step (1):

Gk(u)

,

\phik

, and

gk(x)

are estimates of, respectively,

F(u)

,

\psi

and

f(x)

. In the first step we calculate the Fourier transform of

gk(x)

:

Gk(u)=

i\phik(u)
|G
k(u)|e

=l{F}(gk(x)).

Step (2): The experimental value of

|F(u)|

, calculated from the diffraction pattern via the signal equation, is then substituted for

|Gk(u)|

, giving an estimate of the Fourier transform:

G'k(u)=

i\phik(u)
|F(u)|e

,

where the ' denotes an intermediate result that will be discarded later on.

Step (3): the estimate of the Fourier transform

G'k(u)

is then inverse Fourier transformed:

g'k(x)

i\theta'k(x)
=|g'
k(x)|e

=l{F}-1(G'k(u)).

Step (4):

g'k(x)

then must be changed so that the new estimate of the object,

gk+1(x)

, satisfies the object constraints.

gk+1(x)

is therefore defined piecewise as:

gk+1(x)\equiv\begin{cases} g'k(x),&x\notin\gamma,\\ 0,&x\in\gamma, \end{cases}

where

\gamma

is the domain in which

g'k(x)

does not satisfy the object constraints. A new estimate

gk+1(x)

is obtained and the four step process is repeated.

This process is continued until both the Fourier constraint and object constraint are satisfied. Theoretically, the process will always lead to a convergence, but the large number of iterations needed to produce a satisfactory image (generally >2000) results in the error-reduction algorithm by itself being unsuitable for practical applications.

Hybrid input-output algorithm

See main article: Hybrid input output (HIO) algorithm for phase retrieval.

The hybrid input-output algorithm is a modification of the error-reduction algorithm - the first three stages are identical. However,

gk(x)

no longer acts as an estimate of

f(x)

, but the input function corresponding to the output function

g'k(x)

, which is an estimate of

f(x)

. In the fourth step, when the function

g'k(x)

violates the object constraints, the value of

gk+1(x)

is forced towards zero, but optimally not to zero. The chief advantage of the hybrid input-output algorithm is that the function

gk(x)

contains feedback information concerning previous iterations, reducing the probability of stagnation. It has been shown that the hybrid input-output algorithm converges to a solution significantly faster than the error reduction algorithm. Its convergence rate can be further improved through step size optimization algorithms.[2]

gk+1(x)\equiv\begin{cases} g'k(x),&x\notin\gamma,\\ gk(x)-\beta{g'k(x)},&x\in\gamma. \end{cases}

Here

\beta

is a feedback parameter which can take a value between 0 and 1. For most applications,

\beta0.9

gives optimal results.

Shrinkwrap

For a two dimensional phase retrieval problem, there is a degeneracy of solutions as

f(x)

and its conjugate

f*(-x)

have the same Fourier modulus. This leads to "image twinning" in which the phase retrieval algorithm stagnates producing an image with features of both the object and its conjugate.[3] The shrinkwrap technique periodically updates the estimate of the support by low-pass filtering the current estimate of the object amplitude (by convolution with a Gaussian) and applying a threshold, leading to a reduction in the image ambiguity.[4]

Semidefinite relaxation-based algorithm for short time Fourier transform

The phase retrieval is an ill-posed problem. To uniquely identify the underlying signal, in addition to the methods that adds additional prior information like Gerchberg–Saxton algorithm, the other way is to add magnitude-only measurements like short time Fourier transform (STFT).

The method introduced below mainly based on the work of Jaganathan et al.[5]

Short time Fourier transform

Given a discrete signal

x=(f[0],f[1],...,f[N-1])T

which is sampled from

f(x)

. We use a window of length W:

w=(w[0],w[1],...,w[W-1])T

to compute the STFT of

f

, denoted by

Y

:
N-1
Y[m,r]=\sum
n=0
-i2\pi
mn
N
{f[n]w[rL-n]e
}

for

0\leqm\leqN-1

and

0\leqr\leqR-1

, where the parameter

L

denotes the separation in time between adjacent short-time sections and the parameter

R=\left\lceil

N+W-1
L

\right\rceil

denotes the number of short-time sections considered.

The other interpretation (called sliding window interpretation) of STFT can be used with the help of discrete Fourier transform (DFT). Let

wr[n]=w[rL-n]

denotes the window element obtained from shifted and flipped window

w

. Then we have

Y=[Y0,Y1,...,YR-1]

, where

Yr=x\circwr

.

Problem definition

Let

{Z}w[m,r]=|Y[m,r]|2

be the

N x R

measurements corresponding to the magnitude-square of the STFT of

x

,

Wr

be the

N x N

diagonal matrix with diagonal elements

\left(wr[0],wr[1],\ldots,wr[N-1]\right).

STFT phase retrieval can be stated as:

Find

x

such that

Zw[m,r]=\left|\left\langlefm,Wrx\right\rangle\right|2

for

0\leqm\leqN-1

and

0\leqr\leqR-1

, where

fm

is the

m

-th column of the

N

-point inverse DFT matrix.

Intuitively, the computational complexity growing with

N

makes the method impractical. In fact, however, for the most cases in practical we only need to consider the measurements corresponding to

0\leqm\leqM

, for any parameter

M

satisfying

2W\leqM\leqN

.

To be more specifically, if both the signal and the window are not vanishing, that is,

x[n]0

for all

0\leqn\leqN-1

and

w[n]0

for all

0\leq

n\leqW-1

, signal

x

can be uniquely identified from its STFT magnitude if the following requirements are satisfied:

L<W\leqN/2

,

2W\leqM\leqN

.

The proof can be found in Jaganathan' s work, which reformulates STFT phase retrieval as the following least-squares problem:

minx

R-1
\sum
r=0
N-1
\sum
m=0

\left(Zw[m,r]-\left|\left\langlefm,Wrx\right\rangle\right|2\right)2

.

The algorithm, although without theoretical recovery guarantees, empirically able to converge to the global minimum when there is substantial overlap between adjacent short-time sections.

Semidefinite relaxation-based algorithm

To establish recovery guarantees, one way is to formulate the problems as a semidefinite program (SDP), by embedding the problem in a higher dimensional space using the transformation

X=xx\ast

and relax the rank-one constraint to obtain a convex program. The problem reformulated is stated below:

Obtain

\hat{X

} by solving:\begin&\mathrm~~~\mathrm(\mathbf)\\[6pt]&\mathrm~~Z[m,r]=\mathrm(\mathbf_r^\mathbf_m\mathbf_m^\ast\mathbf_r\mathbf)\\[0pt]&~~~~~~~~~~~~~~~~~~~\mathbf\succeq0\endfor

1\leqm\leqM

and

0\leqr\leqR-1

Once

\hat{X

} is found, we can recover signal

x

by best rank-one approximation.

Applications

Phase retrieval is a key component of coherent diffraction imaging (CDI). In CDI, the intensity of the diffraction pattern scattered from a target is measured. The phase of the diffraction pattern is then obtained using phase retrieval algorithms and an image of the target is constructed. In this way, phase retrieval allows for the conversion of a diffraction pattern into an image without an optical lens.

Using phase retrieval algorithms, it is possible to characterize complex optical systems and their aberrations.[6] For example, phase retrieval was used to diagnose and repair the flawed optics of the Hubble Space Telescope.[7] [8]

Other applications of phase retrieval include X-ray crystallography and transmission electron microscopy.

See also

References

Notes and References

  1. Fienup. J. R.. 1982-08-01. Phase retrieval algorithms: a comparison. Applied Optics. en. 21. 15. 2758–69. 10.1364/AO.21.002758. 20396114. 1982ApOpt..21.2758F . 0003-6935.
  2. Marchesini. S.. 25 January 2007. Invited Article: A unified evaluation of iterative projection algorithms for phase retrieval. Review of Scientific Instruments. en. 78. 1. 011301–011301–10. 10.1063/1.2403783. 17503899. 0034-6748. physics/0603201. 2007RScI...78a1301M . 7462041 .
  3. Fienup. J. R.. Wackerman. C. C.. 1986-11-01. Phase-retrieval stagnation problems and solutions. Journal of the Optical Society of America A. en. 3. 11. 1897. 10.1364/JOSAA.3.001897. 1986JOSAA...3.1897F . 1084-7529.
  4. Marchesini. S.. He. H.. Chapman. H. N.. Hau-Riege. S. P.. Noy. A.. Howells. M. R.. Weierstall. U.. Spence. J. C. H.. 2003-10-28. X-ray image reconstruction from a diffraction pattern alone. Physical Review B. en. 68. 14. 140101. 10.1103/PhysRevB.68.140101. physics/0306174. 2003PhRvB..68n0101M . 14224319 . 0163-1829.
  5. Jaganathan. Kishore. Eldar. Yonina C.. Hassibi. Babak. June 2016. STFT Phase Retrieval: Uniqueness Guarantees and Recovery Algorithms. IEEE Journal of Selected Topics in Signal Processing. 10. 4. 770–781. 10.1109/JSTSP.2016.2549507. 1508.02820 . 2016ISTSP..10..770J . 1941-0484. free.
  6. Fienup. J. R.. 1993-04-01. Phase-retrieval algorithms for a complicated optical system. Applied Optics. EN. 32. 10. 1737–1746. 10.1364/AO.32.001737. 20820307. 1993ApOpt..32.1737F . 2155-3165.
  7. Web site: First person: A scientist's discovery puts space into focus . www.wbur.org . April 2022 . 30 May 2022. Interview with Professor Robert Gonsalves.
  8. Krist. JE. Burrows. CJ. 1995-08-01. Phase-retrieval analysis of pre- and post-repair Hubble Space Telescope images. Applied Optics. EN. 34. 22. 4951–64. 10.1364/AO.34.004951. 21052338. 1995ApOpt..34.4951K .