arXiv:2412.07167cs.LGcs.AI2024-12NeurIPS被引 18

用强化学习做布局优化而非从零放置,提升芯片设计的PPA性能

Reinforcement Learning Policy as Macro Regulator Rather than Macro Placer

  • 将RL用于已有布局的微调,而非从头生成
  • 在ISPD 2005和ICCAD 2015上实现更优的线长与规则性
  • 可适配任意布局方法,适合芯片设计工程师使用

现代芯片设计中,布局需安置数百万个电路模块,直接影响功耗、性能和面积(PPA)。近年来,强化学习(RL)被用于提升布局质量,尤其是宏单元布局。然而,现有基于RL的方法存在训练时间长、泛化能力弱、无法保证PPA结果的问题。根本原因在于问题设定:使用RL从零开始布局,导致训练过程中信息不足且奖励不准确。本文提出一种新方法:将RL应用于布局优化阶段,使策略学习如何调整已有布局,从而获得充分信息和密集精确的奖励。此外,我们在训练中引入规则性指标,该指标在芯片设计行业至关重要,但常被现有方法忽略。我们在ISPD 2005和ICCAD 2015基准上评估,对比了全局半周长布线长度与规则性表现。同时使用商用软件测试了PPA性能,结果表明以RL为调节器可显著改善芯片性能。该方法可对任意布局结果进行微调并提升质量。本工作为RL在布局中的应用开辟新路径,提供更高效有效的优化方案。代码已开源。

原文摘要 · Abstract (English)

In modern chip design, placement aims at placing millions of circuit modules, which is an essential step that significantly influences power, performance, and area (PPA) metrics. Recently, reinforcement learning (RL) has emerged as a promising technique for improving placement quality, especially macro placement. However, current RL-based placement methods suffer from long training times, low generalization ability, and inability to guarantee PPA results. A key issue lies in the problem formulation, i.e., using RL to place from scratch, which results in limits useful information and inaccurate rewards during the training process. In this work, we propose an approach that utilizes RL for the refinement stage, which allows the RL policy to learn how to adjust existing placement layouts, thereby receiving sufficient information for the policy to act and obtain relatively dense and precise rewards. Additionally, we introduce the concept of regularity during training, which is considered an important metric in the chip design industry but is often overlooked in current RL placement methods. We evaluate our approach on the ISPD 2005 and ICCAD 2015 benchmark, comparing the global half-perimeter wirelength and regularity of our proposed method against several competitive approaches. Besides, we test the PPA performance using commercial software, showing that RL as a regulator can achieve significant PPA improvements. Our RL regulator can fine-tune placements from any method and enhance their quality. Our work opens up new possibilities for the application of RL in placement, providing a more effective and efficient approach to optimizing chip design. Our code is available at \url{https://github.com/lamda-bbo/macro-regulator}.

芯片设计强化学习布局优化

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