Skip to content
master
Go to file
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
Jun 23, 2020
Jun 23, 2020
Jun 23, 2020

README.md

Discovering Symbolic Models from Deep Learning with Inductive Biases

This repository is the official implementation of Discovering Symbolic Models from Deep Learning with Inductive Biases.

Miles Cranmer, Alvaro Sanchez-Gonzalez, Peter Battaglia, Rui Xu, Kyle Cranmer, David Spergel, Shirley Ho

Check out our Blog, Paper, Video, and Interactive Demo.

Requirements

For model:

Symbolic regression:

  • PySR, our new open-source Eureqa alternative

For simulations:

  • jax (simple N-body simulations)
  • quijote (Dark matter data; optional)
  • tqdm
  • matplotlib

Training

To train an example model from the paper, try out the demo.

Full model definitions are given in models.py. Data is generated from simulate.py.

Results

We train on simulations produced by the following equations: giving us time series:

We recorded performance for each model: and also measured how well each model's messages correlated with a linear combination of forces:

Finally, we trained on a dark matter simulation and extracted the following equations from the message function:

About

Code for "Discovering Symbolic Models from Deep Learning with Inductive Biases"

Resources

License

Releases

No releases published

Packages

No packages published
You can’t perform that action at this time.