Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
MIRRA, a discrete flow-matching generative model, addresses the challenge of high-fidelity molecular generation by leveraging chemical knowledge to inform identity choices and optimize sampling efficiency. It achieves this by learning to share information across related chemical building blocks and deferring identity probability computations until necessary, thereby avoiding wasted predictions. MIRRA significantly outperforms state-of-the-art baselines in both chemical and graph distribution accuracy on ZINC250k, while also demonstrating substantial speedups and improved property-targeted generation and polymer validity.
Molecular generation models can now achieve 8.6x speedups and drastically reduce distribution errors by deferring identity probability calculations until needed, while retaining access to a full chemical vocabulary.
Fast, high-fidelity molecular generation requires accurate chemical choices and efficient sampling. Chemical building blocks reduce graph size, but make each node a choice among tens of thousands of related identities. We introduce MIRRA (Molecular Identity Representation, Refresh and Assembly), a discrete flow-matching generator that learns from chemical knowledge and evaluates identity probabilities only when needed, while keeping every identity in its vocabulary available. Chemical features share information across related blocks, and learned identity corrections distinguish them in molecular context. By deciding which positions will refresh before computing their probabilities, the sampler avoids unused predictions while preserving each sampling transition. Against the strongest baselines in each task, MIRRA reduces chemical and graph distribution errors on ZINC250k by 29-31% and 53-60%, respectively, with generation up to 8.6 times faster. Property targeting yields an estimated 7.8-20.8 times more distinct novel hits per second. On a held-out polymer test set, MIRRA achieves 100% periodic validity with known chain topology, versus 57.0% for a polymer variational autoencoder. These results demonstrate fast, high-fidelity generation with the full chemical vocabulary available throughout sampling.