Skip to content

Conversation

@jfowkes
Copy link
Collaborator

@jfowkes jfowkes commented Nov 14, 2024

This PR adds the ML with separate gradient steps for probe and object reconstruction engine by @Letizia97 to custom.

Testing has shown that this gives marginally better reconstruction errors than ML at the expense of more computation (as the linesearch step size computation is performed for both probe and object necessitating an extra loop over the data).

Testing comparison on the moonflower example:

ML: Iteration # 300/300 ::  Photons 6.86e+01
MLSeparateGrads: Iteration # 300/300 :: Photons 2.03e+01

Testing on the small nanogold example:

ML: Iteration # 100/100 :: Photons 2.21e+02
MLSeparateGrads: Iteration # 100/100 :: Photons 1.71e+02

This is WIP, some things that need doing:

  • Add Poisson and Euclid noise models
  • Implement all poly_line_coeffs
  • Add GPU support (requires new kernel function)

@jfowkes jfowkes changed the base branch from master to dev January 6, 2025 09:36
@jfowkes jfowkes force-pushed the ML-separate-grads branch from d6df9f1 to 60c7e60 Compare January 6, 2025 09:42
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants