sceki / dsgp4

Differentiable SGP4

$\partial\textrm{SGP4}$

Differentiable SGP4.

Differentiable SGP4
Explore the docs »

Report bug · Request feature

Info

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.

Goals

• 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.)

How to cite

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}
}

Installation

Via pip:

pip install dsgp4

Via conda:

conda install conda-forge::dsgp4

Local installation:

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

Documentation and examples

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

Authors:

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

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

Acknowledgements:

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.

License:

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

Contact:

• [email protected]