arXiv:2509.21886cs.AI2025-09

提出新模型TRACE,精准学习电路图的计算行为。

TRACE: Learning to Compute on Circuit Graphs

  • 用分层Transformer模拟计算步骤,捕捉位置与层级信息。
  • 通过函数偏移学习,准确预测电路全局功能变化。
  • 在多种电路数据上显著优于现有方法,适合芯片设计研究者。

学习计算,即建模电路图的功能行为,是图表示学习中的基础挑战。然而主流范式存在架构缺陷:消息传递神经网络(MPNN)及其基于Transformer的变体普遍采用置换不变聚合,无法捕捉计算的位置敏感性和层次性。为此,我们提出TRACE,一种基于合理架构和原则性学习目标的新范式。首先,TRACE采用分层Transformer,模拟计算的逐步流程,替代错误的置换不变聚合,提供更忠实的架构基础。其次,提出函数偏移学习,将复杂全局函数预测问题解耦:模型只需预测真实函数与假设输入独立性的简单局部近似之间的差异。我们在多种电路模态上验证该范式,包括寄存器传输级图、与非门图及映射后网表。在全面基准测试中,TRACE显著优于所有先前架构。结果表明,架构对齐的骨干网络与解耦学习目标共同构成更稳健的电路功能学习范式。

原文摘要 · Abstract (English)

Learning to compute, the ability to model the functional behavior of a circuit graph, is a fundamental challenge for graph representation learning. Yet, the dominant paradigm is architecturally mismatched for this task. This flawed assumption, central to mainstream message passing neural networks (MPNNs) and their conventional Transformer-based counterparts, prevents models from capturing the position-aware, hierarchical nature of computation. To resolve this, we introduce TRACE, a new paradigm built on an architecturally sound backbone and a principled learning objective. First, TRACE employs a Hierarchical Transformer that mirrors the step-by-step flow of computation, providing a faithful architectural backbone that replaces the flawed permutation-invariant aggregation. Second, we introduce function shift learning, a novel objective that decouples the learning problem. Instead of predicting the complex global function directly, our model is trained to predict only the function shift, the discrepancy between the true global function and a simple local approximation that assumes input independence. We validate this paradigm on various circuits modalities, including Register Transfer Level graphs, And-Inverter Graphs and post-mapping netlists. Across a comprehensive suite of benchmarks, TRACE substantially outperforms all prior architectures. These results demonstrate that our architecturally-aligned backbone and decoupled learning objective form a more robust paradigm for the fundamental challenge of learning the functional behavior of a circuit graph.

电路图图神经网络分层Transformer函数学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。