Q-attention: Enabling Efficient Learning for
Vision-based Robotic Manipulation

Stephen James Andrew J. Davison

Dyson Robotics Lab, Imperial College London

Code: https://github.com/stepjam/ARM

Trained Policies (with Q-attention visualised)

Videos show the Q-attention "heatmap" along with the chosen crop.

Keyframe Discovery

Visualisation of the keyframe discovery for 4 tasks.

Example Demonstrations