Motion planning is integral to robotics applications such as autonomous driving, surgical robots, and industrial manipulators. Existing planning methods lack scalability to higher-dimensional spaces, while recent learning based planners have shown promise in accelerating sampling-based motion planners (SMP) but lack generalizability to out-of-distribution environments. To address this, we present a novel approach, Vector Quantized-Motion Planning Transformers (VQ-MPT) that overcomes the key generalization and scaling drawbacks of previous learning-based methods. VQ-MPT consists of two stages. Stage 1 is a Vector Quantized-Variational AutoEncoder model that learns to represent the planning space using a finite number of sampling distributions, and stage 2 is an Auto-Regressive model that constructs a sampling region for SMPs by selecting from the learned sampling distribution sets. By splitting large planning spaces into discrete sets and selectively choosing the sampling regions, our planner pairs well with out-of-the-box SMPs, generating near-optimal paths faster than without VQ-MPT's aid. It is generalizable in that it can be applied to systems of varying complexities, from 2D planar to 14D bi-manual robots with diverse environment representations, including costmaps and point clouds. Trained VQ-MPT models generalize to environments unseen during training and achieve higher success rates than previous methods.
翻译:运动规划是自动驾驶、手术机器人和工业机械臂等机器人应用的核心技术。现有规划方法在高维空间缺乏可扩展性,而近期基于学习的规划器虽能加速基于采样的运动规划器(SMP),但在分布外环境中泛化能力不足。为此,我们提出了一种新方法——向量量化运动规划Transformer(VQ-MPT),克服了先前基于学习方法的关键泛化与扩展缺陷。VQ-MPT包含两个阶段:第一阶段是向量量化变分自编码器模型,通过学习有限数量的采样分布来表征规划空间;第二阶段是自回归模型,通过从学习到的采样分布集合中选取区域来构建SMP的采样空间。通过将大规划空间分解为离散集合并选择性选择采样区域,我们的规划器能与现成的SMP高效配合,在无需VQ-MPT辅助时更快生成近优路径。该方法具有泛化性,可应用于从二维平面到具备成本地图与点云等多样环境表征的14维双臂机器人等不同复杂度系统。训练后的VQ-MPT模型能泛化至训练中未见的环境,并实现比先前方法更高的成功率。