May, 2023
一种轻量级的CNN-Transformer模型用于学习旅行商问题
A Lightweight CNN-Transformer Model for Learning Traveling Salesman
Problems
TL;DR本研究提出了一种基于CNN嵌入层和局部自注意力的轻量级CNN-Transformer模型,相对于标准Transformer模型,该模型能够更好地从输入数据中学习空间特征,同时通过局部自注意力减少全连接注意模型中的冗余性。实验结果表明,我们提出的模型在解决TSP问题的质量、GPU内存使用和推理时间等方面优于其他最先进的Transformer-based模型。