Large Language Model for Verilog Generation with Code-Structure-Guided Reinforcement Learning

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Wang, Ning, Yao, Bingkun, Zhou, Jie, Wang, Xi, Jiang, Zhe, Guan, Nan
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910914667610112
author Wang, Ning
Yao, Bingkun
Zhou, Jie
Wang, Xi
Jiang, Zhe
Guan, Nan
author_facet Wang, Ning
Yao, Bingkun
Zhou, Jie
Wang, Xi
Jiang, Zhe
Guan, Nan
contents Recent advancements in large language models (LLMs) have sparked significant interest in the automatic generation of Register Transfer Level (RTL) designs, particularly using Verilog. Current research on this topic primarily focuses on pre-training and instruction tuning, but the effectiveness of these methods is constrained by the limited availability of training data, as public Verilog code is far less abundant than software code. In particular, these methods struggle to effectively capture Verilog parallel code structures, which fundamentally differ from the imperative, sequential control flow typical in most software programming languages. This paper introduces VeriSeek, an LLM enhanced by reinforcement learning using a limited amount of high-quality training data to achieve high Verilog code generation performance. Our reinforcement learning approach employs code structure information as feedback signals to refine the pre-trained model, enabling it to effectively learn important patterns from Verilog code with parallel structures. Experiments show that VeriSeek outperforms state-of-the-art methods across multiple benchmarks.
format Preprint
id arxiv_https___arxiv_org_abs_2407_18271
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Large Language Model for Verilog Generation with Code-Structure-Guided Reinforcement Learning
Wang, Ning
Yao, Bingkun
Zhou, Jie
Wang, Xi
Jiang, Zhe
Guan, Nan
Hardware Architecture
Artificial Intelligence
Recent advancements in large language models (LLMs) have sparked significant interest in the automatic generation of Register Transfer Level (RTL) designs, particularly using Verilog. Current research on this topic primarily focuses on pre-training and instruction tuning, but the effectiveness of these methods is constrained by the limited availability of training data, as public Verilog code is far less abundant than software code. In particular, these methods struggle to effectively capture Verilog parallel code structures, which fundamentally differ from the imperative, sequential control flow typical in most software programming languages. This paper introduces VeriSeek, an LLM enhanced by reinforcement learning using a limited amount of high-quality training data to achieve high Verilog code generation performance. Our reinforcement learning approach employs code structure information as feedback signals to refine the pre-trained model, enabling it to effectively learn important patterns from Verilog code with parallel structures. Experiments show that VeriSeek outperforms state-of-the-art methods across multiple benchmarks.
title Large Language Model for Verilog Generation with Code-Structure-Guided Reinforcement Learning
topic Hardware Architecture
Artificial Intelligence
url https://arxiv.org/abs/2407.18271