中文

GAT-Steiner:基于图神经网络的矩形斯坦纳最小树预测

机器学习 2024-07-02 v1

摘要

矩形斯坦纳最小树 (RSMT) 问题是 VLSI 布置与路由中的一个基本问题,已知为 NP-hard 类型。传统的 RSMT 算法在寻找斯坦纳点以减少总线长方面耗费大量时间,或使用启发式方法进行近似求解,产生次优结果。我们表明,图神经网络 (GNN) 可用于以高精度预测 RSMT 中的最优斯坦纳点,并且可以在 GPU 上并行化。在本文中,我们提出了 GAT-Steiner 模型,该模型在 ISPD19 基准测试中正确预测 99.846% 的网络,其平均线长增加仅为 0.480%(针对次优线长网络)。在随机生成的基准测试中,GAT-Steiner 正确预测 99.942% 的网络,其平均线长增加仅为 0.420%。

关键词

引用

@article{arxiv.2407.01440,
  title  = {GAT-Steiner: Rectilinear Steiner Minimal Tree Prediction Using GNNs},
  author = {Bugra Onal and Eren Dogan and Muhammad Hadir Khan and Matthew R. Guthaus},
  journal= {arXiv preprint arXiv:2407.01440},
  year   = {2024}
}

备注

Preprint for The 2024 IEEE/ACM International Conference on Computer-Aided Design (ICCAD 2024)