提出新模型FSA-Heat,快速高精度预测2.5D芯片热分布。
A Novel Frequency-Spatial Domain Aware Network for Fast Thermal Prediction in 2.5D ICs
- 融合频域与空间域特征,捕捉全局热扩散规律。
- 相较GCN+PNA,RMSE降低超99%,推理速度提升4.23倍。
- 适合需要快速热仿真和泛化能力强的芯片设计场景。
后摩尔时代,2.5D芯粒集成芯片因功耗密度升高和热热点问题,面临严峻热管理挑战。基于神经网络的热预测模型可对大量未见的新设计实现实时预测。然而,现有基于CNN和GCN的方法难以有效捕捉高频热特征,限制了预测精度提升。本文提出一种新型频-空双域感知热预测网络(FSA-Heat),通过高频到低频与空间域编码器(FSTE)模块,结合频域跨尺度交互模块(FCIFormer),实现高频到低频及全局到局部热耗散特征提取。此外,设计频-空混合损失函数(FSL),有效抑制高频热梯度噪声与空间错位。实验结果表明,所提方法性能显著优于新提出的2.5D方法GCN+PNA,RMSE降低超过99%,推理速度提升4.23倍。大量实验还验证了FSA-Heat具备强泛化能力。
原文摘要 · Abstract (English)
In the post-Moore era, 2.5D chiplet-based ICs present significant challenges in thermal management due to increased power density and thermal hotspots. Neural network-based thermal prediction models can perform real-time predictions for many unseen new designs. However, existing CNN-based and GCN-based methods cannot effectively capture the global thermal features, especially for high-frequency components, hindering prediction accuracy enhancement. In this paper, we propose a novel frequency-spatial dual domain aware prediction network (FSA-Heat) for fast and high-accuracy thermal prediction in 2.5D ICs. It integrates high-to-low frequency and spatial domain encoder (FSTE) module with frequency domain cross-scale interaction module (FCIFormer) to achieve high-to-low frequency and global-to-local thermal dissipation feature extraction. Additionally, a frequency-spatial hybrid loss (FSL) is designed to effectively attenuate high-frequency thermal gradient noise and spatial misalignments. The experimental results show that the performance enhancements offered by our proposed method are substantial, outperforming the newly-proposed 2.5D method, GCN+PNA, by considerable margins (over 99% RMSE reduction, 4.23X inference time speedup). Moreover, extensive experiments demonstrate that FSA-Heat also exhibits robust generalization capabilities.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。