Differentiable SGP4.

Differentiable SGP4
Explore the docs »

Report bug · Request feature

This repository contains the code discussed in this paper. $\partial \textrm{SGP4}$ is a differentiable version of SGP4 implemented using PyTorch. By making SGP4 differentiable, $\partial \textrm{SGP4}$ facilitates various space-related applications, including spacecraft orbit determination, covariance transformation, state transition matrix computation, and covariance propagation. Additionally, $\partial \textrm{SGP4}$'s PyTorch implementation allows for parallel orbital propagation across batches of Two-Line Element Sets (TLEs), leveraging the computational power of CPUs, GPUs, and advanced hardware for distributed prediction of satellite positions at future times. Furthermore, $\partial \textrm{SGP4}$'s differentiability enables integration with modern machine learning techniques. Thus, we propose a novel orbital propagation paradigm, $\textrm{ML}-\partial \textrm{SGP4}$, where neural networks are integrated into the orbital propagator. Through stochastic gradient descent, this combined model's inputs, outputs, and parameters can be iteratively refined, surpassing SGP4's precision while maintaining computational speed. This empowers satellite operators and researchers to train the model using high-precision simulated or observed data, advancing orbital prediction capabilities compared to the standard SGP4.

• Differentiable version of SGP4 (implemented in PyTorch)
• Hybrid SGP4 and machine learning propagation: input/output/parameters corrections of SGP4 from accurate simulated or observed data are learned
• Parallel TLE propagation
• Use of differentiable SGP4 on several spaceflight mechanics problems (state transition matrix computation, covariance transformation, and propagation, orbit determination, ML hybrid orbit propagation, etc.)

If you use dsgp4, we would be grateful if you could star the repository and/or cite our work. $\partial \textrm{SGP4}$ and its applications for ML hybrid propagation and more, can be found in our publication:

@article{acciarini2024closing,
  title={Closing the Gap Between SGP4 and High-Precision Propagation via Differentiable Programming},
  author={Acciarini, Giacomo and Baydin, At{\i}l{\i}m G{\"u}ne{\c{s}} and Izzo, Dario},
  journal={arXiv preprint arXiv:2402.04830},
  year={2024}
}

Via pip:

pip install dsgp4

Via conda:

conda install conda-forge::dsgp4

or mamba:

mamba install dsgp4

Local installation:

git clone https://github.com/esa/dSGP4.git
cd dSGP4
pip install -e .

To get started, follow the examples in the documentation. You will find tutorials with basic and more advanced functionalities and applications.

• Giacomo Acciarini
• Atılım Güneş Baydin
• Dario Izzo

The project originated after the work of the authors at the University of Oxford AI4Science Lab.

We would like to thank Dr. T.S. Kelso for his support and useful pointers on how to correctly validate the code with respect to the official release by Space-Track.

$\partial\textrm{SGP4}$ is distributed under the GNU General Public License version 3. Get in touch with the authors for other licensing options.

• [email protected]