arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1907.08489cs.AIcs.CYcs.LG
收藏

基于神经网络的A* 搜索算法的个性化路径推荐

Empowering A* Search Algorithms with Neural Networks for Personalized Route Recommendation

Beihang University(北京航空航天大学);Renmin University of China(中国人民大学)

Jingyuan Wang, Ning Wu, Wayne Xin Zhao, Fanzhang Peng, Xin Lin

摘要:

个性化路线推荐(PRR)的目的是生成用户特定的路线建议,以响应用户的路线查询。早期的研究将PRR任务视为图上的寻路问题,并通过整合启发式策略来采用自适应搜索算法。虽然这些方法在一定程度上是有效的,但它们需要用数学方法来设置成本函数。此外,在搜索过程中很难利用有用的上下文信息。为了解决这些问题,我们建议使用神经网络自动学习成本函数的经典启发式算法,即A* 算法,PRR任务。我们的模型由两个部分组成。首先,我们采用基于注意力的递归神经网络(RNN),通过结合有用的上下文信息来建模从源到候选位置的成本。RNN组件不是学习单个成本值,而是能够学习用户移动状态的时变向量化表示。其次,我们建议使用价值网络来估计从候选位置到目的地的成本。为了捕捉结构特征,通过结合用户的移动状态和其他上下文信息,在改进的图注意力网络之上构建价值网络。这两个分量以原则性的方式被集成,用于导出候选位置的更准确的成本。在三个真实数据集上的实验结果表明了该模型的有效性和鲁棒性。

英文摘要:

Personalized Route Recommendation (PRR) aims to generate user-specific route suggestions in response to users' route queries. Early studies cast the PRR task as a pathfinding problem on graphs, and adopt adapted search algorithms by integrating heuristic strategies. Although these methods are effective to some extent, they require setting the cost functions with heuristics. In addition, it is difficult to utilize useful context information in the search procedure. To address these issues, we propose using neural networks to automatically learn the cost functions of a classic heuristic algorithm, namely A* algorithm, for the PRR task. Our model consists of two components. First, we employ attention-based Recurrent Neural Networks (RNN) to model the cost from the source to the candidate location by incorporating useful context information. Instead of learning a single cost value, the RNN component is able to learn a time-varying vectorized representation for the moving state of a user. Second, we propose to use a value network for estimating the cost from a candidate location to the destination. For capturing structural characteristics, the value network is built on top of improved graph attention networks by incorporating the moving state of a user and other context information. The two components are integrated in a principled way for deriving a more accurate cost of a candidate location. Extensive experiment results on three real-world datasets have shown the effectiveness and robustness of the proposed model.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑