Tolerance-Aware Deep Optics

1Shanghai AI Laboratory, 2KAUST, 3NVIDIA, 4 The Chinese University of Hong Kong
Teaser Image

By employing tolerance-aware end-to-end optimization, our approach significantly enhances the tolerance robustness of deep optics.

Abstract

Deep optics has emerged as a promising approach by co-designing optical elements with deep learning algorithms. However, current research typically overlooks the analysis and optimization of manufacturing and assembly tolerances. This oversight creates a significant performance gap between designed and fabricated optical systems. To address this challenge, we present the first end-to-end tolerance-aware optimization framework that incorporates multiple tolerance types into the deep optics design pipeline. Our method combines physics-informed modelling with data-driven training to enhance optical design by accounting for and compensating for structural deviations in manufacturing and assembly. We validate our approach through computational imaging applications, demonstrating results in both simulations and real-world experiments. We further examine how our proposed solution improves the robustness of optical systems and vision algorithms against tolerances through qualitative and quantitative analyses.

Main idea

Teaser Image How we enhance the tolerance robustness of deep optics:
  • We first implement a tolerance sampler that can convert random tolerances into variations in optical parameters and spatial transformations. The tolerance sampler is capable of applying realistic tolerance perturbations to lens systems, thereby reducing the gap between design and manufacture.
  • We design two loss functions aimed at improving the optical tolerance robustness: PSF Loss and Spot Loss. The combined use of these two losses significantly enhances the stability of end-to-end deep optical training.
Ablation Image
Top: Demonstration of tolerance effects; Bottom: Ray tracing with tolerance results comparison against ZEMAX.

Results Comparison

Preview 1
Preview 2
Preview 3
Preview 4
Preview 5
Preview 6
Preview 7
Preview 8
Preview 9
Preview 10
Preview 11
Preview 12
Preview 13
w/o TOLR
w/ TOLR
Clear Image
Blur Image

Drag the slider to compare results with and without our TOLR. Click on the previews above to view different cases.

Numerical Results

Ablation Image
After tolerance-aware optimization, Deep Optics demonstrates enhanced robustness to the actual tolerances, significantly improving its manufacturing yield, thereby narrowing the gap between design and manufacture.

Ablation Results

Ablation Image
After tolerance-aware optimization, the optical component of Deep Optics exhibits minimal changes in the PSF when subjected to random tolerance disturbances, allowing the decoder to handle the relatively small variations in imaging properties more easily.
Ablation Image
After tolerance-aware optimization, the decoder in Deep Optics demonstrates enhanced capability to handle variations in optical imaging properties induced by random tolerances.
Ablation Image
To ensure a more stable training process, it is essential to sample a statistically meaningful number of tolerance disturbance patterns.

BibTeX

@article{dai2025toleranceawaredeepoptics,
      title={Tolerance-Aware Deep Optics}, 
      author={Jun Dai and Liqun Chen and Xinge Yang and Yuyao Hu and Jinwei Gu and Tianfan Xue},
      journal={arXiv preprint arXiv:2502.04719},
      year={2025}
}