Machine Learning for Gravitational Waves from Binary Neutron Star mergers
This package's purpose is to speed up the generation of template gravitational waveforms for binary neutron star mergers by training a machine learning model on a dataset of waveforms generated with some physically-motivated surrogate.
It is able to reconstruct them with mismatches lower than 1/10000, with as little as 1000 training waveforms; the accuracy then steadily drops as more training waveforms are used.
Currently, the only model used for training is TEOBResumS,
but it is planned to introduce the possibility to use others.
The documentation can be found here.
Installation
To install the package, use
pip install mlgw-bns
For more details see the documentation.
Inner workings
The main steps taken by mlgw_bns to train on a dataset are as follows:
- generate the dataset
- decompose the Fourier transforms of the waveforms into phase and amplitude
- downsample the dataset to a few thousand points
- apply a PCA to reduce the dimensionality to a few tens of real numbers
- train a neural network on the relation between the waveform parameters and the PCA components
In several of these steps data-driven optimizations are performed:
- the points at which the waveforms are downsampled are not uniformly chosen: instead, a greedy downsampling algorithm determines them
- the PCA is trained on a separate downsampled dataset, which is then thrown out
- the hyperparameters for the neural network are optimized according to both the time taken for the training and the estimated reconstruction error
1 maintainer
Release files for mlgw_bns 0.6.1
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| mlgw_bns-0.6.1.tar.gz | 3.6 MB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| mlgw_bns-0.6.1-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 7.1 MB
Release files / mlgw_bns-0.6.1.tar.gz
| Download URL | mlgw_bns-0.6.1.tar.gz |
|---|---|
| Size | 3.6 MB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
6ce4799635aaa7f4ebd69992303722933cabeef3892e27878c442be2500aaba8
|
|
BLAKE2b-256 checksum How to use checksums |
8e3fda9e54030e2a6065c089fa7503a25995409ed2be04f60fc1a0710b84f77b
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
poetry/1.1.12 CPython/3.9.9 Linux/5.13.0-28-generic
|
Release files / mlgw_bns-0.6.1-py3-none-any.whl
| Download URL | mlgw_bns-0.6.1-py3-none-any.whl |
|---|---|
| Size | 3.6 MB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
8cad098d9400fbac247d29b35c625f22d544e08373fecb55aca8b0b33c8500f6
|
|
BLAKE2b-256 checksum How to use checksums |
1c1e18a90775b64d5b7020fd38a7f86ffb06c6dc1bab2e72b82ef61b3bce4115
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
poetry/1.1.12 CPython/3.9.9 Linux/5.13.0-28-generic
|
