Skip to yearly menu bar Skip to main content


Poster
in
Workshop: AI4MAT-ICLR-2025: AI for Accelerated Materials Design

MatInvent: Reinforcement Learning for 3D Crystal Diffusion Generation

Junwu Chen · Jeff Guo · Philippe Schwaller

Keywords: [ diffusion model ] [ crystal generation ] [ reinforcement learning ]


Abstract:

Recent advances in diffusion models have enabled increasing capabilities for inverse materials design. The key capability is achieving tailored design towards desired property profiles, with wide applications for climate change, semiconductor design, and catalysis. In this work, we present MatInvent, a reinforcement learning (RL) framework tailored to optimize diffusion models for goal-directed crystal generation. By formulating equivariant denoising as a multi-step decision-making problem, MatInvent leverages policy optimization with reward-weighted KL regularization, including experience replay and diversity filters to enhance sample efficiency and diversity. Experimental results demonstrate that MatInvent outperforms existing baselines, offering an effective strategy for crystal generation with single or multiple property optimization.

Chat is not available.