用图神经网络快速预测芯片封装热变形,精度高且可跨配置迁移。
WarPGNN: A Parametric Thermal Warpage Analysis Framework with Physics-aware Graph Neural Network
- 基于图结构建模芯片布局,通过GNN提取热变形特征
- 相比传统仿真提速超11万倍,误差仅1.26%全范围归一化RMSE
- 针对极端情况优化损失函数与网络结构,适合芯片设计迭代
随着系统级封装(SiP)芯片小芯片架构和异构2.5D/3D集成的发展,热致翘曲已成为关键可靠性问题。传统数值方法虽精度高但计算成本巨大,难以扩展至复杂小芯片-封装系统。本文提出WarPGNN,一种基于图神经网络的高效精准参数化热翘曲分析框架。首先将多芯片版图编码为简化传递闭包图(rTCG),再通过基于图卷积网络(GCN)的编码器提取层次化结构特征,最后采用类U-Net解码器从图嵌入重建翘曲分布。为应对翘曲数据的长尾分布,设计物理感知损失函数,并改进基于图同构网络(GIN)的消息传递编码器,提升极端情况下的学习性能与嵌入表达能力。数值结果表明,WarPGNN相较于二维高效有限元法实现超过205.91倍加速,对比三维有限元软件COMSOL更达119,766.64倍,同时保持1.26%全范围归一化均方根误差与2.21%翘曲值误差。相比近期基于DeepONet的模型,在预测精度相当的前提下,训练时间降低3.4倍,且在未见数据集上展现优异泛化能力,最高归一化误差仅3.69%,运行时间相近。
原文摘要 · Abstract (English)
With the advent of system-in-package (SiP) chiplet-based design and heterogeneous 2.5D/3D integration, thermal-induced warpage has become a critical reliability concern. While conventional numerical approaches can deliver highly accurate results, they often incur prohibitively high computational costs, limiting their scalability for complex chiplet-package systems. In this paper, we present WarPGNN, an efficient and accurate parametric thermal warpage analysis framework powered by Graph Neural Networks (GNNs). By operating directly on graphs constructed from the floorplans, WarPGNN enables fast warpage-aware floorplan exploration and exhibits strong transferability across diverse package configurations. Our method first encodes multi-die floorplans into reduced Transitive Closure Graphs (rTCGs), then a Graph Convolution Network (GCN)-based encoder extracts hierarchical structural features, followed by a U-Net inspired decoder that reconstructs warpage maps from graph feature embeddings. Furthermore, to address the long-tailed pattern of warpage data distribution, we developed a physics-informed loss and revised a message-passing encoder based on Graph Isomorphic Network (GIN) that further enhance learning performance for extreme cases and expressiveness of graph embeddings. Numerical results show that WarPGNN achieves more than 205.91x speedup compared with the 2-D efficient FEM-based method and over 119766.64x acceleration with 3-D FEM method COMSOL, respectively, while maintaining comparable accuracy at only 1.26% full-scale normalized RMSE and 2.21% warpage value error. Compared with recent DeepONet-based model, our method achieved comparable prediction accuracy and inference speedup with 3.4x lower training time. In addition, WarPGNN demonstrates remarkable transferability on unseen datasets with up to 3.69% normalized RMSE and similar runtime.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。