Skip to content

PINNAir: Physics-Informed Neural Network for Multirotor Slung Load Systems Modeling

License

Notifications You must be signed in to change notification settings

GilSerrano/pinn-air

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Physics-Informed Neural Network for Multirotor Slung Load Systems Modeling

In this work, we explore the use of physics-informed neural networks to learn an end-to-end model of the multirotor-slung-load system and, at a given time, estimate a sequence of the future system states. To this end, a sequence of observations of states and control inputs of the system is fed into a set of fully connected layers, which encodes the input data into a more meaningful representation. An LSTM encoder-decoder with an attention mechanism is then used to capture the dynamics of the system and another set of fully connected layers is employed to represent the decoded sequence in the original state space. To guarantee the cohesiveness between the multiple predicted states of the system and restrict the space of admissible predictions, we propose the use of a physics-based loss term in the loss function, which includes a discretized physical model derived from first principles, together with slack variables. To train the model, a dataset using a real-world quadrotor carrying a slung load was curated and is made available. Prediction results and ablations studies are presented and corroborate the feasibility of the approach. The proposed method outperforms both the first principles physical model and a comparable neural network model trained without the physics regularization proposed.

PINN-Air and the associated dataset is released under the BSD-3 License.

Citation

If you use PINN-Air or the associated dataset in your research, please cite:

@misc{2024pinnair,
      title={Physics-Informed Neural Network for Multirotor Slung Load Systems Modeling}, 
      author={Gil Serrano and Marcelo Jacinto and Jose Ribeiro-Gomes and Joao Pinto and Bruno J. Guerreiro and Alexandre Bernardino and Rita Cunha},
      year={2024},
      eprint={2405.09428},
      archivePrefix={arXiv},
      primaryClass={cs.RO}
}

Training the Model and Generating Statistics

./train_model.sh

Launching Tensorboard

python -m tensorboard.main --logdir "trained_models"

Then navigate to http://localhost:6006/ in your browser.

Developer Team

This work was developed by the following team of researchers:

Supervised by:

  • Prof. Rita Cunha
  • Prof. Bruno Guerreiro
  • Prof. Alexandre Bernardino

The authors gratefully acknowledge Chrysoula Zerva and André F. T. Martins for their suggestions to improve the quality of this work.

Project Sponsors

  • Institute for Systems and Robotics (ISR), a research unit of the Laboratory of Robotics and Engineering Systems (LARSyS)
  • Instituto Superior Técnico, Universidade de Lisboa

The work developed by Gil Serrano, Marcelo Jacinto, José Gomes and João Pinto was supported by Ph.D. grants funded by Fundação para a Ciência e Tecnologia (FCT).

About

PINNAir: Physics-Informed Neural Network for Multirotor Slung Load Systems Modeling

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 4

  •  
  •  
  •  
  •