OriGen: Enhancing RTL Code Generation with Code-to-Code Augmentation and Self-Reflection
Recent studies have demonstrated the significant potential of Large Language Models (LLMs) in generating Register Transfer Level (RTL) code, with notable advancements showcased by commercial models such as GPT-4 and Claude3-Opus. However, these proprietary LLMs often raise concerns regarding privacy and security. While open-source LLMs offer solutions to these concerns, they typically underperform commercial models in RTL code generation tasks, primarily due to the scarcity of high-quality open-source RTL datasets. To address this challenge, we introduce OriGen, a fully open-source framework that incorporates self-reflection capabilities and a novel dataset augmentation methodology for generating high-quality, large-scale RTL code. Our approach employs a code-to-code augmentation technique to enhance the quality of open-source RTL code datasets. Furthermore, OriGen can rectify syntactic errors through a self-reflection process that leverages compiler feedback.
Authors
- Kexing Zhou (ORCID: https://orcid.org/0000-0001-7635-3425)
- Dahua Lin (ORCID: https://orcid.org/0000-0002-8865-7896)
- Youwei Xiao (ORCID: https://orcid.org/0000-0002-3636-3685)
- Demin Song
- Qiang Xu (ORCID: https://orcid.org/0000-0001-6747-126X)
- Guangyu Sun (ORCID: https://orcid.org/0000-0002-7315-6589)
- Yun Liang (ORCID: https://orcid.org/0000-0002-9076-7998)
- Qipeng Guo (ORCID: https://orcid.org/0000-0002-8805-8789)
- Xingcheng Zhang (ORCID: https://orcid.org/0009-0006-8525-0608)
- Fan Cui
- Chenyang Yin
Institutions
- Chinese University of Hong Kong (HK)
- Peking University (CN)
- Shanghai Artificial Intelligence Laboratory (CN)
Publication Details
- Published
- 2024-10-27
- DOI
- https://doi.org/10.1145/3676536.3676830
- Citations
- 21
- Primary Topic
- Embedded Systems Design Techniques
- Type
- article
- Field-Weighted Citation Impact
- 10.27