sbi icon indicating copy to clipboard operation
sbi copied to clipboard

Continuous normalizing flows via flow matching

Open janfb opened this issue 1 year ago • 3 comments

It would be great to have FMPE implemented in sbi.

There are implementations available that we could depend on, e.g.,

  • dingo: https://github.com/dingo-gw/dingo/blob/FMPE/dingo/core/posterior_models/flow_matching.py
  • lampe via zuko: https://lampe.readthedocs.io/en/stable/tutorials/fmpe.html

To Do:

  • check how we could integrate existing implementations into sbi
  • or implement from scratch
  • add CNF via flow matching as density estimator
  • add FMPE as inference method

janfb avatar Feb 28 '24 10:02 janfb

I'd like to have a look at this.

turnmanh avatar Mar 15 '24 14:03 turnmanh

I am also up for it, but happy to work on something else if it's better as a one person job.

bkmi avatar Mar 18 '24 09:03 bkmi

This is heavily related to #962! So it would probably be good to coordinate some aspects :)

jsvetter avatar Mar 18 '24 14:03 jsvetter