An unsupervised learning approach to solving heat equations on chip based on Auto Encoder and Image Gradient

Abstract

Solving heat transfer equations on chip becomes very critical in the upcoming 5G and AI chip-package-systems. However, batches of simulations have to be performed for data driven supervised machine learning models. Data driven methods are data hungry, to address this, Physics Informed Neural Networks (PINN) have been proposed. However, vanilla PINN models solve one fixed heat equation at a time, so the models have to be retrained for heat equations with different source terms. Additionally, issues related to multi-objective optimization have to be resolved while using PINN to minimize the PDE residual, satisfy boundary conditions and fit the observed data etc. Therefore, this paper investigates an unsupervised learning approach for solving heat transfer equations on chip without using solution data and generalizing the trained network for predicting solutions for heat equations with unseen source terms. Specifically, a hybrid framework of Auto Encoder (AE) and Image Gradient (IG) based network is designed. The AE is used to encode different source terms of the heat equations. The IG based network implements a second order central difference algorithm for structured grids and minimizes the PDE residual. The effectiveness of the designed network is evaluated by solving heat equations for various use cases. It is proved that with limited number of source terms to train the AE network, the framework can not only solve the given heat transfer problems with a single training process, but also make reasonable predictions for unseen cases (heat equations with new source terms) without retraining.

18 pages, 13 figures

References in corpus (9)

• DeepXDE: A deep learning library for solving differential equations
• DGM: A deep learning algorithm for solving partial differential equations
• Physics Informed Deep Learning (Part I): Data-driven Solutions of Nonlinear Partial Differential Equations
• Data-Driven Deep Learning of Partial Differential Equations in Modal Space
• Solving inverse-PDE problems with physics-aware neural networks
• Physics-informed Autoencoders for Lyapunov-stable Fluid Flow Prediction
• Solving Bayesian Inverse Problems via Variational Autoencoders
• A machine learning framework for data driven acceleration of computations of differential equations
• Zero-shot User Intent Detection via Capsule Neural Networks

Cited by in corpus (5)

• Interpretable and Explainable Machine Learning for Materials Science and Chemistry
• Conditional physics informed neural networks
• A composable autoencoder-based iterative algorithm for accelerating numerical simulations
• Geometry encoding for numerical simulations
• A Latent space solver for PDE generalization