Saved in:
Bibliographic Details
Main Authors: Keramati, Hadi, Kirchen, Patrick, Hannan, Mohammed, Jaiman, Rajeev K.
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2508.01509
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912516716625920
author Keramati, Hadi
Kirchen, Patrick
Hannan, Mohammed
Jaiman, Rajeev K.
author_facet Keramati, Hadi
Kirchen, Patrick
Hannan, Mohammed
Jaiman, Rajeev K.
contents This study presents a generative optimization framework that builds on a fine-tuned diffusion model and reward-directed sampling to generate high-performance engineering designs. The framework adopts a parametric representation of the design geometry and produces new parameter sets corresponding to designs with enhanced performance metrics. A key advantage of the reward-directed approach is its suitability for scenarios in which performance metrics rely on costly engineering simulations or surrogate models (e.g. graph-based, ensemble models, or tree-based) are non-differentiable or prohibitively expensive to differentiate. This work introduces the iterative use of a soft value function within a Markov decision process framework to achieve reward-guided decoding in the diffusion model. By incorporating soft-value guidance during both the training and inference phases, the proposed approach reduces computational and memory costs to achieve high-reward designs, even beyond the training data. Empirical results indicate that this iterative reward-directed method substantially improves the ability of the diffusion models to generate samples with reduced resistance in 3D ship hull design and enhanced hydrodynamic performance in 2D airfoil design tasks. The proposed framework generates samples that extend beyond the training data distribution, resulting in a greater 25 percent reduction in resistance for ship design and over 10 percent improvement in the lift-to-drag ratio for the 2D airfoil design. Successful integration of this model into the engineering design life cycle can enhance both designer productivity and overall design performance.
format Preprint
id arxiv_https___arxiv_org_abs_2508_01509
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle A Reward-Directed Diffusion Framework for Generative Design Optimization
Keramati, Hadi
Kirchen, Patrick
Hannan, Mohammed
Jaiman, Rajeev K.
Machine Learning
Computational Engineering, Finance, and Science
This study presents a generative optimization framework that builds on a fine-tuned diffusion model and reward-directed sampling to generate high-performance engineering designs. The framework adopts a parametric representation of the design geometry and produces new parameter sets corresponding to designs with enhanced performance metrics. A key advantage of the reward-directed approach is its suitability for scenarios in which performance metrics rely on costly engineering simulations or surrogate models (e.g. graph-based, ensemble models, or tree-based) are non-differentiable or prohibitively expensive to differentiate. This work introduces the iterative use of a soft value function within a Markov decision process framework to achieve reward-guided decoding in the diffusion model. By incorporating soft-value guidance during both the training and inference phases, the proposed approach reduces computational and memory costs to achieve high-reward designs, even beyond the training data. Empirical results indicate that this iterative reward-directed method substantially improves the ability of the diffusion models to generate samples with reduced resistance in 3D ship hull design and enhanced hydrodynamic performance in 2D airfoil design tasks. The proposed framework generates samples that extend beyond the training data distribution, resulting in a greater 25 percent reduction in resistance for ship design and over 10 percent improvement in the lift-to-drag ratio for the 2D airfoil design. Successful integration of this model into the engineering design life cycle can enhance both designer productivity and overall design performance.
title A Reward-Directed Diffusion Framework for Generative Design Optimization
topic Machine Learning
Computational Engineering, Finance, and Science
url https://arxiv.org/abs/2508.01509