Composable Crystals: Controllable Materials Discovery via Concept Learning

• URL: http://arxiv.org/abs/2605.14769v1
• Date: Thu, 14 May 2026 12:36:23 GMT
• Title: Composable Crystals: Controllable Materials Discovery via Concept Learning
• Abstract summary: We introduce a concept-based compositional framework for crystal generation.<n>We train a vector-quantized variational autoencoder to automatically discover a shared set of reusable crystal concepts.<n>By recombining such concepts, our framework enables controllable exploration of novel crystals beyond the training distribution.
• Score: 25.01951802810537
• License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
• Abstract: 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.

Related papers

• Symmetry-Breaking De Novo Crystal Generation via Markovian Jump Diffusion [50.136954476803766]
  We propose a novel diffusion-based framework that generates full structure specifications by reversing from the lowest-symmetry priors.<n>Our model, dubbed emphSymmetry-breaking Crystal Diffusion (SbCD), introduces a principled approach to explicitly incorporate inter-space-group transitions into the generative process.<n>In de novo generation experiments on MP20 and MPTS-52, SbCD outperforms its symmetry-preserving counterpart by a substantial margin.
  arXiv  Detail & Related papers  (2026-08-13T16:41:48Z)
• CrystalREPA: Transferring Physical Priors from Universal MLIPs to Crystal Generative Models [10.446104330081189]
  We show a substantial representation gap between state-of-the-art crystal generative models and pretrained universal machine learning interatomic potentials.<n>We propose Crystal REPresentation Alignment (CrystalREPA), a plug-and-play framework that aligns the atom-wise hidden states of generative encoders with frozen MLIP representations.<n>CrystalREPA consistently improves the thermodynamic stability, structural validity, and structural fidelity of generated crystals.
  arXiv  Detail & Related papers  (2026-05-09T13:56:23Z)
• OXtal: An All-Atom Diffusion Model for Organic Crystal Structure Prediction [63.318434943975255]
  We introduce OXtal, a large-scale 100M parameter all-atom diffusion model that learns the conditional joint distribution over intramolecular conformations and periodic packing.<n>By leveraging a large dataset of 600K experimentally validated crystal structures, OXtal achieves orders-of-improvement over prior ab initio machine learning CSP methods.<n> OXtal attains over 80% packing similarity rate, demonstrating its ability to model both thermodynamic and kinetic regularities of molecular crystallization.
  arXiv  Detail & Related papers  (2025-12-07T20:46:30Z)
• CrystalICL: Enabling In-Context Learning for Crystal Generation [12.641999605656409]
  Large language models (LLMs) have demonstrated strong in-context learning (ICL) capabilities.<n>Existing LLM-based crystal generation approaches are limited to zero-shot scenarios and are unable to benefit from few-shot scenarios.<n>We propose CrystalICL, a novel model designed for few-shot crystal generation.
  arXiv  Detail & Related papers  (2025-08-27T07:49:27Z)
• CrystalDiT: A Diffusion Transformer for Crystal Generation [52.45780803467369]
  We present CrystalDiT, a diffusion transformer for crystal structure generation that achieves state-of-the-art performance.<n>CrystalDiT employs a unified transformer that imposes a powerful inductive bias: treating lattice and atomic properties as a single, interdependent system.
  arXiv  Detail & Related papers  (2025-08-13T12:53:32Z)
• A Periodic Bayesian Flow for Material Generation [20.62877413439857]
  We introduce CrysBFN, a novel crystal generation method by proposing a periodic Bayesian flow.<n>To successfully realize the concept of periodic Bayesian flow, CrysBFN integrates a new entropy conditioning mechanism.<n>Experiments over both crystal ab initio generation and crystal structure prediction tasks demonstrate the superiority of CrysBFN.
  arXiv  Detail & Related papers  (2025-02-04T05:07:13Z)
• Complete and Efficient Graph Transformers for Crystal Material Property Prediction [53.32754046881189]
  Crystal structures are characterized by atomic bases within a primitive unit cell that repeats along a regular lattice throughout 3D space. We introduce a novel approach that utilizes the periodic patterns of unit cells to establish the lattice-based representation for each atom. We propose ComFormer, a SE(3) transformer designed specifically for crystalline materials.
  arXiv  Detail & Related papers  (2024-03-18T15:06:37Z)
• Scalable Diffusion for Materials Generation [99.71001883652211]
  We develop a unified crystal representation that can represent any crystal structure (UniMat) UniMat can generate high fidelity crystal structures from larger and more complex chemical systems. We propose additional metrics for evaluating generative models of materials.
  arXiv  Detail & Related papers  (2023-10-18T15:49:39Z)
• Latent Conservative Objective Models for Data-Driven Crystal Structure Prediction [62.36797874900395]
  In computational chemistry, crystal structure prediction is an optimization problem. One approach to tackle this problem involves building simulators based on density functional theory (DFT) followed by running search in simulation. We show that our approach, dubbed LCOMs (latent conservative objective models), performs comparably to the best current approaches in terms of success rate of structure prediction.
  arXiv  Detail & Related papers  (2023-10-16T04:35:44Z)
• Data-Driven Score-Based Models for Generating Stable Structures with Adaptive Crystal Cells [1.515687944002438]
  This work aims at the generation of new crystal structures with desired properties, such as chemical stability and specified chemical composition. The novelty of the presented approach resides in the fact that the lattice of the crystal cell is not fixed. A multigraph crystal representation is introduced that respects symmetry constraints, yielding computational advantages.
  arXiv  Detail & Related papers  (2023-10-16T02:53:24Z)
• Crystal-GFN: sampling crystals with desirable properties and constraints [103.79058968784163]
  We introduce Crystal-GFN, a generative model of crystal structures that sequentially samples structural properties of crystalline materials. In this paper, we use as objective the formation energy per atom of a crystal structure predicted by a new proxy machine learning model trained on MatBench. The results demonstrate that Crystal-GFN is able to sample highly diverse crystals with low (median -3.1 eV/atom) predicted formation energy.
  arXiv  Detail & Related papers  (2023-10-07T21:36:55Z)
• An invertible crystallographic representation for general inverse design of inorganic crystals with targeted properties [10.853822721106205]
  We present a framework capable of general inverse design (not limited to a given set of elements or crystal structures) The framework generates new crystals with user-defined formation energies, bandgap, thermoelectric (TE) power factor, and combinations thereof. Results represent a significant step toward property-driven general inverse design using generative models.
  arXiv  Detail & Related papers  (2020-05-15T15:58:31Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.