用扩散模型生成高效散热器,压降降低10%且无需重新训练。
HeatGen: A Guided Diffusion Framework for Multiphysics Heat Sink Design Optimization
- 用扩散模型结合代理梯度引导设计,自动优化散热结构。
- 生成的散热器压降比传统方法低10%,表面温度达标。
- 一次训练后可快速适配新温控需求,适合工业设计场景。
本文提出一种基于引导去噪扩散概率模型(DDPM)的生成优化框架,利用代理梯度生成满足压降最小且表面温度低于阈值的散热器结构。几何体采用多鳍片边界表示,通过多保真度方法生成训练数据。以边界表示向量与几何数据联合训练去噪扩散模型,生成符合数据特征的散热器。训练两个残差神经网络分别预测压降与表面温度,并利用其对设计变量的梯度,在推理阶段引导生成过程,实现低压降与防过热双重优化。相比传统黑箱优化方法如CMA-ES,生成样本压降最高降低10%。该方法在数据充足时具备良好可扩展性,且一旦模型训练完成并保存热沉世界模型,即可在不重训的情况下快速响应新温度约束,适用于电子冷却领域的高效设计生成。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。