Personalized Route Recommendation with Neural Network Enhanced A* Search Algorithm
Jingyuan Wang,Ning Wu,Wayne Xin Zhao,Xin Zhao
DOI: https://doi.org/10.1109/tkde.2021.3068479
IF: 9.235
2021-01-01
IEEE Transactions on Knowledge and Data Engineering
Abstract:In this work, we study an important task in location-based services, namely Personalized Route Recommendation (PRR). Given a road network, the PRR task aims to generate user-specific route suggestions for replying to users' route queries. A classic approach is to adapt search algorithms to construct pathfinding-like solutions. These methods typically focus on reducing search space with suitable heuristic strategies. For these search algorithms, heuristic strategies are often handcrafted, which are not flexible to work in complicated task settings. In addition, it is difficult to utilize useful context information in the search procedure. To develop a more principled solution to the PRR task, we propose to improve search algorithms with neural networks for solving the PRR task based on the widely used <span class="mjpage"><svg xmlns:xlink="http://www.w3.org/1999/xlink" width="2.797ex" height="2.176ex" style="vertical-align: -0.338ex;" viewBox="0 -791.3 1204.4 936.9" role="img" focusable="false" xmlns="http://www.w3.org/2000/svg"><g stroke="currentColor" fill="currentColor" stroke-width="0" transform="matrix(1 0 0 -1 0 0)"> <use xlink:href="#MJMATHI-41" x="0" y="0"></use> <use transform="scale(0.707)" xlink:href="#MJMAIN-2217" x="1061" y="513"></use></g></svg></span>A* algorithm. The main idea of our solution is to automatically learn the cost functions in <span class="mjpage"><svg xmlns:xlink="http://www.w3.org/1999/xlink" width="2.797ex" height="2.176ex" style="vertical-align: -0.338ex;" viewBox="0 -791.3 1204.4 936.9" role="img" focusable="false" xmlns="http://www.w3.org/2000/svg"><g stroke="currentColor" fill="currentColor" stroke-width="0" transform="matrix(1 0 0 -1 0 0)"> <use xlink:href="#MJMATHI-41" x="0" y="0"></use> <use transform="scale(0.707)" xlink:href="#MJMAIN-2217" x="1061" y="513"></use></g></svg></span>A* algorithms, which is the key of heuristic search algorithms. Our model consists of two main 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 an estimation network for predicting the cost from a candidate location to the destination. For capturing structural characteristics, the-estimation network is built on top of position-aware graph attention networks. The two components are integrated in a principled way for deriving a more accurate cost of a candidate location for the <span class="mjpage"><svg xmlns:xlink="http://www.w3.org/1999/xlink" width="2.797ex" height="2.176ex" style="vertical-align: -0.338ex;" viewBox="0 -791.3 1204.4 936.9" role="img" focusable="false" xmlns="http://www.w3.org/2000/svg"><g stroke="currentColor" fill="currentColor" stroke-width="0" transform="matrix(1 0 0 -1 0 0)"> <use xlink:href="#MJMATHI-41" x="0" y="0"></use> <use transform="scale(0.707)" xlink:href="#MJMAIN-2217" x="1061" y="513"></use></g></svg></span>A* algorithm. Extensive experiment results on three real-world datasets have shown the effectiveness and robustness of the proposed model.<svg xmlns="http://www.w3.org/2000/svg" style="display: none;"><defs id="MathJax_SVG_glyphs"><path stroke-width="1" id="MJMATHI-41" d="M208 74Q208 50 254 46Q272 46 272 35Q272 34 270 22Q267 8 264 4T251 0Q249 0 239 0T205 1T141 2Q70 2 50 0H42Q35 7 35 11Q37 38 48 46H62Q132 49 164 96Q170 102 345 401T523 704Q530 716 547 716H555H572Q578 707 578 706L606 383Q634 60 636 57Q641 46 701 46Q726 46 726 36Q726 34 723 22Q720 7 718 4T704 0Q701 0 690 0T651 1T578 2Q484 2 455 0H443Q437 6 437 9T439 27Q443 40 445 43L449 46H469Q523 49 533 63L521 213H283L249 155Q208 86 208 74ZM516 260Q516 271 504 416T490 562L463 519Q447 492 400 412L310 260L413 259Q516 259 516 260Z"></path><path stroke-width="1" id="MJMAIN-2217" d="M229 286Q216 420 216 436Q216 454 240 464Q241 464 245 464T251 465Q263 464 273 456T283 436Q283 419 277 356T270 286L328 328Q384 369 389 372T399 375Q412 375 423 365T435 338Q435 325 425 315Q420 312 357 282T289 250L355 219L425 184Q434 175 434 161Q434 146 425 136T401 125Q393 125 383 131T328 171L270 213Q283 79 283 63Q283 53 276 44T250 35Q231 35 224 44T216 63Q216 80 222 143T229 213L171 171Q115 130 110 127Q106 124 100 124Q87 124 76 134T64 161Q64 166 64 169T67 175T72 181T81 188T94 195T113 204T138 215T170 230T210 250L74 315Q65 324 65 338Q65 353 74 363T98 374Q106 374 116 368T171 328L229 286Z"></path></defs></svg>
computer science, information systems, artificial intelligence,engineering, electrical & electronic