herikem / CVAE-for-Gravitational-Waves-from-EMRI

Wei Liang and fellow researchers exploring CVAE algorithms
GNU General Public License v3.0
3 stars 0 forks source link

This is an attentative program to perform the Bayesian parameter analysis [1] based on the CVAE network [2]. We are developing this project with the understanding that CVAE is rapidly evolving and being utilized in theoretical physics. We believe that another public code would be a valuable addition to this collaborative community. Anyone is welcome to monitor and discuss with us about the project as it progresses.

The code is distributed under the GNU General Public License and the related theoretical studies will be published under the Creative Commons Attribution license. The database link can be requested from me, Hong Guo, or Wei-Liang Qian.

[1] Bayesian parameter estimation using conditional variational autoencoders for gravitational-wave astronomy, by Hunter Gabbard et al., arXiv:1909.06296.

[2] Learning Structured Output Representation using Deep Conditional Generative Models, by Kihyuk Sohn et al., NIPS15, 3483, 2015.

Update:

20240520 The model is almost functional. At this point, there are still some problems with batch_size and the loss is evaluted to NAN.