Skip to main content

Accelerating gravitational wave template generation with machine learning.

Project description

CI Pipeline for mlgw_bns Documentation Status PyPI version Code style: black

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 improves 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, consisting of EOB waveforms
  • decompose the Fourier transforms of the waveforms into phase and amplitude
  • downsample the dataset to a few thousand points
  • compute the residuals of the EOB waveforms from PN ones
  • 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

After this, the model can reconstruct a waveform within its parameter space, resampled at arbitrary points in frequency space.

In several of the training 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 hyperparameters for the neural network are optimized, according to both the time taken for the training and the estimated reconstruction error, also varying the number of training waveforms available.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

mlgw_bns-0.9.2.tar.gz (1.4 MB view details)

Uploaded Source

Built Distribution

mlgw_bns-0.9.2-py3-none-any.whl (1.4 MB view details)

Uploaded Python 3

File details

Details for the file mlgw_bns-0.9.2.tar.gz.

File metadata

  • Download URL: mlgw_bns-0.9.2.tar.gz
  • Upload date:
  • Size: 1.4 MB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/1.1.13 CPython/3.9.9 Linux/5.13.0-30-generic

File hashes

Hashes for mlgw_bns-0.9.2.tar.gz
Algorithm Hash digest
SHA256 91abfd0d42b84b734545813784760bb0929d19105a98736d5ff8129c6e30cc06
MD5 558767bb6475074b0dad69b0bacc6ab2
BLAKE2b-256 4dcebaeb8db2354dc433d0fdec7d2bd2d3bf70a20d9732d94572098e2e54577b

See more details on using hashes here.

File details

Details for the file mlgw_bns-0.9.2-py3-none-any.whl.

File metadata

  • Download URL: mlgw_bns-0.9.2-py3-none-any.whl
  • Upload date:
  • Size: 1.4 MB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/1.1.13 CPython/3.9.9 Linux/5.13.0-30-generic

File hashes

Hashes for mlgw_bns-0.9.2-py3-none-any.whl
Algorithm Hash digest
SHA256 9cbeb3623991c4edd5b169fd0c5c9ad608a92a332d7bccd94b7164573ac0a6ec
MD5 1cce69f79619a1ba180ac4e488b44de1
BLAKE2b-256 7bc7ef866d557c48b17ea89c78014cb9b0a094d2f240a705e45450b5370a8029

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page