search

probabilists / lampe

Likelihood-free AMortized Posterior Estimation with PyTorch
https://lampe.readthedocs.io
MIT License
117 stars 11 forks source link

Implement differentiable coverage probability regularizer #14

Closed macio232 closed 1 month ago

macio232 commented 8 months ago

Description

The method described in Calibrating Neural Simulation-Based Inference with Differentiable Coverage Probability paper.

Implementation

New loss class in the inference module, similar to the NRELoss ->BNRELoss enhancement.

Two separate classes, one for NRE, and one for NPE.

Alternatives

I did not consider alternatives.

francois-rozet commented 8 months ago

Hello @macio232 thanks for the feature request! I'll take a look at the PR soon!