End-to-End Hybrid Refractive-Diffractive Lens Design with Differentiable Ray-Wave Model

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Yang, Xinge, Souza, Matheus, Wang, Kunyi, Chakravarthula, Praneeth, Fu, Qiang, Heidrich, Wolfgang
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913812589838336
author Yang, Xinge
Souza, Matheus
Wang, Kunyi
Chakravarthula, Praneeth
Fu, Qiang
Heidrich, Wolfgang
author_facet Yang, Xinge
Souza, Matheus
Wang, Kunyi
Chakravarthula, Praneeth
Fu, Qiang
Heidrich, Wolfgang
contents Hybrid refractive-diffractive lenses combine the light efficiency of refractive lenses with the information encoding power of diffractive optical elements (DOE), showing great potential as the next generation of imaging systems. However, accurately simulating such hybrid designs is generally difficult, and in particular, there are no existing differentiable image formation models for hybrid lenses with sufficient accuracy. In this work, we propose a new hybrid ray-tracing and wave-propagation (ray-wave) model for accurate simulation of both optical aberrations and diffractive phase modulation, where the DOE is placed between the last refractive surface and the image sensor, i.e. away from the Fourier plane that is often used as a DOE position. The proposed ray-wave model is fully differentiable, enabling gradient back-propagation for end-to-end co-design of refractive-diffractive lens optimization and the image reconstruction network. We validate the accuracy of the proposed model by comparing the simulated point spread functions (PSFs) with theoretical results, as well as simulation experiments that show our model to be more accurate than solutions implemented in commercial software packages like Zemax. We demonstrate the effectiveness of the proposed model through real-world experiments and show significant improvements in both aberration correction and extended depth-of-field (EDoF) imaging. We believe the proposed model will motivate further investigation into a wide range of applications in computational imaging, computational photography, and advanced optical design. Code will be released upon publication.
format Preprint
id arxiv_https___arxiv_org_abs_2406_00834
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle End-to-End Hybrid Refractive-Diffractive Lens Design with Differentiable Ray-Wave Model
Yang, Xinge
Souza, Matheus
Wang, Kunyi
Chakravarthula, Praneeth
Fu, Qiang
Heidrich, Wolfgang
Graphics
Computer Vision and Pattern Recognition
Optics
Hybrid refractive-diffractive lenses combine the light efficiency of refractive lenses with the information encoding power of diffractive optical elements (DOE), showing great potential as the next generation of imaging systems. However, accurately simulating such hybrid designs is generally difficult, and in particular, there are no existing differentiable image formation models for hybrid lenses with sufficient accuracy. In this work, we propose a new hybrid ray-tracing and wave-propagation (ray-wave) model for accurate simulation of both optical aberrations and diffractive phase modulation, where the DOE is placed between the last refractive surface and the image sensor, i.e. away from the Fourier plane that is often used as a DOE position. The proposed ray-wave model is fully differentiable, enabling gradient back-propagation for end-to-end co-design of refractive-diffractive lens optimization and the image reconstruction network. We validate the accuracy of the proposed model by comparing the simulated point spread functions (PSFs) with theoretical results, as well as simulation experiments that show our model to be more accurate than solutions implemented in commercial software packages like Zemax. We demonstrate the effectiveness of the proposed model through real-world experiments and show significant improvements in both aberration correction and extended depth-of-field (EDoF) imaging. We believe the proposed model will motivate further investigation into a wide range of applications in computational imaging, computational photography, and advanced optical design. Code will be released upon publication.
title End-to-End Hybrid Refractive-Diffractive Lens Design with Differentiable Ray-Wave Model
topic Graphics
Computer Vision and Pattern Recognition
Optics
url https://arxiv.org/abs/2406.00834