DeepGate4让大电路图表示学习更高效,突破传统模型扩展瓶颈。
DeepGate4: Efficient and Effective Representation Learning for Circuit Design at Scale
- 设计专用更新策略与稀疏注意力机制,降低内存复杂度至亚线性。
- 在ITC99和EPFL数据集上分别提升15.5%和31.1%性能,超越现有方法。
- 适合需要处理大规模电路的EDA工程师或研究者使用。
电路表示学习在电子设计自动化中至关重要,支持可测性分析、逻辑推理、功耗估计和SAT求解等任务。然而,现有模型在扩展至大型电路时面临显著挑战,如图神经网络的过压缩问题以及基于Transformer模型的二次复杂度。为此,我们提出DeepGate4,一种专为大规模电路设计的可扩展图Transformer。该模型包含三项关键创新:(1) 针对电路图设计的更新策略,将内存复杂度降至亚线性,并适用于任意图Transformer;(2) 基于GAT的稀疏Transformer,结合全局与局部结构编码,用于AIG;(3) 充分利用AIG独特稀疏模式的推理加速CUDA内核。在ITC99和EPFL基准上的大量实验表明,DeepGate4显著优于现有最先进方法,在性能上分别实现15.5%和31.1%的提升。此外,Fused-DeepGate4变体将运行时间减少35.1%,内存使用降低46.8%,极大提升了大规模电路分析的效率。这些结果展示了DeepGate4在处理复杂EDA任务中的潜力,具备优异的可扩展性与效率。代码已开源:https://github.com/zyzheng17/DeepGate4-ICLR-25。
原文摘要 · Abstract (English)
Circuit representation learning has become pivotal in electronic design automation, enabling critical tasks such as testability analysis, logic reasoning, power estimation, and SAT solving. However, existing models face significant challenges in scaling to large circuits due to limitations like over-squashing in graph neural networks and the quadratic complexity of transformer-based models. To address these issues, we introduce DeepGate4, a scalable and efficient graph transformer specifically designed for large-scale circuits. DeepGate4 incorporates several key innovations: (1) an update strategy tailored for circuit graphs, which reduce memory complexity to sub-linear and is adaptable to any graph transformer; (2) a GAT-based sparse transformer with global and local structural encodings for AIGs; and (3) an inference acceleration CUDA kernel that fully exploit the unique sparsity patterns of AIGs. Our extensive experiments on the ITC99 and EPFL benchmarks show that DeepGate4 significantly surpasses state-of-the-art methods, achieving 15.5% and 31.1% performance improvements over the next-best models. Furthermore, the Fused-DeepGate4 variant reduces runtime by 35.1% and memory usage by 46.8%, making it highly efficient for large-scale circuit analysis. These results demonstrate the potential of DeepGate4 to handle complex EDA tasks while offering superior scalability and efficiency. Code is available at https://github.com/zyzheng17/DeepGate4-ICLR-25.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。