HeatGen: A Guided Diffusion Framework for Multiphysics Heat Sink Design Optimization

• URL: http://arxiv.org/abs/2511.09578v1
• Date: Fri, 14 Nov 2025 01:00:35 GMT
• Title: HeatGen: A Guided Diffusion Framework for Multiphysics Heat Sink Design Optimization
• Authors: Hadi Keramati, Morteza Sadeghi, Rajeev K. Jaiman,
• Abstract summary: Geometries are represented using boundary representations of multiple fins.<n>We train a denoising diffusion probabilistic model to generate heat sinks with characteristics consistent with those observed in the data.<n>We train two different residual neural networks to predict the pressure drop and surface temperature for each geometry.
• Score: 0.0
• License: http://creativecommons.org/licenses/by/4.0/
• Abstract: This study presents a generative optimization framework based on a guided denoising diffusion probabilistic model (DDPM) that leverages surrogate gradients to generate heat sink designs minimizing pressure drop while maintaining surface temperatures below a specified threshold. Geometries are represented using boundary representations of multiple fins, and a multi-fidelity approach is employed to generate training data. Using this dataset, along with vectors representing the boundary representation geometries, we train a denoising diffusion probabilistic model to generate heat sinks with characteristics consistent with those observed in the data. We train two different residual neural networks to predict the pressure drop and surface temperature for each geometry. We use the gradients of these surrogate models with respect to the design variables to guide the geometry generation process toward satisfying the low-pressure and surface temperature constraints. This inference-time guidance directs the generative process toward heat sink designs that not only prevent overheating but also achieve lower pressure drops compared to traditional optimization methods such as CMA-ES. In contrast to traditional black-box optimization approaches, our method is scalable, provided sufficient training data is available. Unlike traditional topology optimization methods, once the model is trained and the heat sink world model is saved, inference under new constraints (e.g., temperature) is computationally inexpensive and does not require retraining. Samples generated using the guided diffusion model achieve pressure drops up to 10 percent lower than the limits obtained by traditional black-box optimization methods. This work represents a step toward building a foundational generative model for electronics cooling.

Related papers

• Machine Learning for Electron-Scale Turbulence Modeling in W7-X [35.18016233072556]
  This paper presents machine-learning-driven reduced models for turbulence in the Wendelstein 7-X stellarator.<n>Each model predicts the ETG heat flux as a function of three plasma parameters.<n>Our models demonstrate robust performance and predictive accuracy comparable to the original reference simulations.
  arXiv  Detail & Related papers  (2025-11-06T17:24:37Z)
• Equilibrium Matching: Generative Modeling with Implicit Energy-Based Models [52.74448905289362]
  EqM is a generative modeling framework built from an equilibrium dynamics perspective.<n>By replacing time-conditional velocities with a unified equilibrium landscape, EqM offers a tighter bridge between flow and energy-based models.
  arXiv  Detail & Related papers  (2025-10-02T17:59:06Z)
• Multi-Stage Graph Neural Networks for Data-Driven Prediction of Natural Convection in Enclosed Cavities [0.0]
  Graph Neural Networks (GNNs) offer new alternatives for learning thermal-fluid behavior directly from simulation data.<n>We propose a novel multi-stage GNN architecture that leverages hierarchical pooling and unpooling operations to progressively model global-to-local interactions.<n> Experimental results demonstrate that the proposed model achieves higher predictive accuracy, improved training efficiency, and reduced long-term error accumulation.
  arXiv  Detail & Related papers  (2025-09-07T13:05:39Z)
• Progressive Inference-Time Annealing of Diffusion Models for Sampling from Boltzmann Densities [93.13866975467549]
  We propose Progressive Inference-Time Annealing (PITA) to learn diffusion-based samplers.<n>PITA combines two complementary techniques: Annealing of the Boltzmann distribution and Diffusion smoothing.<n>It enables equilibrium sampling of N-body particle systems, Alanine Dipeptide, and tripeptides in Cartesian coordinates.
  arXiv  Detail & Related papers  (2025-06-19T17:14:22Z)
• PearSAN: A Machine Learning Method for Inverse Design using Pearson Correlated Surrogate Annealing [66.27103948750306]
  PearSAN is a machine learning-assisted optimization algorithm applicable to inverse design problems with large design spaces.<n>It uses a Pearson correlated surrogate model to predict the figure of merit of the true design metric.<n>It achieves a state-of-the-art maximum design efficiency of 97%, and is at least an order of magnitude faster than previous methods.
  arXiv  Detail & Related papers  (2024-12-26T17:02:19Z)
• Point Cloud Resampling with Learnable Heat Diffusion [58.050130177241186]
  We propose a learnable heat diffusion framework for point cloud resampling. Unlike previous diffusion models with a fixed prior, the adaptive conditional prior selectively preserves geometric features of the point cloud.
  arXiv  Detail & Related papers  (2024-11-21T13:44:18Z)
• Bridging Model-Based Optimization and Generative Modeling via Conservative Fine-Tuning of Diffusion Models [54.132297393662654]
  We introduce a hybrid method that fine-tunes cutting-edge diffusion models by optimizing reward models through RL. We demonstrate the capability of our approach to outperform the best designs in offline data, leveraging the extrapolation capabilities of reward models.
  arXiv  Detail & Related papers  (2024-05-30T03:57:29Z)
• Deep encoder-decoder hierarchical convolutional neural networks for conjugate heat transfer surrogate modeling [0.0]
  Conjugate heat transfer (CHT) analyses are vital for the design of many energy systems.<n>High-fidelity CHT numerical simulations are computationally intensive.<n>We develop a modular deep encoder-decoder hierarchical (DeepEDH) convolutional neural network for CHT analyses.
  arXiv  Detail & Related papers  (2023-11-24T21:45:11Z)
• Machine learning based surrogate models for microchannel heat sink optimization [0.0]
  In this paper, microchannel designs with secondary channels and with ribs are investigated using computational fluid dynamics. A workflow that combines Latin hypercube sampling, machine learning-based surrogate modeling and multi-objective optimization is proposed.
  arXiv  Detail & Related papers  (2022-08-20T13:49:11Z)
• A physics and data co-driven surrogate modeling approach for temperature field prediction on irregular geometric domain [12.264200001067797]
  We propose a novel physics and data co-driven surrogate modeling method for temperature field prediction. Numerical results demonstrate that our method can significantly improve accuracy prediction on a smaller dataset.
  arXiv  Detail & Related papers  (2022-03-15T08:43:24Z)
• Heat Conduction Plate Layout Optimization using Physics-driven Convolutional Neural Networks [14.198900757461555]
  The layout optimization of the heat conduction is essential during design in engineering, especially for sensible thermal products. Data-driven approaches are used to train a surrogate model as a mapping between the prescribed external loads and various geometry. This paper proposes a Physics-driven Convolutional Neural Networks (PD-CNN) method to infer the physical field solutions for varied loading cases.
  arXiv  Detail & Related papers  (2022-01-21T10:43:57Z)
• A Near-Optimal Gradient Flow for Learning Neural Energy-Based Models [93.24030378630175]
  We propose a novel numerical scheme to optimize the gradient flows for learning energy-based models (EBMs) We derive a second-order Wasserstein gradient flow of the global relative entropy from Fokker-Planck equation. Compared with existing schemes, Wasserstein gradient flow is a smoother and near-optimal numerical scheme to approximate real data densities.
  arXiv  Detail & Related papers  (2019-10-31T02:26:20Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.