Chip Floorplanning Optimization Using Deep Reinforcement Learning

Authors

  • Shikai Wang Electrical and Computer Engineering, New York University, NY, USA
  • Haodong Zhang Computer Science, New York University, NY, USA
  • Shiji Zhou Computer Science, University of Southern California, CA, USA
  • Jun Sun Business Analytics and Project Management, University of Connecticut, CT, USA
  • Qi Shen Master of Business Administration, Columbia University, NY, USA

Keywords:

Deep Reinforcement Learning, Graph Neural Networks, Chip Floorplanning, Electronic Design Automation

Abstract

This paper presents a new method for chip floorplanning optimization using deep learning (DRL) combined with graph neural networks (GNNs). The plan addresses the challenges of traditional floor plans by applying AI to space design and intelligent space decisions. Three-head network architecture, including a policy network, cost network, and reconstruction head, is introduced to improve feature extraction and overall performance. GNNs are employed for state representation and feature extraction, enabling the capture of intricate topological information from chip netlists. A carefully designed reward function incorporating wire length minimization, area utilization, and timing constraint satisfaction guides the DRL agent toward high-quality floorplan solutions. An exploration bonus based on reconstruction error addresses the sparse reward problem. Extensive testing of the ISPD 2005 benchmarks demonstrated the effectiveness of the proposed approach, consistently operating on a state-of-the-art basis. Significant improvements include an average 31.4% reduction in half-perimeter wire length (HPWL) and a 34.2% reduction in breach time compared to the best baseline performance. The process scalability and robustness are evaluated, showing performance in various circuits and different perturbations. This research advances AI-driven electronic device design and paves the way for better chip design processes.

References

T. Andersen, "AI Chips Built by AI-Promise or Reality? An Industry Perspective," in Proc. 2022 ACM/IEEE Workshop on Machine Learning for CAD, Sep. 2022, pp. 51-51. Available From: https://doi.org/10.1145/3551901.3557043

D. Zhao, S. Yuan, Y. Sun, S. Tu, and L. Xu, "DeepTH: Chip Placement with Deep Reinforcement Learning Using a Three-Head Policy Network," in 2023 Design, Automation & Test in Europe Conf. & Exhibition (DATE), Apr. 2023, pp. 1-2. Available From: https://doi.org/10.23919/DATE56975.2023.10137100

M. E. Yanık, İ. Çiçek, and E. Afacan, "ShortCircuit: An Open-Source ChatGPT Driven Digital Integrated Circuit Front-End Design Automation Tool," in 2023 30th IEEE Int. Conf. Electronics, Circuits and Systems (ICECS), Dec. 2023, pp. 1-4. Available From: https://doi.org/10.1109/ICECS58634.2023.10382808

A. Malhotra and A. Singh, "Implementation of AI in the Field of VLSI: A Review," in 2022 2nd Int. Conf. Power, Control and Computing Technologies (ICPC2T), Mar. 2022, pp. 1-5. Available From: https://doi.org/10.1109/ICPC2T53885.2022.9776845

V. Janpoladov, "A Machine Learning-Based Post-Route PVT-Aware Power Prediction of Benchmark Circuits at Floorplan Stage of Physical Design," in 2023 IEEE East-West Design & Test Symp. (EWDTS), Sep. 2023, pp. 1-6. Available From: https://doi.org/10.1109/EWDTS59469.2023.10297036

S. Li, H. Xu, T. Lu, G. Cao, and X. Zhang, "Emerging Technologies in Finance: Revolutionizing Investment Strategies and Tax Management in the Digital Era," Manage. J. Adv. Res., vol. 4, no. 4, pp. 35-49, 2024. Available From: https://doi.org/10.5281/zenodo.13283670

J. Shi, F. Shang, S. Zhou, et al., "Applications of Quantum Machine Learning in Large-Scale E-commerce Recommendation Systems: Enhancing Efficiency and Accuracy," J. Ind. Eng. Appl. Sci., vol. 2, no. 4, pp. 90-103, 2024. Available From: https://doi.org/10.5281/zenodo.13117899

S. Wang, H. Zheng, X. Wen, and S. Fu, "Distributed High-Performance Computing Methods for Accelerating Deep Learning Training," J. Knowl. Learn. Sci. Technol., vol. 3, no. 3, pp. 108-126, 2024. Available From: https://doi.org/10.60087/jklst.v3.n3.p108-126

M. Zhang, B. Yuan, H. Li, and K. Xu, "LLM-Cloud Complete: Leveraging Cloud Computing for Efficient Large Language Model-based Code Completion," J. Artif. Intell. Gen. Sci., vol. 5, no. 1, pp. 295-326, 2024. Available From: https://doi.org/10.60087/jaigs.v5i1.200

H. Lei, B. Wang, Z. Shui, P. Yang, and P. Liang, "Automated Lane Change Behavior Prediction and Environmental Perception Based on SLAM Technology," arXiv preprint, arXiv:2404.04492, 2024. Available From: https://doi.org/10.48550/arXiv.2404.04492

B. Wang, Y. He, Z. Shui, Q. Xin, and H. Lei, "Predictive Optimization of DDoS Attack Mitigation in Distributed Systems Using Machine Learning," Appl. Comput. Eng., vol. 64, pp. 95-100, 2024. Available From: https://www.researchgate.net/profile/Qi-Xin-32/publication/379897526_Predictive_Optimization_of_DDoS_Attack_Mitigation_in_Distributed_Systems_using_Machine_Learning/links/6620b89166ba7e2359e6379f/Predictive-Optimization-of-DDoS-Attack-Mitigation-in-Distributed-Systems-using-Machine-Learning.pdf

B. Wang, H. Zheng, K. Qian, X. Zhan, and J. Wang, "Edge Computing and AI-Driven Intelligent Traffic Monitoring and Optimization," Appl. Comput. Eng., vol. 77, pp. 225-230, 2024. https://doi.org/10.54254/2755-2721/77/2024MA0062

Y. Xu, Y. Liu, H. Xu, and H. Tan, "AI-Driven UX/UI Design: Empirical Research and Applications in FinTech," Int. J. Innov. Res. Comput. Sci. Technol., vol. 12, no. 4, pp. 99-109, 2024. Available From: https://doi.org/10.55524/ijircst.2024.12.4.16

Y. Liu, Y. Xu, and R. Song, "Transforming User Experience (UX) through Artificial Intelligence (AI) in Interactive Media Design," Eng. Sci. Technol. J., vol. 5, no. 7, pp. 2273-2283, 2024. Available From: https://doi.org/10.20944/preprints202409.0168.v1

P. Zhang, "A Study on the Location Selection of Logistics Distribution Centers Based on E-Commerce," J. Knowl. Learn. Sci. Technol., vol. 3, no. 3, pp. 103-107, 2024. https://doi.org/10.60087/jklst.vol3.n3.p103-107

P. Zhang and L. I. U. Gan, "Optimization of Vehicle Scheduling for Joint Distribution in the Logistics Park Based on Priority," J. Ind. Eng. Appl. Sci., vol. 2, no. 4, pp. 116-121, 2024. https://n2t.net/ark:/40704/JIEAS.v2n4a17

H. Li, S. X. Wang, F. Shang, K. Niu, and R. Song, "Applications of Large Language Models in Cloud Computing: An Empirical Study Using Real-World Data," Int. J. Innov. Res. Comput. Sci. Technol., vol. 12, no. 4, pp. 59-69, 2024. https://doi.org/10.55524/ijircst.2024.12.4.10

G. Ping, S. X. Wang, F. Zhao, Z. Wang, and X. Zhang, "Blockchain-Based Reverse Logistics Data Tracking: An Innovative Approach to Enhance E-Waste Recycling Efficiency," 2024. https://doi.org/10.53469/wjimt.2024.07(04).02

H. Xu, K. Niu, T. Lu, and S. Li, "Leveraging Artificial Intelligence for Enhanced Risk Management in Financial Services: Current Applications and Prospects," Eng. Sci. Technol. J., vol. 5, no. 8, pp. 2402-2426, 2024. Available From: https://doi.org/10.5281/zenodo.13765819

Y. Shi, F. Shang, Z. Xu, and S. Zhou, "Emotion-Driven Deep Learning Recommendation Systems: Mining Preferences from User Reviews and Predicting Scores," J. Artif. Intell. Dev., vol. 3, no. 1, pp. 40-46, 2024. Available From: https://edujavare.com/index.php/JAI/article/view/472

S. Wang, K. Xu, and Z. Ling, "Deep Learning-Based Chip Power Prediction and Optimization: An Intelligent EDA Approach," Int. J. Innov. Res. Comput. Sci. Technol., vol. 12, no. 4, pp. 77-87, 2024. Available From: https://doi.org/10.55524/ijircst.2024.12.4.13

G. Ping, M. Zhu, Z. Ling, and K. Niu, "Research on Optimizing Logistics Transportation Routes Using AI Large Models," Appl. Sci. Eng. J. Adv. Res., vol. 3, no. 4, pp. 14-27, 2024. Available From: https://doi.org/10.5281/zenodo.12787012

F. Shang, J. Shi, Y. Shi, and S. Zhou, "Enhancing E-Commerce Recommendation Systems with Deep Learning-Based Sentiment Analysis of User Reviews," Int. J. Eng. Manage. Res., vol. 14, no. 4, pp. 19-34, 2024. Available From: https://doi.org/10.5281/zenodo.13221409

H. Xu, S. Li, K. Niu, and G. Ping, "Utilizing Deep Learning to Detect Fraud in Financial Transactions and Tax Reporting," J. Econ. Theory Bus. Manage., vol. 1, no. 4, pp. 61-71, 2024. https://doi.org/10.5281/zenodo.13294459

K. Xu, H. Zhou, H. Zheng, M. Zhu, and Q. Xin, "Intelligent Classification and Personalized Recommendation of E-Commerce Products Based on Machine Learning," arXiv preprint, arXiv:2403.19345, 2024. Available From: https://doi.org/10.48550/arXiv.2403.19345

K. Xu, H. Zheng, X. Zhan, S. Zhou, and K. Niu, "Evaluation and Optimization of Intelligent Recommendation System Performance with Cloud Resource Automation Compatibility," 2024. Available From: https://doi.org/10.54254/2755-2721/87/20241620

H. Zheng, K. Xu, H. Zhou, Y. Wang, and G. Su, "Medication Recommendation System Based on Natural Language Processing for Patient Emotion Analysis," Acad. J. Sci. Technol., vol. 10, no. 1, pp. 62-68, 2024. Available From: https://doi.org/10.54097/v160aa61

H. Zheng, J. Wu, R. Song, L. Guo, and Z. Xu, "Predicting Financial Enterprise Stocks and Economic Data Trends Using Machine Learning Time Series Analysis," Appl. Comput. Eng., vol. 87, pp. 26-32, 2024. Available From: https://doi.org/10.54254/2755-2721/87/20241562

X. Zhan, C. Shi, L. Li, K. Xu, and H. Zheng, "Aspect Category Sentiment Analysis Based on Multiple Attention Mechanisms and Pre-Trained Models," Appl. Comput. Eng., vol. 71, pp. 21-26, 2024. Available From: https://doi.org/10.54254/2755-2721/67/2024MA0055

B. Liu, X. Zhao, H. Hu, Q. Lin, and J. Huang, "Detection of Esophageal Cancer Lesions Based on CBAM Faster R-CNN," J. Theory Pract. Eng. Sci., vol. 3, no. 12, pp. 36-42, 2023. Available From: https://doi.org/10.53469/jtpes.2023.03(12).06

B. Liu, L. Yu, C. Che, Q. Lin, H. Hu, and X. Zhao, "Integration and performance analysis of artificial intelligence and computer vision based on deep learning algorithms," Applied and Computational Engineering, vol. 64, pp. 36-41, 2024. Available From: https://doi.org/10.48550/arXiv.2312.12872

B. Liu, "Based on intelligent advertising recommendations and abnormal advertising monitoring systems in machine learning," International Journal of Computer Science and Information Technology, vol. 1, no. 1, pp. 17-23, 2023. Available From: https://doi.org/10.62051/ijcsit.v1n1.03

P. Liang, B. Song, X. Zhan, Z. Chen, and J. Yuan, "Automating the training and deployment of models in MLOps by integrating systems with machine learning," Applied and Computational Engineering, vol. 67, pp. 1-7, 2024. Available From: https://doi.org/10.48550/arXiv.2405.09819

B. Wu, Y. Gong, H. Zheng, Y. Zhang, J. Huang, and J. Xu, "Enterprise cloud resource optimization and management based on cloud operations," Applied and Computational Engineering, vol. 67, pp. 8-14, 2024. Available From: https://doi.org/10.54254/2755-2721/76/20240667

B. Liu and Y. Zhang, "Implementation of seamless assistance with Google Assistant leveraging cloud computing," Journal of Cloud Computing, vol. 12, no. 4, pp. 1-15, 2023. Available From: http://dx.doi.org/10.54254/2755-2721/64/20241383

L. Guo, Z. Li, K. Qian, W. Ding, and Z. Chen, "Bank credit risk early warning model based on machine learning decision trees," Journal of Economic Theory and Business Management, vol. 1, no. 3, pp. 24-30, 2024. Available From: https://doi.org/10.5281/zenodo.11627011

Z. Xu, L. Guo, S. Zhou, R. Song, and K. Niu, "Enterprise supply chain risk management and decision support driven by large language models," Applied Science and Engineering Journal for Advanced Research, vol. 3, no. 4, pp. 1-7, 2024. Available From: https://doi.org/10.5281/zenodo.12670581

R. Song, Z. Wang, L. Guo, F. Zhao, and Z. Xu, "Deep belief networks (DBN) for financial time series analysis and market trends prediction," World Journal of Innovative Medical Technologies, vol. 5, no. 3, pp. 27-34, 2024. Available From: https://doi.org/10.53469/wjimt.2024.07(04).01

L. Guo, R. Song, J. Wu, Z. Xu, and F. Zhao, "Integrating a machine learning-driven fraud detection system based on a risk management framework," Preprints, 2024. Available From: https://doi.org/10.54254/2755-2721/87/20241541

Y. Feng, Y. Qi, H. Li, X. Wang, and J. Tian, "Leveraging federated learning and edge computing for recommendation systems within cloud computing networks," in Proc. Third Int. Symp. Computer Applications and Information Systems (ISCAIS 2024), Jul. 2024, vol. 13210, pp. 279-287. Available From: https://doi.org/10.1117/12.3034773

S. Wang, K. Xu, and Z. Ling, "Deep learning-based chip power prediction and optimization: An intelligent EDA approach," International Journal of Innovative Research in Computer Science & Technology, vol. 12, no. 4, pp. 77-87, 2024. Available From: https://doi.org/10.55524/ijircst.2024.12.4.13

S. Wang, H. Zheng, X. Wen, and S. Fu, "Distributed high-performance computing methods for accelerating deep learning training," Journal of Knowledge Learning and Science Technology, vol. 3, no. 3, pp. 108-126, 2024. Available From: https://doi.org/10.60087/jklst.v3.n3.p108-126

Z. Yuan, J. Yang, Y. Zhang, S. Wang, and T. Xu, "Mass transport optimization in the anode diffusion layer of a micro direct methanol fuel cell," Energy, vol. 93, pp. 599-605, 2015. Available From: https://doi.org/10.1016/j.energy.2015.09.067

S. Wang, Y. Zhu, Q. Lou, and M. Wei, "Utilizing artificial intelligence for financial risk monitoring in asset management," Academic Journal of Sociology and Management, vol. 2, no. 5, pp. 11-19, 2024. Available From: https://doi.org/10.5281/zenodo.13762069

S. Wang, H. Zheng, X. Wen, K. Xu, and H. Tan, "Enhancing chip design verification through AI-powered bug detection in RTL code," Applied and Computational Engineering, vol. 92, pp. 27-33, 2024. Available From: https://doi.org/10.54254/2755-2721/92/20241685

H. Zheng, K. Xu, H. Zhou, Y. Wang, and G. Su, "Medication recommendation system based on natural language processing for patient emotion analysis," Academic Journal of Science and Technology, vol. 10, no. 1, pp. 62-68, 2024. Available From: https://doi.org/10.54097/v160aa61

H. Zheng, J. Wu, R. Song, L. Guo, and Z. Xu, "Predicting financial enterprise stocks and economic data trends using machine learning time series analysis," Preprints, 2024. Available From: https://doi.org/10.54254/2755-2721/87/20241562

S. Wang, H. Zheng, X. Wen, and S. Fu, "Distributed high-performance computing methods for accelerating deep learning training," Journal of Knowledge Learning and Science Technology, vol. 3, no. 3, pp. 108-126, 2024. Available From: https://doi.org/10.60087/jklst.v3.n3.p108-126

K. Xu, H. Zheng, X. Zhan, S. Zhou, and K. Niu, "Evaluation and optimization of intelligent recommendation system performance with cloud resource automation compatibility," Preprints, 2024. Available From: https://doi.org/10.54254/2755-2721/87/20241620

G. Ruan, D. S. Kirschen, H. Zhong, Q. Xia, and C. Kang, "Estimating demand flexibility using Siamese LSTM neural networks," IEEE Transactions on Power Systems, vol. 37, no. 3, pp. 2360-2370, 2021. Available From: https://doi.org/10.1109/TPWRS.2021.3110723

Y. Yang, Z. Tan, H. Yang, G. Ruan, H. Zhong, and F. Liu, "Short-term electricity price forecasting based on graph convolution network and attention mechanism," IET Renewable Power Generation, vol. 16, no. 12, pp. 2481-2492, 2022. Available From: https://doi.org/10.1049/rpg2.12413

Z. Tan, G. Ruan, H. Zhong, and Q. Xia, "Security pre-check method of bilateral trading adapted to independence of power exchange," Automation of Electric Power Systems, vol. 42, no. 10, pp. 106-113, 2018. Available From: http://dx.doi.org/10.7500/AEPS20171005002

G. Ruan, D. Qiu, S. Sivaranjani, A. S. Awad, and G. Strbac, "Data-driven energy management of virtual power plants: A review," Advances in Applied Energy, vol. 100170, 2024. Available From: https://doi.org/10.1016/j.adapen.2024.100170

A. Li, S. Zhuang, T. Yang, W. Lu, and J. Xu, "Optimization of logistics cargo tracking and transportation efficiency based on data science deep learning models," Preprints, 2024. Available From: https://doi.org/10.54254/2755-2721/69/20241522

F. Zhao, H. Li, K. Niu, J. Shi, and R. Song, "Application of deep learning-based intrusion detection system (IDS) in network anomaly traffic detection," Preprints, 2024. Available From: https://doi.org/10.54254/2755-2721/86/20241604

Downloads

Published

2024-10-02

How to Cite

[1]
S. Wang, H. Zhang, S. Zhou, J. Sun, and Q. Shen, “Chip Floorplanning Optimization Using Deep Reinforcement Learning”, IJIRCST, vol. 12, no. 5, pp. 100–109, Oct. 2024.