paper-with-me

Papers

DeepOHeat-v1: Efficient Operator Learning for Fast and Trustworthy Thermal Simulation and Optimization in 3D-IC Design

2025-04-04 · Xinling Yu, Ziyue Liu, Hai Li, Yixing Li, Xin Ai, Zhiyu Zeng, Ian Young, Zheng Zhang

Thermal analysis is crucial in three-dimensional integrated circuit (3D-IC) design due to increased power density and complex heat dissipation paths. Although operator learning frameworks such as DeepOHeat 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.25\times$ and $6.29\times$ 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 $62\times$ training speedup and $31\times$ 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.

📄 PDF Abstract BibTeX arXiv:2504.03955

Code (1)

xlyu0127/DeepOHeat-v1 jax

Tasks

GPUKolmogorov-Arnold NetworksOperator learning

Similar Papers 제목 키워드 기반

DeepOHeat: Operator Learning-based Ultra-fast Thermal Simulation in 3D-IC Design

2023-02-25 · Ziyue Liu, Yixing Li, Jing Hu, Xinling Yu 외

Thermal issue is a major concern in 3D integrated circuit (IC) design. Thermal optimization of 3D IC often requires massive expensive PDE simulations. Neural network-based thermal prediction models can perform real-time …

Operator learning

DeepOHeat-v2: Self-Improving Operator Learning for Fast and Trustworthy Thermal Optimization in 3D-IC Design

2026-08-17 · Xinling Yu, Yixing Li, Ziyue Liu, Xin Ai 외 arxiv

Thermal-aware optimization of multi-die 3D integrated circuits evaluates many designs, each a costly heat-equation solve. Operator-learning surrogates replace this solve with a fast forward pass, ideally trained from phy…

Self-Attention to Operator Learning-based 3D-IC Thermal Simulation

2025-10-12 · Zhen Huang, Hong Wang, Wenkai Yang, Muxi Tang 외 arxiv

Thermal management in 3D ICs is increasingly challenging due to higher power densities. Traditional PDE-solving-based methods, while accurate, are too slow for iterative design. Machine learning approaches like FNO provi…

Transfer Learning

A fast general thermal simulation model based on MultiBranch Physics-Informed deep operator neural network

2024-03-19 · Physics of Fluids 2024 3 · Lu, Zibo ; Zhou, Yuanye ; Zhang, Yanbo ; Hu 외

Thermal simulation plays a crucial role in various fields, often involving complex partial differential equation (PDE) simulations for thermal optimization. To tackle this challenge, we have harnessed neural networks for…

Transfer Learning

Accelerating IC Thermal Simulation Data Generation via Block Krylov and Operator Action

2025-10-27 · Hong Wang, Wenkai Yang, Jie Wang, Huanshuo Dong 외 arxiv

Recent advances in data-driven approaches, such as neural operators (NOs), have shown substantial efficacy in reducing the solution time for integrated circuit (IC) thermal simulations. However, a limitation of these app…