MOSS: Multi-Modal Representation Learning on Sequential Circuits
Deep learning has significantly advanced Electronic Design Automation (EDA), with circuit representation learning emerging as a key area for modeling the relationship between a circuit’s structure and functionality. Existing methods primarily use either Large Language Models (LLMs) for Register Transfer Level (RTL) code analysis or Graph Neural Networks (GNNs) for netlist modeling. While LLMs excel at high-level functional understanding, they struggle with detailed netlist behavior. GNNs, however, face challenges when scaling to larger sequential circuits due to long-range information dependencies and insufficient functional supervision, leading to decreased accuracy and limited generalization. To address these challenges, we propose MOSS, a multimodal framework that integrates GNNs with LLMs for sequential circuit modeling. By enhancing D-type Flip-Flop (DFF) node features with embeddings from fine-tuned LLMs on RTL code, we focus the GNN on critical anchor points, reducing reliance on long-range dependencies. The LLM also provides global circuit embeddings, offering efficient supervision for functionality-related tasks. Additionally, MOSS introduces an adaptive aggregation method and a two-phase propagation mechanism in the GNN to better model signal propagation and sequential feedback within the circuit. Experimental results demonstrate that MOSS significantly improves the accuracy of functionality and performance predictions for sequential circuits compared to existing methods, particularly in larger circuits where previous models struggle. Specifically, MOSS achieves a $\mathbf{9 5. 2 \%}$ accuracy in arrival time prediction.
Authors
- Bin Sun (ORCID: https://orcid.org/0000-0002-5663-4294)
- Jianan Mu (ORCID: https://orcid.org/0000-0001-8513-0792)
- Silin Liu (ORCID: https://orcid.org/0000-0001-7406-4179)
- Xiaowei Li (ORCID: https://orcid.org/0000-0002-0874-814X)
- Zhiteng Chao (ORCID: https://orcid.org/0009-0006-2926-7499)
- Bei Yu (ORCID: https://orcid.org/0000-0001-6406-4810)
- Husheng Han (ORCID: https://orcid.org/0000-0003-0873-3471)
- Tianmeng Yang
- Xinyu Zhang (ORCID: https://orcid.org/0000-0001-9688-8056)
- Jing Ye (ORCID: https://orcid.org/0000-0002-8023-5090)
- Hui Wang (ORCID: https://orcid.org/0000-0002-8745-4042)
- Mingjun Wang (ORCID: https://orcid.org/0000-0003-0829-0359)
- Huawei Li (ORCID: https://orcid.org/0000-0001-8082-4218)
- Jun Gao (ORCID: https://orcid.org/0000-0003-2959-2778)
- Shengwen Liang
- Zizhen Liu (ORCID: https://orcid.org/0000-0002-1674-9361)
- Yihan Wen
- Buxing Han (ORCID: https://orcid.org/0000-0003-0440-809X)
- Feng Gu
Institutions
- Chinese University of Hong Kong (HK)
- Chinese Academy of Sciences (CN)
- Peking University (CN)
- Beijing University of Technology (CN)
- Institute of Computing Technology (CN)
- Stanford University (US)
Publication Details
- Published
- 2025-06-22
- DOI
- https://doi.org/10.1109/dac63849.2025.11132437
- Citations
- 1
- Primary Topic
- Neural Networks and Applications
- Type
- article
- Field-Weighted Citation Impact
- 2.48