The Implementation of a Pointer Network Model for Traveling Salesman Problem on a Xilinx PYNQ Board

被引:5
作者
Gu, Shenshen [1 ]
Hao, Tao [1 ]
Yang, Shaofu [2 ]
机构
[1] Shanghai Univ, Sch Mechatron Engn & Automat, Shanghai 200072, Peoples R China
[2] Southeast Univ, Sch Comp Sci & Engn, Nanjing 211189, Jiangsu, Peoples R China
来源
ADVANCES IN NEURAL NETWORKS - ISNN 2018 | 2018年 / 10878卷
基金
美国国家科学基金会;
关键词
Pointer networks; Traveling salesman problem; Theano; PYNQ; FPGA;
D O I
10.1007/978-3-319-92537-0_16
中图分类号
TP301 [理论、方法];
学科分类号
081202 ;
摘要
In this paper, a pointer network model for traveling salesman problem (TSP) was implemented on a Xilinx PYNQ board which supports Python and Jupyter notebook and is equipped with ZYNQ SOC. We implement a pointer network model for solving TSP with Python and Theano firstly, then train the model on a GPU platform, and eventually deploy the model on a PYNQ board. Unlike traditional neural network implementation, hardware libraries on PYNQ (Overlays) are used to accelerate the pointer network model application. The experimental results show that the pointer network model for TSP can be deployed on the embedded system successfully and achieve good performance.
引用
收藏
页码:130 / 138
页数:9
相关论文
共 13 条
[1]  
[Anonymous], 2017, IMPLEMENTATION DEEP
[2]  
Bahdanau D, 2016, Arxiv, DOI arXiv:1409.0473
[3]  
Chang A.X. M., 2016, Comput. Sci.
[4]   LSTM: A Search Space Odyssey [J].
Greff, Klaus ;
Srivastava, Rupesh K. ;
Koutnik, Jan ;
Steunebrink, Bas R. ;
Schmidhuber, Juergen .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2017, 28 (10) :2222-2232
[5]  
Guan YJ, 2017, ASIA S PACIF DES AUT, P629, DOI 10.1109/ASPDAC.2017.7858394
[6]  
Hochreiter S, 1997, NEURAL COMPUT, V9, P1735, DOI [10.1162/neco.1997.9.8.1735, 10.1007/978-3-642-24797-2, 10.1162/neco.1997.9.1.1]
[7]   Deep learning [J].
LeCun, Yann ;
Bengio, Yoshua ;
Hinton, Geoffrey .
NATURE, 2015, 521 (7553) :436-444
[8]   Deep learning for detecting robotic grasps [J].
Lenz, Ian ;
Lee, Honglak ;
Saxena, Ashutosh .
INTERNATIONAL JOURNAL OF ROBOTICS RESEARCH, 2015, 34 (4-5) :705-724
[9]   FPGA Acceleration of Recurrent Neural Network based Language Model [J].
Li, Sicheng ;
Wu, Chunpeng ;
Li, Hai ;
Li, Boxun ;
Wang, Yu ;
Qiu, Qinru .
2015 IEEE 23RD ANNUAL INTERNATIONAL SYMPOSIUM ON FIELD-PROGRAMMABLE CUSTOM COMPUTING MACHINES (FCCM), 2015, :111-118
[10]  
Milan A, 2017, AAAI CONF ARTIF INTE, P1453