DiffRayve: Differentiable Ray-Wave Method for Polarized and Unpolarized Diffractive-Refractive Optical Systems

Abstract

Current deep optics simulations struggle to balance high fidelity with computational efficiency, particularly for complex polarized and diffractive systems. To address this, we introduce a fully differentiable Shooting and Bouncing Ray (SBR) algorithm that simultaneously models geometric and physical optics fields. Unlike standard wave optics methods, our approach enables accurate, gradient-based optimization of polarized compound refractive-diffractive systems without the computational cost of full-wave simulations. We validate our method against analytical Fraunhofer diffraction patterns and state-of-the-art Fourier optics and wave optics methods. We showcase the practical utility of our differentiable engine through one application: designing a wide field-of-view achromatic lens with multiple Diffractive Optical Elements (DOEs).