arXiv AI

AtomSteer: Test-Time Constraint Steering for Atomistic Generative Models

AtomSteer is a test‑time inference framework that steers pretrained atomistic generative models to satisfy structural constraints such as crystallographic symmetry, geometry, and stereochemistry. It applies bounded local projections, graph‑rigid stereochemical transformations, and periodic symmetry projections during sampling, accepting updates that improve constraint satisfaction while preserving the rest of the structure. The method is demonstrated on three backbones—FlowMM, ET‑Flow, and Boltz‑2—across tasks including de novo crystal generation, crystal structure prediction, molecular conformer generation, and protein‑ligand complex prediction, yielding improvements in space‑group coverage, stereochemical accuracy, and ligand E/Z error rates.

arXiv Machine Learning
Aug 28

Packora: Systematic Design for Generative Molecular Crystal Structure Prediction

Packora is a flow-based generative model designed for molecular crystal structure prediction (CSP). It jointly predicts atomic coordinates and lattice parameters from molecular graphs, supporting multi-component and organometallic crystals and allowing conditioning on conformers, stereochemistry, and space-group data. In evaluations inspired by the CCDC CSP blind test, Packora outperforms baselines on generation and ranking benchmarks, achieving superior matched-budget coverage, higher experimental-form recovery, lower ranks, and faster convergence.

By Nayoung Kim, Kiyoung Seong, Sungsoo Ahn
arXiv AI
Oct 7

Inference-Time Projection for Physically Valid Biomolecular Diffusion Models

The paper introduces two closed‑form projection operators that enforce physical validity in biomolecular diffusion models during inference. One operator resolves inter‑chain van der Waals clashes, while the other restores ligand geometry, including bond lengths, angles, planarity, and chirality. These lightweight, model‑agnostic corrections preserve structural accuracy and ligand placement across multiple benchmarks without adding runtime or memory overhead.

By Qurat-ul-ain, Yee Whye Teh, Charlotte M. Deane, Matteo Cagiada
arXiv AI
Oct 9

Zatom-2: Multitask Pretraining on Atomistic Data for Generative Modeling across Domains

Zatom-2 is a multitask generative model for atomistic data that has been pretrained on about five million structures from the OMol25 and OMat24 datasets. It uses a multiscale Transformer with conditional flow matching to support tasks such as generation, structure prediction, and energy/force prediction for both molecules and materials. The model outperforms its predecessor, Zatom-1, on molecular distribution fidelity and benchmark generation tasks, and improves protein backbone designability from 67.8% to 74.8% after finetuning on 2,000 protein domains.

By Miruna Cretu, Alex Abrudan, Antonia Panescu, Tynan Perez, Rishabh Anand, N. Benjamin Erichson, Michael W. Mahoney, Samuel Blau, Joseph Jacobson, Rafael G\'omez-Bombarelli, Rex Ying, Tuomas Knowles, Pietro Li\`o, Alex Morehead
arXiv Machine Learning
Jun 9

Few-step Cofolding with All-Atom Flow Maps

arXiv:2606. 08375v1 Announce Type: new Abstract: All-atom generative modeling of 3D biomolecular complexes has emerged as the dominant paradigm for predicting the structure of proteins and protein-ligand systems.

By Gianluca Scarpellini, Ron Shprints, Peter Holderrieth, Juno Nam, Pranav Murugan, Rafael G\'omez-Bombarelli, Tommi Jaakola, Maruan Al-Shedivat, Nicholas Matthew Boffi, Avishek Joey Bose
arXiv Machine Learning
Oct 2

Riemannian Flow Models with Reinforcement Learning for Molecular Crystal Structure Prediction

The paper introduces CG-OMatG, an equivariant Riemannian flow-based generative model that predicts molecular crystal structures using a coarse-grained, hierarchical representation. It treats molecules as rigid bodies, performs inter- and intra-molecular message passing, and learns to reconstruct molecule centroids, orientations, and lattice parameters conditioned on chemical species and conformer geometry. The model is trained on OMC25 and CSD datasets, fine-tuned with policy gradient reinforcement learning to favor low-energy structures, and validated against a CSP blind test benchmark using COMPACK packing-similarity analysis.

By Thomas Egg, Harry Winston Sullivan, Maya M. Martirossyan, Philipp H\"ollmer, Cheng Zeng, Adrian Roitberg, Mingjie Liu, Richard Hennig, Sapna Sarupria, Ellad B. Tadmor, Stefano Martiniani
arXiv AI
Aug 19

Leveraging generative hallucination and biophysics-informed modeling for unified biomolecular sequence-structure co-design

The paper introduces MCTH (Monte Carlo Tree Hallucination), an inference-only framework that performs all‑atom biomolecular sequence‑structure co‑design by treating pretrained folding and inverse‑folding models as black‑box operators. MCTH uses Monte Carlo Tree Search to allocate a fixed inference budget across competing design trajectories, incorporating model confidence, uncertainty, and cross‑expert consensus. Experiments across protein‑RNA, protein‑DNA, protein‑protein, and protein‑ligand design show that adaptive search outperforms simpler sampling strategies, and evaluations with AlphaFold3 and Chai‑1 demonstrate transferability beyond the search‑time oracle.

By Xuefeng Liu, Mingxuan Cao, Xiao Luo, Songhao Jiang, Tobin Sosnick, Jinbo Xu, Louis Maher, Rick Stevens
arXiv AI
Sep 30

Where Should Physics Enter a Molecular Crystal Generator?

The paper introduces CrystAF, an all‑atom crystal flow‑map generation model, and evaluates where physics should be incorporated into generative crystal structure models. By applying physics‑informed post‑training, the authors improve molecular validity and crystal packing without altering sampling speed, while inference‑time corrections further refine the structures. The study demonstrates that post‑training and inference‑time physics are complementary, and that the post‑training approach transfers to other generators such as Clari‑M and MolCrystalFlow.

By Haocheng Tang, Junmei Wang, Wengong Jin
arXiv AI
Jun 19

Emyx: Fast and efficient all-atom protein generation

arXiv:2606. 19377v1 Announce Type: cross Abstract: Computational enzyme design requires generating proteins that scaffold catalytic residues and ligands, a task that demands both geometric accuracy and structural diversity from the underlying generative model.

By Nicholas J. Williams, Ward Haddadin, Matteo P. Ferla, Constantin Schneider, Nicholas B. Woodall, Ruby Sedgwick, Christian D. Madsen, Andrew L. Hopkins, Edward O. Pyzer-Knapp