Reinforcement Learning Within Tree Search for Fast Macro Placement

Abstract

Macro placement is a crucial step in modern chip design, and reinforcement learning (RL) has recently emerged as a promising technique for improving the placement quality. However, existing RL-based techniques are hindered by their low sample efficiency, requiring numerous online rollouts or substantial offline expert data to achieve bootstrap, which are often impractical in industrial scenarios. To address this challenge, we propose a novel sample-efficient framework, namely EfficientPlace, for fast macro placement. EfficientPlace integrates a global tree search algorithm to strategically direct the optimization process, as well as a RL agent for local policy learning to advance the tree search. Experiments on commonly used benchmarks demonstrate that EfficientPlace achieves remarkable placement quality within a short timeframe, outperforming recent state-of-the-art approaches.

Cite

Text

Geng et al. "Reinforcement Learning Within Tree Search for Fast Macro Placement." International Conference on Machine Learning, 2024.

Markdown

[Geng et al. "Reinforcement Learning Within Tree Search for Fast Macro Placement." International Conference on Machine Learning, 2024.](https://mlanthology.org/icml/2024/geng2024icml-reinforcement/)

BibTeX

@inproceedings{geng2024icml-reinforcement,
  title     = {{Reinforcement Learning Within Tree Search for Fast Macro Placement}},
  author    = {Geng, Zijie and Wang, Jie and Liu, Ziyan and Xu, Siyuan and Tang, Zhentao and Yuan, Mingxuan and Hao, Jianye and Zhang, Yongdong and Wu, Feng},
  booktitle = {International Conference on Machine Learning},
  year      = {2024},
  pages     = {15402-15417},
  volume    = {235},
  url       = {https://mlanthology.org/icml/2024/geng2024icml-reinforcement/}
}