Skip to content

Bayesian Optimization for MPPI Control of Robot Arm Planar Pushing

License

Notifications You must be signed in to change notification settings

caokaifa/bayesian-opt-gpytorch

 
 

Repository files navigation

Bayesian Optimization for MPPI Planar Pushing

In this project, we implement the Bayesian Optimization Algorithm (BOA) for autotuning the hyperparameters of Model Predictive Path Integral (MPPI) control to solve a planar box pushing task with non-trivial obstacles.

Quick Start

  1. Install all necessary dependencies using bash install.sh.
  2. Run our demo using python demo.py. Instructions and results will be printed in the terminal and a live pushing demo should show up in pybullet GUI.

API for Easy BOA Application

This implementation provides an easy-to-use Bayesian optimizer with full GPU pipeline support. The optimization results are at least comparable to the most popular BOA implementation in python.

Check here for more details.

Dependencies

Reference

BOA

f: objective function
X: support of variables
S: acquisition function
M: GP model
D: dataset
x: variable
y: observation

MPPI

About

Bayesian Optimization for MPPI Control of Robot Arm Planar Pushing

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%