Saved in:
Bibliographic Details
Main Authors: Liu, Nian, Zeng, Yuwei, Kubo, Ryoji, Kazeev, Nikita, Dale, Stephen Gregory, Maevskiy, Artem, Huang, Pengru, Laurent, Thomas, Novoselov, Kostya S., Bresson, Xavier
Format: Preprint
Published: 2026
Subjects:
Online Access:https://arxiv.org/abs/2605.14769
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910253372669952
author Liu, Nian
Zeng, Yuwei
Kubo, Ryoji
Kazeev, Nikita
Dale, Stephen Gregory
Maevskiy, Artem
Huang, Pengru
Laurent, Thomas
Novoselov, Kostya S.
Bresson, Xavier
author_facet Liu, Nian
Zeng, Yuwei
Kubo, Ryoji
Kazeev, Nikita
Dale, Stephen Gregory
Maevskiy, Artem
Huang, Pengru
Laurent, Thomas
Novoselov, Kostya S.
Bresson, Xavier
contents De novo crystal generation, a central task in materials discovery, aims to generate crystals that are simultaneously valid, stable, unique, and novel. Existing methods mainly rely on black-box stochastic sampling, providing limited control over how generated structures move beyond the observed distribution. In this paper, we introduce a concept-based compositional framework for crystal generation. We train a vector-quantized variational autoencoder to automatically discover a shared set of reusable crystal concepts, which serve as building blocks for guided generation. These learned concepts naturally exhibit interpretability from both local atomic environments and global symmetry patterns, and generalize to crystals from different distributions. By recombining such concepts, our framework enables controllable exploration of novel crystals beyond the training distribution, rather than relying solely on unconstrained random sampling. To further improve composition efficiency, we introduce a composition generator and iteratively refine it using high-quality samples generated by the model itself. The resulting concept compositions are then used to condition downstream crystal generation. Numerical experiments on MP-20 and Alex-MP-20 show that compositing concepts separately increase base model up to 53.2% and 51.7% on V.S.U.N metric, with particular gains in novelty.
format Preprint
id arxiv_https___arxiv_org_abs_2605_14769
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Composable Crystals: Controllable Materials Discovery via Concept Learning
Liu, Nian
Zeng, Yuwei
Kubo, Ryoji
Kazeev, Nikita
Dale, Stephen Gregory
Maevskiy, Artem
Huang, Pengru
Laurent, Thomas
Novoselov, Kostya S.
Bresson, Xavier
Machine Learning
De novo crystal generation, a central task in materials discovery, aims to generate crystals that are simultaneously valid, stable, unique, and novel. Existing methods mainly rely on black-box stochastic sampling, providing limited control over how generated structures move beyond the observed distribution. In this paper, we introduce a concept-based compositional framework for crystal generation. We train a vector-quantized variational autoencoder to automatically discover a shared set of reusable crystal concepts, which serve as building blocks for guided generation. These learned concepts naturally exhibit interpretability from both local atomic environments and global symmetry patterns, and generalize to crystals from different distributions. By recombining such concepts, our framework enables controllable exploration of novel crystals beyond the training distribution, rather than relying solely on unconstrained random sampling. To further improve composition efficiency, we introduce a composition generator and iteratively refine it using high-quality samples generated by the model itself. The resulting concept compositions are then used to condition downstream crystal generation. Numerical experiments on MP-20 and Alex-MP-20 show that compositing concepts separately increase base model up to 53.2% and 51.7% on V.S.U.N metric, with particular gains in novelty.
title Composable Crystals: Controllable Materials Discovery via Concept Learning
topic Machine Learning
url https://arxiv.org/abs/2605.14769