WarpagePINN: Thermal Warpage Prediction in Advanced Packaging via a Two-Stage Physics-Informed Neural Networks
Pith reviewed 2026-07-02 08:26 UTC · model grok-4.3
The pith
A two-stage neural network predicts chiplet thermal warpage from physical equations alone, matching finite element results at 0.2 micrometer error.
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
Core claim
The WarpagePINN framework computes both the temperature profile and warpage deformation of chiplets by training neural networks exclusively through losses derived from physical laws, without any labeled deformation data. The first stage employs a Fourier-series representation to satisfy boundary conditions inherently while minimizing the residual of the governing thermal equation. The second stage uses a multilayer perceptron with a hybrid supervisory strategy to optimize an energy-based loss function for the deformation field. A parametric extension allows quantification of uncertainties in the coefficients of thermal expansion. Numerical tests demonstrate a mean absolute error of 0.2 micro
What carries the argument
The two-stage WarpagePINN, consisting of a Fourier-series temperature network trained on PDE loss followed by an MLP warpage network trained on hybrid energy loss.
If this is right
- Enables rapid evaluation of warpage across many different CTE values without repeated finite element runs.
- Provides built-in uncertainty quantification for material property variations in packaging design.
- Removes the need to generate large labeled datasets from conventional simulators before training.
- Maintains pointwise agreement with finite element solutions at the level of 0.2 micrometers for the geometries considered.
Where Pith is reading between the lines
- The same two-stage structure could be extended to time-dependent thermal cycles or fully three-dimensional multilayer stacks.
- Embedding the parametric network inside an outer optimization loop would allow automatic selection of material properties that minimize warpage.
- If sensor data were available during operation, the energy loss could be augmented with real-time measurements to refine predictions on the fly.
Load-bearing premise
The hybrid supervisory strategy can successfully optimize the energy-based loss for warpage without any labeled deformation data, and the Fourier-series temperature representation remains accurate for the chiplet geometries and boundary conditions tested.
What would settle it
Run WarpagePINN and a finite element solver on a chiplet geometry or set of boundary conditions outside the original test set and check whether the mean absolute error in predicted warpage stays below 0.2 micrometers.
Figures
read the original abstract
Thermal warpage has become a critical issue in advanced packaging, primarily caused by the mismatch in coefficients of thermal expansion (CTE) among heterogeneously integrated materials. However, only a limited number of studies have focused on developing computational methods for coupled thermal-warpage prediction in the chiplet. This paper proposes a two-stage physics-informed neural network (WarpagePINN) framework to compute both temperature profile and warpage deformation of chiplets. The neural networks are trained without relying on labeled datasets generated by conventional simulators. In the first stage, the temperature field is modeled using a Fourier series representation that inherently satisfies boundary conditions, and the network is trained solely through a loss function derived from the governing equation. In the second stage, a multilayer perceptron (MLP) is employed for warpage prediction, utilizing a novel hybrid supervisory strategy to optimize the energy-based loss function instead of residual loss. A parametric WarpagePINN is also developed to quantify uncertainties associated with the CTE. Numerical results show that the proposed WarpagePINN framework achieves excellent agreement with conventional finite element methods, with a mean absolute error (MAE) of 0.2 {\mu}m, while achieving a speedup of approximately 1000 {\times} in CTE parameterization studies.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The manuscript proposes WarpagePINN, a two-stage physics-informed neural network for coupled thermal-warpage prediction in chiplet packaging. Stage 1 represents the temperature field via a Fourier series that satisfies boundary conditions and is trained solely on the governing heat equation residual. Stage 2 employs an MLP for out-of-plane deformation whose loss is an energy functional minimized via a novel hybrid supervisory strategy that requires no labeled displacement data. A parametric extension allows rapid CTE sweeps. Numerical experiments report MAE of 0.2 μm against FEM reference solutions together with an approximately 1000× speedup on parameterization studies.
Significance. If the hybrid energy minimization in the second stage can be shown to enforce equilibrium and interface conditions without auxiliary labeled data, the framework would supply a genuinely data-free surrogate for thermal-mechanical analysis of heterogeneous packages. The reported speedup on CTE sweeps would then be directly useful for uncertainty quantification and design-space exploration in advanced packaging, where repeated FEM runs are currently prohibitive.
major comments (3)
- [§3.2] §3.2 (second-stage loss): the hybrid supervisory strategy is introduced without an explicit statement of the individual loss terms or their weighting; it is therefore impossible to verify whether the energy functional is minimized subject to strong enforcement of traction-free boundaries and displacement continuity at material interfaces, or whether auxiliary signals are implicitly present.
- [§4.1, Table 2] §4.1 and Table 2: the reported MAE of 0.2 μm is given as a single scalar without accompanying standard deviation across random seeds, mesh-convergence data for the FEM reference, or a statement of the maximum element size used in the comparison; this leaves open whether the agreement is robust or case-specific.
- [§3.3] §3.3 (parametric extension): the manner in which the CTE vector is injected into the network (embedding, conditioning, or separate input branch) is not specified, nor is any analysis provided that the learned mapping remains accurate when CTE values lie outside the training interval.
minor comments (3)
- [Abstract / §2.1] The abstract states that the Fourier series “inherently satisfies boundary conditions,” yet the precise form of the series and the treatment of non-homogeneous Dirichlet data on the chiplet edges are not shown until §2.1; a short equation block would improve readability.
- [Figure 3] Figure 3 caption refers to “warpage contours” but the color bar units are omitted; add units (μm) for immediate interpretability.
- [§4.3] The speedup factor of 1000× is stated for “CTE parameterization studies” without specifying the number of parameter samples or the wall-clock time of the reference FEM campaign; a brief table of timings would strengthen the claim.
Simulated Author's Rebuttal
We thank the referee for the constructive comments, which help clarify several aspects of the WarpagePINN framework. We address each major comment below and indicate planned revisions.
read point-by-point responses
-
Referee: [§3.2] §3.2 (second-stage loss): the hybrid supervisory strategy is introduced without an explicit statement of the individual loss terms or their weighting; it is therefore impossible to verify whether the energy functional is minimized subject to strong enforcement of traction-free boundaries and displacement continuity at material interfaces, or whether auxiliary signals are implicitly present.
Authors: We agree that the hybrid supervisory strategy in §3.2 needs explicit formulation. The revised manuscript will list each loss term (variational energy, traction-free boundary penalty, and interface continuity penalty) with their respective weighting coefficients. No labeled displacement data or auxiliary signals are used; enforcement occurs through the energy functional and penalty terms only. Equations detailing the full loss will be added. revision: yes
-
Referee: [§4.1, Table 2] §4.1 and Table 2: the reported MAE of 0.2 μm is given as a single scalar without accompanying standard deviation across random seeds, mesh-convergence data for the FEM reference, or a statement of the maximum element size used in the comparison; this leaves open whether the agreement is robust or case-specific.
Authors: We will augment §4.1 and Table 2 with the standard deviation of MAE computed over five independent random seeds, a mesh-convergence study for the FEM reference solutions (showing stabilization of warpage values), and the maximum element size employed in the comparisons. These additions will confirm the robustness of the reported 0.2 μm agreement. revision: yes
-
Referee: [§3.3] §3.3 (parametric extension): the manner in which the CTE vector is injected into the network (embedding, conditioning, or separate input branch) is not specified, nor is any analysis provided that the learned mapping remains accurate when CTE values lie outside the training interval.
Authors: In the revision we will specify that the CTE vector is provided through a dedicated input branch concatenated with the spatial coordinates before the first hidden layer of the MLP. The current parametric study is restricted to interpolation within the sampled CTE interval; we will add an explicit statement of this scope and a brief note on the absence of guaranteed accuracy for extrapolation. A dedicated out-of-range analysis lies outside the present scope. revision: partial
Circularity Check
No circularity: derivation uses independent physics losses and external FEM validation
full rationale
The abstract and description present a standard two-stage PINN: stage 1 enforces the heat equation via Fourier series (inherently satisfying BCs) and residual loss from the governing PDE; stage 2 optimizes an energy-based loss for warpage via a hybrid supervisory strategy with no deformation labels. No quoted step reduces a prediction to a fitted input by construction, invokes self-citation for uniqueness, or renames a known result. The reported MAE vs. FEM and speedup constitute external verification rather than internal tautology. The hybrid strategy is described as novel but not shown to collapse to the inputs.
Axiom & Free-Parameter Ledger
axioms (2)
- domain assumption The governing partial differential equations for steady heat conduction and linear elasticity are known and can be directly encoded as loss terms.
- domain assumption A Fourier-series representation can be constructed that inherently satisfies the boundary conditions of the temperature problem for the geometries considered.
Reference graph
Works this paper leans on
-
[1]
1.1 innovation for the next decade of compute efficiency,
L. Su and S. Naffziger, “1.1 innovation for the next decade of compute efficiency,” in Proc. IEEE Int. Solid-State Circuits Conf. (ISSCC), pp. 8–12, 2023
2023
-
[2]
Warpage and thermal characteriza- tion of fan-out wafer-level packaging,
J. H. Lau, M. Li, D. Tian, N. Fan, E. Kuah, W. Kai, M. Li, J. Hao, Y. M. Cheung, Z. Li, K. H. Tan, R. Beica, T. Taylor, C.- T. Ko, H. Yang, Y.-H. Chen, S. P. Lim, N. C. Lee, J. Ran, C. Xi, K. S. Wee, and Q. Yong, “Warpage and thermal characteriza- tion of fan-out wafer-level packaging,” IEEE Trans. Compon., Packag., Manuf. Technol., vol. 7, no. 10, pp. 17...
2017
-
[3]
Warpage measurements and characterizations of fan-out wafer-level packaging with large chips and multiple redistributed layers,
J. H. Lau, M. Li, L. Yang, M. Li, I. Xu, T. Chen, S. Chen, Q. X. Yong, J. P. Madhukumar, W. Kai, N. Fan, E. Kuah, Z. Li, K. H. Tan, W. Bao, S. P. Lim, R. Beica, C.-T. Ko, and C. Xi, “Warpage measurements and characterizations of fan-out wafer-level packaging with large chips and multiple redistributed layers,” IEEE Trans. Compon., Packag., Manuf. Technol....
2018
-
[4]
Package warpage reduction for large cowos-r packages,
Y.-H. Hu, C.-H. Lee, J.-S. Peng, H.-Y. Chen, P.-H. Lee, J. Li, J. Shieh, E. Chen, M. Yew, S.-P. Jeng, K. Yan, and J. He, “Package warpage reduction for large cowos-r packages,” in Proc. IEEE Electron. Compon. Technol. Conf., pp. 116–121, 2025
2025
-
[5]
Physical design for heterogeneous integration: Challenges and opportunities,
Y.-W. Chang and D. Z. Pan, “Physical design for heterogeneous integration: Challenges and opportunities,” IEEE Des. Test, vol. 42, no. 6, pp. 15–29, 2025
2025
-
[6]
Analysis of bi-metal thermostats,
S. Timoshenko, “Analysis of bi-metal thermostats,” J. Opt. Soc. Amer, vol. 11, no. 3, pp. 233–255, 1925
1925
-
[7]
Warpage analysis and prediction of the advanced fan-out technology based on process mechanics,
S. Wang, Y. Sun, C. Sheng, Z. Feng, R. Li, L. Xue, J. Liu, and S. Liu, “Warpage analysis and prediction of the advanced fan-out technology based on process mechanics,” IEEE Trans. Compon., Packag., Manuf. Technol., vol. 11, no. 12, pp. 2201– 2213, 2021
2021
-
[8]
Predicted bow of plastic packages of integrated circuit (ic) devices,
E. Suhir, “Predicted bow of plastic packages of integrated circuit (ic) devices,” J. Reinforced Plastics Composites, vol. 12, no. 9, pp. 951–972, 1993
1993
-
[9]
To extend suhir theory for predicting thermally-induced warpage of flip-chip packages,
Y. W. Wang and M. Y. Tsai, “To extend suhir theory for predicting thermally-induced warpage of flip-chip packages,” in Int. Conf. Electron. Mater. Packag. (EMAP), pp. 1–4, 2018
2018
-
[10]
A theoretical solution for thermal warpage of flip-chip packages,
M.-Y. Tsai and Y.-W. Wang, “A theoretical solution for thermal warpage of flip-chip packages,” IEEE Trans. Compon., Packag., Manuf. Technol., vol. 10, no. 1, pp. 72–78, 2020
2020
-
[11]
Application of lamination theory to study warpage across pwb and pwba during convective reflow process,
W. Tan and I. C. Ume, “Application of lamination theory to study warpage across pwb and pwba during convective reflow process,” IEEE Trans. Compon., Packag., Manuf. Technol., vol. 2, no. 2, pp. 217–223, 2012
2012
-
[12]
Efficient high-fidelity two- dimensional warpage modeling for advanced packaging anal- ysis,
S.-Y. Lo, M. Liu, and Y.-W. Chang, “Efficient high-fidelity two- dimensional warpage modeling for advanced packaging anal- ysis,” in Proc. IEEE/ACM Int. Conf. Comput.-Aided Design (ICCAD), (New York, NY, USA), 2024
2024
-
[13]
Efficient high-fidelity warpage modeling for advanced packaging analysis,
S.-Y. Lo, M. Liu, and Y.-W. Chang, “Efficient high-fidelity warpage modeling for advanced packaging analysis,” IEEE Trans. Comput.-Aided Des. Integr. Circuits Syst., pp. 1–1, 2025
2025
-
[14]
More-stress: Model order reduction based efficient numerical algorithm for thermal stress simulation of tsv arrays in 2.5d/3d ic,
T. Zhu, Q. Wang, Y. Lin, R. Wang, and R. Huang, “More-stress: Model order reduction based efficient numerical algorithm for thermal stress simulation of tsv arrays in 2.5d/3d ic,” in Proc. IEEE Des., Autom. Test Eur. Conf., pp. 1–7, 2025
2025
-
[15]
Artificial intelligence-based warpage prediction model for accelerating thermo-mechanical simulation in advanced packaging,
J. Lee, S.-W. Lee, T.-S. Kim, and D. Kwon, “Artificial intelligence-based warpage prediction model for accelerating thermo-mechanical simulation in advanced packaging,” in Proc. IEEE Electron. Compon. Technol. Conf., pp. 1570–1576, 2025
2025
-
[16]
Ai-based prediction of warpage in organic substrates,
J. Zhao, M. Su, and R. Ma, “Ai-based prediction of warpage in organic substrates,” IEEE Access, vol. 13, pp. 142535–142545, 2025
2025
-
[17]
Advanced packaging warpage modeling with deeponet-based operator learning,
S.-Y. Lo, C.-M. Chang, and Y.-W. Chang, “Advanced packaging warpage modeling with deeponet-based operator learning,” in Proc. IEEE/ACM Int. Conf. Comput.-Aided Design (ICCAD), (Munich, Germany), 2025
2025
-
[18]
Physics- informed neural networks: A deep learning framework for solving forward and inverse problems involving nonlinear partial differential equations,
M. Raissi, P. Perdikaris, and G. E. Karniadakis, “Physics- informed neural networks: A deep learning framework for solving forward and inverse problems involving nonlinear partial differential equations,” J. Comput. Phys., vol. 378, pp. 686–707, 2019
2019
-
[19]
Physics-informed machine learning,
G. E. Karniadakis, I. G. Kevrekidis, L. Lu, P. Perdikaris, S. Wang, and L. Yang, “Physics-informed machine learning,” Nat. Rev. Phys., vol. 3, no. 6, pp. 422–440, 2021
2021
-
[20]
Fast full-chip parametric thermal analysis based on enhanced physics enforced neural network,
L. Chen, J. Lu, W. Jin, and S. X.-D. Tan, “Fast full-chip parametric thermal analysis based on enhanced physics enforced neural network,” in Proc. IEEE/ACM Int. Conf. Comput.- Aided Design (ICCAD), pp. 1–8, 2023
2023
-
[21]
Asrr-pinn: Adaptive sub- regional random resampling-based pinn for thermal analysis of 3d-ics,
Z. Zhou, M. Tang, and L. Chen, “Asrr-pinn: Adaptive sub- regional random resampling-based pinn for thermal analysis of 3d-ics,” in Proc. 62nd ACM/IEEE Design Autom. Conf. (DAC), pp. 1–7, 2025
2025
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.