Dynamic gravitational field dataset - Latent Field Discovery in Interacting Dynamical Systems with Neural Fields

This repository contains the "Dynamic gravitational field" dataset from the paper

Latent Field Discovery in Interacting Dynamical Systems with Neural Fields
Miltiadis Kofinas, Erik J Bekkers, Naveen Shankar Nagaraja, Efstratios Gavves
NeurIPS 2023
https://arxiv.org/abs/2310.20679
https://github.com/mkofinas/aether

It contains simulations of trajectories of 5 charged particles in 3 dimensions, interacting via gravitational forces.

Particles move under the influence of 1 immovable and unknown source, which is different in each simulation. The source has a mass of 10, while each particle has a mass of 1.

There are 50,000 simulations for training, 10,000 for validation, and 10,000 for testing. Simulations last for 49 timesteps.

The features comprise positions and velocities of particles. The dataset also contains the positions of the field sources, meant to be used for visualization.