PEPL: A Two-Stage Method for Submillimeter Insertion Tasks with Pose Estimation and Primitives Learning

Yuxuan Bai,Mingshuai Dong,Shimin Wei,Jian Li
DOI: https://doi.org/10.1109/iceemt63201.2024.10692425
2024-01-01
Abstract:Sub-millimeter insertion tasks are indispensable tasks in intelligent manufacturing and aerospace unmanned assembly. To date, traditional methods based on visual positioning and force perception rely on establishing contact state models, which exhibit poor robustness. Meanwhile, reinforcement learning-based methods need a larger exploration space, leading to low learning efficiency. This paper proposed PEPL, a two-stage insertion method combining visual positioning and Primitives Learning. Firstly, a visual positioning method based on Siamese networks is used for pose estimation, narrowing down the reinforcement learning workspace. Next, leveraging the pose estimation results as prior knowledge, the reinforcement learning insertion algorithm based on primitive learning completes the final insertion process. Additionally, the design of a hybrid action space enhances the algorithm’s learning efficiency and facilitates its transferability to other objects. Extensive experiments have demonstrated the effectiveness of the two-stage paradigm. Furthermore, experiments conducted on real robots have confirmed that PEPL can achieve a high success rate in completing sub-millimeter insertion tasks. In summary, the proposed PEPL meets the requirements for sub-millimeter precision assembly and ensures the algorithm’s efficiency and robustness.
What problem does this paper attempt to address?