We present Neural A*, a novel data-driven search method for path planning problems. Despite the recent increasing attention to data-driven path planning, a machine learning approach to search-based planning is still challenging due to the discrete nature of search algorithms. In this work, we reformulate a canonical A* search algorithm to be differentiable and couple it with a convolutional encoder to form an end-to-end trainable neural network planner. Neural A* solves a path planning problem by encoding a problem instance to a guidance map and then performing the differentiable A* search with the guidance map. By learning to match the search results with ground-truth paths provided by experts, Neural A* can produce a path consistent with the ground truth accurately and efficiently. Our extensive experiments confirmed that Neural A* outperformed state-of-the-art data-driven planners in terms of the search optimality and efficiency trade-off, and furthermore, successfully predicted realistic human trajectories by directly performing search-based planning on natural image inputs.
翻译:我们提出“神经A* ”,这是用于路径规划问题的一种新颖的数据驱动搜索方法。尽管最近人们日益关注数据驱动路径规划,但是,由于搜索算法的离散性质,基于搜索的规划的机械学习方法仍然具有挑战性。在这项工作中,我们重塑了一种卡通A* 搜索算法,使之具有差异性,并将其与进化编码器相匹配,以形成一个端到端的可训练神经网络规划师。神经A* 解决了路径规划问题,将一个问题实例编码在指导地图上,然后用指导地图进行不同的 A* 搜索。通过学习将搜索结果与专家提供的地面路径相匹配,神经A* 能够准确和高效地产生一条与地面真相相符的道路。我们的广泛实验证实,神经A* 在搜索的最佳性和效率交易方面超越了由数据驱动的状态规划师,此外,通过直接对自然图像输入进行搜索规划,成功地预测了现实的人类轨迹。