Ensemble Latent Space Roadmap for Improved Robustness in Visual Action Planning

Martina Lippi*, Michael C. Welle*, Andrea Gasparri, and Danica Kragic

Planning in learned latent spaces helps to decrease the dimensionality of raw observations. In this work, we propose to leverage the ensemble paradigm to enhance the robustness of latent planning systems. We rely on our Latent Space Roadmap (LSR) framework, which builds a graph in a learned structured latent space to perform planning. Given multiple LSR framework instances, that differ either on their latent spaces or on the parameters for constructing the graph, we use the action information as well as the embedded nodes of the produced plans to define similarity measures. These are then utilized to select the most promising plans. We validate the performance of our Ensemble LSR (ENS-LSR) on simulated box stacking and grape harvesting tasks as well as on a real-world robotic T-shirt folding experiment.

*Contributed equally and listed in alphabetical order

Download Preprint

Method

Our Method:

Folding execution videos

Fold 1

Start state

Goal state

Execution videos fold 1

Fold 2

Start state

Goal state

Execution videos fold 2

Fold 3

Start state

Goal state

Execution videos fold 3

Fold 4

Start state

Goal state

Execution videos fold 4

Fold 5

Start state

Goal state

Execution videos fold 5

HTM examples

Example of erroneous plans obtained with method from the paper Liu, Kara, et al. "Hallucinative topological memory for zero-shot visual planning." International Conference on Machine Learning, 2020

Check the examples

Code Repository

Available after paper acceptance

Datasets

Available after paper acceptance

Contact

  • Martina Lippi; lippi(at)kth.se, martina.lippi(at)uniroma3.it; KTH Royal Institute of Technology, Sweden and Roma Tre University, Italy
  • Michael C. Welle; mwelle(at)kth.se; KTH Royal Institute of Technology, Sweden
  • Andrea Gasparri; andrea.gasparri(at)uniroma3.it; Roma Tre University, Italy
  • Danica Kragic; dani(at)kth.se; KTH Royal Institute of Technology, Sweden