Search papers, labs, and topics across Lattice.
The authors diagnose the failure of scaling all-atom crystal generation to large systems as a particle-space correspondence bottleneck and resolve it with GLASS, a framework combining a permutation-invariant global latent autoencoder, learned-slot set decoding, and flow matching. By offloading atomic correspondence to the slot decoder rather than computing particle-space transport trajectories, the model bypasses the combinatorial complexity that degrades equivariant flows at high particle counts and densities. Across MP20 and QMOF, GLASS approaches the validity of empirical training data across all system sizes and generates unconditioned metal–organic frameworks up to 150 atoms per unit cell.
Scaling all-atom crystal generation to 150-atom metal–organic frameworks doesn't require rigid chemical building blocks or topology priors—it just requires decoupling atomic correspondence from generative transport.
Generative models for crystals enable the discovery of novel structures, but scaling all-atom generation to larger systems such as metal--organic frameworks remains challenging. We connect this difficulty to the correspondence problem of particle-space generation. Even on a single fixed target set, index-free permutation-equivariant particle flows require substantially more training for reliable generation as set size and density increase, under both independent and optimal-transport couplings. To resolve this challenge, we introduce GLASS---Global Latent Aggregation with Slot-based Set Decoding, which encodes structures in a permutation-invariant global latent space and learns their distribution via flow matching. A learned-slot decoder constructs all atoms in parallel, removing atom-wise correspondence from generative transport. On MP20, GLASS is competitive with particle-space models, and flow training can reach the validity of the training data at every structure size. On a QMOF subset, GLASS generates MOFs with up to 150 atoms per unit cell without conditioning on building blocks, topology, or composition, and approaches the structural validity of the training data. On both datasets, flow training exposes a validity--novelty tradeoff, and MOF novelty remains limited by autoencoder generalization on the available data. These results show that separating correspondence assignment from generative transport provides a simple route toward high-validity generation of larger atomistic systems.