Sample Efficient Grasp Learning Using Equivariant Models
Sample Efficient Grasp Learning Using Equivariant Models
复制标题
DOI:
10.15607/rss.2022.xviii.071
复制
发表时间:
2022-02
期刊:
影响因子:
--
通讯作者:
Xu Zhu;Dian Wang;Ondrej Biza;Guanang Su;R. Walters;Robert W. Platt
中科院分区:
文献类型:
--
作者:
Xu Zhu;Dian Wang;Ondrej Biza;Guanang Su;R. Walters;Robert W. Platt
In planar grasp detection, the goal is to learn a function from an image of a scene onto a set of feasible grasp poses in $\mathrm{SE}(2)$. In this paper, we recognize that the optimal grasp function is $\mathrm{SE}(2)$-equivariant and can be modeled using an equivariant convolutional neural network. As a result, we are able to significantly improve the sample efficiency of grasp learning, obtaining a good approximation of the grasp function after only 600 grasp attempts. This is few enough that we can learn to grasp completely on a physical robot in about 1.5 hours.