Solving Fourier Ptychographic Imaging Problems Via Neural Network Modeling and Tensorflow
Total Page:16
File Type:pdf, Size:1020Kb
Solving Fourier ptychographic imaging problems via neural network modeling and TensorFlow SHAOWEI JIANG,1,3 KAIKAI GUO,1,3 JUN LIAO,1,3 AND GUOAN ZHENG1, 2* 1Biomedical Engineering, University of Connecticut, Storrs, CT, 06269, USA 2Electrical and Computer Engineering, University of Connecticut, Storrs, CT, 06269, USA 3These authors contributed equally to this work *[email protected] Abstract: Fourier ptychography is a recently developed imaging approach for large field-of- view and high-resolution microscopy. Here we model the Fourier ptychographic forward imaging process using a convolution neural network (CNN) and recover the complex object information in the network training process. In this approach, the input of the network is the point spread function in the spatial domain or the coherent transfer function in the Fourier domain. The object is treated as 2D learnable weights of a convolution or a multiplication layer. The output of the network is modeled as the loss function we aim to minimize. The batch size of the network corresponds to the number of captured low-resolution images in one forward / backward pass. We use a popular open-source machine learning library, TensorFlow, for setting up the network and conducting the optimization process. We analyze the performance of different learning rates, different solvers, and different batch sizes. It is shown that a large batch size with the Adam optimizer achieves the best performance in general. To accelerate the phase retrieval process, we also discuss a strategy to implement Fourier-magnitude projection using a multiplication neural network model. Since convolution and multiplication are the two most- common operations in imaging modeling, the reported approach may provide a new perspective to examine many coherent and incoherent systems. As a demonstration, we discuss the extensions of the reported networks for modeling single-pixel imaging and structured illumination microscopy (SIM). 4-frame resolution doubling is demonstrated using a neural network for SIM. The link between imaging systems and neural network modeling may enable the use of machine-learning hardware such as neural engine and tensor processing unit for accelerating the image reconstruction process. We have made our implementation code open- source for the broad research community. Key words: (100.4996) Pattern recognition, neural networks; (170.3010) Image reconstruction techniques; (170.0180) Microscopy. References and links 1. G. Zheng, R. Horstmeyer, and C. Yang, "Wide-field, high-resolution Fourier ptychographic microscopy," Nature Photonics 7, 739-745 (2013). 2. X. Ou, R. Horstmeyer, C. Yang, and G. Zheng, "Quantitative phase imaging via Fourier ptychographic microscopy," Optics Letters 38, 4845-4848 (2013). 3. L. Tian, X. Li, K. Ramchandran, and L. Waller, "Multiplexed coded illumination for Fourier Ptychography with an LED array microscope," Biomedical optics express 5, 2376-2389 (2014). 4. K. Guo, S. Dong, and G. Zheng, "Fourier Ptychography for Brightfield, Phase, Darkfield, Reflective, Multi-Slice, and Fluorescence Imaging," IEEE Journal of Selected Topics in Quantum Electronics 22, 1-12 (2016). 5. V. Mico, Z. Zalevsky, P. García-Martínez, and J. García, "Synthetic aperture superresolution with multiple off- axis holograms," JOSA A 23, 3162-3170 (2006). 6. J. Di, J. Zhao, H. Jiang, P. Zhang, Q. Fan, and W. Sun, "High resolution digital holographic microscopy with a wide field of view based on a synthetic aperture technique and use of linear CCD scanning," Appl. Opt. 47, 5654- 5659 (2008). 7. T. R. Hillman, T. Gutzler, S. A. Alexandrov, and D. D. Sampson, "High-resolution, wide-field object reconstruction with synthetic aperture Fourier holographic optical microscopy," Optics Express 17, 7873-7892 (2009). 8. L. Granero, V. Micó, Z. Zalevsky, and J. García, "Synthetic aperture superresolved microscopy in digital lensless Fourier holography by time and angular multiplexing of the object information," Appl. Opt. 49, 845-857 (2010). 9. T. Gutzler, T. R. Hillman, S. A. Alexandrov, and D. D. Sampson, "Coherent aperture-synthesis, wide-field, high- resolution holographic microscopy of biological tissue," Opt. Lett. 35, 1136-1138 (2010). 10. A. Meinel, "Aperture synthesis using independent telescopes," Applied Optics 9, 2501-2501 (1970). 11. T. Turpin, L. Gesell, J. Lapides, and C. Price, "Theory of the synthetic aperture microscope," Proc. SPIE 2566, 230-240 (1995). 12. J. R. Fienup, "Phase retrieval algorithms: a comparison," Applied optics 21, 2758-2769 (1982). 13. V. Elser, "Phase retrieval by iterated projections," JOSA A 20, 40-55 (2003). 14. H. M. L. Faulkner and J. M. Rodenburg, "Movable Aperture Lensless Transmission Microscopy: A Novel Phase Retrieval Algorithm," Physical Review Letters 93, 023903 (2004). 15. A. M. Maiden and J. M. Rodenburg, "An improved ptychographical phase retrieval algorithm for diffractive imaging," Ultramicroscopy 109, 1256-1262 (2009). 16. R. Gonsalves, "Phase retrieval from modulus data," JOSA 66, 961-964 (1976). 17. J. R. Fienup, "Reconstruction of a complex-valued object from the modulus of its Fourier transform using a support constraint," JOSA A 4, 118-123 (1987). 18. E. J. Candes, T. Strohmer, and V. Voroninski, "Phaselift: Exact and stable signal recovery from magnitude measurements via convex programming," Communications on Pure and Applied Mathematics 66, 1241-1274 (2013). 19. E. J. Candes, Y. C. Eldar, T. Strohmer, and V. Voroninski, "Phase retrieval via matrix completion," SIAM Review 57, 225-251 (2015). 20. I. Waldspurger, A. d’Aspremont, and S. Mallat, "Phase recovery, maxcut and complex semidefinite programming," Mathematical Programming 149, 47-81 (2015). 21. X. Ou, G. Zheng, and C. Yang, "Embedded pupil function recovery for Fourier ptychographic microscopy," Optics Express 22, 4960-4972 (2014). 22. G. Zheng, "Breakthroughs in Photonics 2013: Fourier Ptychographic Imaging," Photonics Journal, IEEE 6, 1-7 (2014). 23. G. Zheng, X. Ou, R. Horstmeyer, J. Chung, and C. Yang, "Fourier Ptychographic Microscopy: A Gigapixel Superscope for Biomedicine," Optics and Photonics News, April Issue, 26-33 (2014). 24. W. Hoppe and G. Strube, "Diffraction in inhomogeneous primary wave fields. 2. Optical experiments for phase determination of lattice interferences," Acta Crystallogr. A 25, 502-507 (1969). 25. J. Rodenburg, A. Hurst, A. Cullis, B. Dobson, F. Pfeiffer, O. Bunk, C. David, K. Jefimovs, and I. Johnson, "Hard- x-ray lensless imaging of extended objects," Physical review letters 98, 034801 (2007). 26. J. Rodenburg, "Ptychography and related diffractive imaging methods," Advances in Imaging and Electron Physics 150, 87 (2008). 27. R. Horstmeyer and C. Yang, "A phase space model of Fourier ptychographic microscopy," Optics Express 22, 338-358 (2014). 28. T. B. Edo, D. J. Batey, A. M. Maiden, C. Rau, U. Wagner, Z. D. Pešić, T. A. Waigh, and J. M. Rodenburg, "Sampling in x-ray ptychography," Physical Review A 87, 053850 (2013). 29. D. J. Batey, D. Claus, and J. M. Rodenburg, "Information multiplexing in ptychography," Ultramicroscopy 138, 13-21 (2014). 30. P. Thibault and A. Menzel, "Reconstructing state mixtures from diffraction measurements," Nature 494, 68-71 (2013). 31. A. M. Maiden, M. J. Humphry, and J. M. Rodenburg, "Ptychographic transmission microscopy in three dimensions using a multi-slice approach," Journal of the Optical Society of America A 29, 1606-1614 (2012). 32. T. Godden, R. Suman, M. Humphry, J. Rodenburg, and A. Maiden, "Ptychographic microscope for three- dimensional imaging," Opt. Express 22, 12513 (2014). 33. P. Thibault, M. Dierolf, O. Bunk, A. Menzel, and F. Pfeiffer, "Probe retrieval in ptychographic coherent diffractive imaging," Ultramicroscopy 109, 338-343 (2009). 34. M. Abadi, P. Barham, J. Chen, Z. Chen, A. Davis, J. Dean, M. Devin, S. Ghemawat, G. Irving, and M. Isard, "TensorFlow: A System for Large-Scale Machine Learning," in OSDI, 2016), 265-283. 35. N. P. Jouppi, C. Young, N. Patil, D. Patterson, G. Agrawal, R. Bajwa, S. Bates, S. Bhatia, N. Boden, and A. Borchers, "In-datacenter performance analysis of a tensor processing unit," in Proceedings of the 44th Annual International Symposium on Computer Architecture, (ACM, 2017), 1-12. 36. L. Bian, J. Suo, G. Zheng, K. Guo, F. Chen, and Q. Dai, "Fourier ptychographic reconstruction using Wirtinger flow optimization," Optics Express 23, 4856-4866 (2015). 37. D. P. Kingma and J. Ba, "Adam: A method for stochastic optimization," arXiv preprint arXiv:1412.6980 (2014). 38. A. Maiden, D. Johnson, and P. Li, "Further improvements to the ptychographical iterative engine," Optica 4, 736- 745 (2017). 39. M. Odstrčil, A. Menzel, and M. Guizar-Sicairos, "Iterative least-squares solver for generalized maximum- likelihood ptychography," Optics Express 26, 3108-3123 (2018). 40. M. F. Duarte, M. A. Davenport, D. Takhar, J. N. Laska, T. Sun, K. F. Kelly, and R. G. Baraniuk, "Single-pixel imaging via compressive sampling," IEEE signal processing magazine 25, 83-91 (2008). 41. B. Sun, M. P. Edgar, R. Bowman, L. E. Vittert, S. Welsh, A. Bowman, and M. Padgett, "3D computational imaging with single-pixel detectors," Science 340, 844-847 (2013). 42. Z. Zhang, X. Ma, and J. Zhong, "Single-pixel imaging by means of Fourier spectrum acquisition," Nature communications 6, 6225 (2015). 43. M. G. Gustafsson, "Surpassing the lateral resolution limit by a factor of two using structured illumination microscopy," Journal of microscopy 198, 82-87 (2000). 44. M. G. Gustafsson, "Nonlinear structured-illumination