用AI加速3D芯片热仿真,速度快70倍且结果可信。
DeepOHeat-v1: Efficient Operator Learning for Fast and Trustworthy Thermal Simulation and Optimization in 3D-IC Design
- 用可学习激活的柯尔莫戈洛夫网络自适应捕捉多尺度温度分布。
- 训练速度提升62倍,显存减少31倍,支持更高分辨率仿真。
- 引入置信度评分与混合优化,让AI预测结果更可靠,适合芯片设计优化。
三维集成电路(3D-IC)因功耗密度上升和复杂散热路径,热分析至关重要。尽管深度算子学习框架如DeepOHeat已在加速热仿真方面展现潜力,但仍面临多尺度温度模式预测能力不足、训练效率低及优化过程中结果可信度不高的问题。本文提出DeepOHeat-v1,通过三项创新解决上述挑战:首先,采用带可学习激活函数的柯尔莫戈洛夫-阿诺德网络作为主干网络,自适应表征多尺度热分布,在两个典型测试案例中误差分别降低1.25倍和6.29倍;其次,提出可分离训练方法,沿坐标轴分解基函数,实现62倍训练加速与31倍GPU显存减少,使此前受显存限制无法实现的高分辨率热分析成为可能;第三,引入置信度评分评估预测可靠性,并设计结合有限差分(FD)与广义最小残差(GMRES)法的混合优化流程,实现增量式解精炼,保障热优化过程高效可信。实验表明,DeepOHeat-v1在精度上接近高保真有限差分求解器,同时使整个优化流程提速70.6倍,有效降低峰值温度。代码已开源。
原文摘要 · Abstract (English)
Thermal analysis is crucial in 3D-IC design due to increased power density and complex heat dissipation paths. Although operator learning frameworks such as DeepOHeat~\cite{liu2023deepoheat} have demonstrated promising preliminary results in accelerating thermal simulation, they face critical limitations in prediction capability for multi-scale thermal patterns, training efficiency, and trustworthiness of results during design optimization. This paper presents DeepOHeat-v1, an enhanced physics-informed operator learning framework that addresses these challenges through three key innovations. First, we integrate Kolmogorov-Arnold Networks with learnable activation functions as trunk networks, enabling an adaptive representation of multi-scale thermal patterns. This approach achieves a 1.25x and 6.29x reduction in error in two representative test cases. Second, we introduce a separable training method that decomposes the basis function along the coordinate axes, achieving 62x training speedup and 31x GPU memory reduction in our baseline case, and enabling thermal analysis at resolutions previously infeasible due to GPU memory constraints. Third, we propose a confidence score to evaluate the trustworthiness of the predicted results, and further develop a hybrid optimization workflow that combines operator learning with finite difference (FD) using Generalized Minimal Residual (GMRES) method for incremental solution refinement, enabling efficient and trustworthy thermal optimization. Experimental results demonstrate that DeepOHeat-v1 achieves accuracy comparable to optimization using high-fidelity finite difference solvers, while speeding up the entire optimization process by $70.6\times$ in our test cases, effectively minimizing the peak temperature through optimal placement of heat-generating components. Open source code is available at https://github.com/xlyu0127/DeepOHeat-v1.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。