基于物理的可微光学湍流模拟器,支持端到端系统优化。
TurPy: a physics-based and differentiable optical turbulence simulator for algorithmic development and system optimization
- 采用参数化相位屏生成与自回归时间演化,实现高保真湍流建模。
- 在弱到强湍流下对齐理论模型,波束扩展误差仅2%,精度达98%。
- 适合光学算法开发、神经网络训练及湍流环境系统设计者使用。
面向自由空间光学系统的算法开发与系统优化,需精确模拟湍流引起的波前畸变并支持梯度优化。本文提出TurPy,一个基于GPU加速、完全可微的波前光学湍流仿真框架。该框架整合了次谐波相位屏生成、自回归时间演化及自动屏幕布置机制,在满足傅里叶混叠约束与弱湍流近似前提下,统一实现高保真仿真。由于相位屏生成由介质特异性功率谱密度参数化,该框架可无缝扩展至大气、海洋与生物传播环境,仅需折射率结构常数与功率谱密度即可输入。通过匹配二阶高斯光束展宽与四阶平面波闪烁,验证结果在弱至强湍流范围内与闭合形式模型一致,准确率达98%。为展示其梯度优化能力,我们使用双掩码双域架构的衍射深度神经网络(D2NN)在弱湍流路径中恢复高斯光束,相比未补偿接收机,闪烁抑制超过58%。TurPy已开源,可用于合成数据生成、湍流感知算法开发及湍流环境下光学平台的端到端设计。
原文摘要 · Abstract (English)
Developing optical systems for free-space applications requires simulation tools that accurately capture turbulence-induced wavefront distortions and support gradient-based optimization. Here we introduce TurPy, a GPU-accelerated, fully differentiable wave optics turbulence simulator to bridge high fidelity simulation with end-to-end optical system design. TurPy incorporates subharmonic phase screen generation, autoregressive temporal evolution, and an automated screen placement routine balancing Fourier aliasing constraints and weak-turbulence approximations into a unified, user-ready framework. Because TurPy's phase screen generation is parameterized through a media-specific power spectral density, the framework extends to atmospheric, oceanic, and biological propagation environments with minimal modification. We validate TurPy against established atmospheric turbulence theory by matching 2nd order Gaussian beam broadening and 4th order plane wave scintillation to closed-form models with 98% accuracy across weak to strong turbulence regimes, requiring only the medium's refractive index structure constant and power spectral density as inputs. To demonstrate TurPy as a gradient-based training platform, we optimize a dual-domain diffractive deep neural network (D2NN) in a two-mask dual-domain architecture to recover a Gaussian beam from a weakly turbulent path and achieving over 58% reduction in scintillation relative to an uncompensated receiver in simulation. TurPy is released as an open-source package to support synthetic data generation, turbulence-informed algorithm development, and the end-to-end design of optical platforms operating in turbulent environments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。