LouisDesdoigts / dLux

Differentiable optical models as parameterised neural networks in Jax using Zodiax
https://louisdesdoigts.github.io/dLux/
BSD 3-Clause "New" or "Revised" License
52 stars 6 forks source link

Improve FP-precision by setting wavefront amplitude as opposed to scaling output PSFs? #248

Open LouisDesdoigts opened 1 year ago

LouisDesdoigts commented 1 year ago

If wavefront amplitudes are set upstream before the propagation, then the necessarily small values output by unitary PSFs can be avoided. This should help alleviate FP precision issues, but is likely difficult to implement since the Wavefront can be normalized at any point during the transformations. Ideally this would be done before each propagation.