三维TSP问题

[wangyan-hlab]

@neo-pan 你好，

如果我想将这个算法应用于三维TSP问题（顶点位于三维空间中），需要进行哪些修改呢？

谢谢！

[neo-pan]

你好，

要应用于三维 TSP，需要修改的主要是两个部分：

1. State 和 Environment 相关的修改
   在 state 和 environment 中，有许多与距离计算和 action 执行相关的代码，这些都与 TSP 问题的维度相关，因此需要做出相应的修改。

   • 参考代码段：代码链接
   • 另一个相关代码段：代码链接

2. 模型的 Encoder 修改
   当前使用的 encoder 参考了 3D 点云的处理方式来对大规模 TSP 问题进行编码。对于三维 TSP 问题，可以考虑直接使用针对 3D 点云设计的 encoder 结构。例如，可以参考以下文献中的 PointPillars 结构来进行修改：

   • 文献：Lang, A. H.; Vora, S.; Caesar, H.; Zhou, L.; Yang, J.; and Beijbom, O. 2019. PointPillars: Fast Encoders for Object Detection From Point Clouds. In 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 12689–12697.

   • 相关代码段：代码链接