Style-constrained inverse design of microstructures with tailored mechanical properties using unconditional diffusion models

TL;DR

This paper introduces an inverse design framework using unconditional diffusion models and differentiable programming to generate microstructures with tailored mechanical properties without extensive retraining.

Contribution

It presents a novel approach that reinterprets the noise input as an optimizable variable, eliminating the need for labeled datasets and retraining for different design tasks.

Findings

Successfully designed microstructures with specified properties.

Demonstrated control over style and performance in generated microstructures.

Showcased versatility across various mechanical behavior targets.

Abstract

Deep generative models, particularly denoising diffusion models, have achieved remarkable success in high-fidelity generation of architected microstructures with desired properties and styles. Nevertheless, these recent methods typically rely on conditional training mechanisms and demand substantial computational effort to prepare the labeled training dataset, which makes them inflexible since any change in the governing equations or boundary conditions requires a complete retraining process. In this study, we propose a new inverse design framework that integrates unconditional denoising diffusion models with differentiable programming techniques for architected microstructure generation. Our approach eliminates the need for expensive labeled dataset preparation and retraining for different problem settings. By reinterpreting the noise input to the diffusion model as an optimizable design variable, we formulate the design task as an optimization problem over the noise input, enabling control over the reverse denoising trajectory to guide the generated microstructure toward the desired mechanical properties while preserving the stylistic constraints encoded in the training dataset. A unified differentiation pipeline via vector-Jacobian product concatenations is developed to enable end-to-end gradient evaluation through backpropagation. Several numerical examples, ranging from the design of microstructures with specified homogenized properties to those with targeted hyperelastic and elasto-plastic behaviors, showcase the effectiveness of the framework and its potential for advanced design tasks involving diverse performance and style requirements.