Energy-Based Flow Matching for Generating 3D Molecular Structure

• URL: http://arxiv.org/abs/2508.18949v1
• Date: Tue, 26 Aug 2025 11:42:57 GMT
• Title: Energy-Based Flow Matching for Generating 3D Molecular Structure
• Authors: Wenyin Zhou, Christopher Iliffe Sprague, Vsevolod Viliuga, Matteo Tadiello, Arne Elofsson, Hossein Azizpour,
• Abstract summary: We focus on flow matching and adopt an energy-based perspective to improve training and inference of structure generation models.<n>Our view results in a mapping function, that is directly learned to textititeratively map random configurations.<n>Experiments on protein docking and protein backbone generation consistently demonstrate the method's effectiveness.
• Score: 7.055961224018033
• License: http://creativecommons.org/licenses/by/4.0/
• Abstract: Molecular structure generation is a fundamental problem that involves determining the 3D positions of molecules' constituents. It has crucial biological applications, such as molecular docking, protein folding, and molecular design. Recent advances in generative modeling, such as diffusion models and flow matching, have made great progress on these tasks by modeling molecular conformations as a distribution. In this work, we focus on flow matching and adopt an energy-based perspective to improve training and inference of structure generation models. Our view results in a mapping function, represented by a deep network, that is directly learned to \textit{iteratively} map random configurations, i.e. samples from the source distribution, to target structures, i.e. points in the data manifold. This yields a conceptually simple and empirically effective flow matching setup that is theoretically justified and has interesting connections to fundamental properties such as idempotency and stability, as well as the empirically useful techniques such as structure refinement in AlphaFold. Experiments on protein docking as well as protein backbone generation consistently demonstrate the method's effectiveness, where it outperforms recent baselines of task-associated flow matching and diffusion models, using a similar computational budget.

Related papers

• Zatom-1: A Multimodal Flow Foundation Model for 3D Molecules and Materials [51.342983349686556]
  General-purpose 3D chemical modeling encompasses molecules and materials, requiring both generative and predictive capabilities.<n>We introduce Zatom-1, the first end-to-end, fully open-source foundation model that unifies generative and predictive learning of 3D molecules and materials.
  arXiv  Detail & Related papers  (2026-02-24T20:52:39Z)
• Molecular Representations in Implicit Functional Space via Hyper-Networks [53.70982267248536]
  We argue that molecular learning can instead be formulated as learning in function space.<n>We instantiate this formulation with MolField, a hyper-network-based framework that learns distributions over molecular fields.<n>Our results show that treating molecules as continuous functions fundamentally changes how molecular representations generalize across tasks.
  arXiv  Detail & Related papers  (2026-01-29T21:13:37Z)
• FlexiFlow: decomposable flow matching for generation of flexible molecular ensemble [43.16267651837197]
  FlexiFlow is a novel architecture that extends flow-matching models.<n>We demonstrate the effectiveness of our approach on the QM9 and GEOM Drugs datasets.
  arXiv  Detail & Related papers  (2025-11-21T13:50:54Z)
• Guiding Diffusion Models with Reinforcement Learning for Stable Molecule Generation [16.01877423456416]
  Reinforcement Learning with Physical Feedback (RLPF) is a novel framework that extends Denoising Diffusion Policy Optimization to 3D molecular generation.<n>RLPF introduces reward functions derived from force-field evaluations to guide the generation toward energetically stable and physically meaningful structures.<n> Experiments on the QM9 and GEOM-drug datasets demonstrate that RLPF significantly improves molecular stability compared to existing methods.
  arXiv  Detail & Related papers  (2025-08-22T16:44:55Z)
• FlowMol3: Flow Matching for 3D De Novo Small-Molecule Generation [0.0]
  FlowMol3 is an open-source, multi-modal flow matching model that advances the state of the art for all-atom, small-molecule generation.<n>Our results highlight simple, transferable strategies for improving the stability and quality of diffusion- and flow-based molecular generative models.
  arXiv  Detail & Related papers  (2025-08-18T05:13:27Z)
• Conditional Synthesis of 3D Molecules with Time Correction Sampler [58.0834973489875]
  Time-Aware Conditional Synthesis (TACS) is a novel approach to conditional generation on diffusion models. It integrates adaptively controlled plug-and-play "online" guidance into a diffusion model, driving samples toward the desired properties.
  arXiv  Detail & Related papers  (2024-11-01T12:59:25Z)
• Structure Language Models for Protein Conformation Generation [66.42864253026053]
  Traditional physics-based simulation methods often struggle with sampling equilibrium conformations.<n>Deep generative models have shown promise in generating protein conformations as a more efficient alternative.<n>We introduce Structure Language Modeling as a novel framework for efficient protein conformation generation.
  arXiv  Detail & Related papers  (2024-10-24T03:38:51Z)
• Diffusion Models in $\textit{De Novo}$ Drug Design [0.0]
  Diffusion models have emerged as powerful tools for molecular generation, particularly in the context of 3D molecular structures. This review focuses on the technical implementation of diffusion models tailored for 3D molecular generation.
  arXiv  Detail & Related papers  (2024-06-07T06:34:13Z)
• DecompDiff: Diffusion Models with Decomposed Priors for Structure-Based Drug Design [62.68420322996345]
  Existing structured-based drug design methods treat all ligand atoms equally. We propose a new diffusion model, DecompDiff, with decomposed priors over arms and scaffold. Our approach achieves state-of-the-art performance in generating high-affinity molecules.
  arXiv  Detail & Related papers  (2024-02-26T05:21:21Z)
• Molecule Design by Latent Space Energy-Based Modeling and Gradual Distribution Shifting [53.44684898432997]
  Generation of molecules with desired chemical and biological properties is critical for drug discovery. We propose a probabilistic generative model to capture the joint distribution of molecules and their properties. Our method achieves very strong performances on various molecule design tasks.
  arXiv  Detail & Related papers  (2023-06-09T03:04:21Z)
• Towards Predicting Equilibrium Distributions for Molecular Systems with Deep Learning [60.02391969049972]
  We introduce a novel deep learning framework, called Distributional Graphormer (DiG), in an attempt to predict the equilibrium distribution of molecular systems. DiG employs deep neural networks to transform a simple distribution towards the equilibrium distribution, conditioned on a descriptor of a molecular system.
  arXiv  Detail & Related papers  (2023-06-08T17:12:08Z)
• Modeling Molecular Structures with Intrinsic Diffusion Models [2.487445341407889]
  This thesis proposes Intrinsic Diffusion Modeling. It combines diffusion generative models with scientific knowledge about the flexibility of biological complexes. We demonstrate the effectiveness of this approach on two fundamental tasks at the basis of computational chemistry and biology.
  arXiv  Detail & Related papers  (2023-02-23T03:26:48Z)
• Learning Neural Generative Dynamics for Molecular Conformation Generation [89.03173504444415]
  We study how to generate molecule conformations (textiti.e., 3D structures) from a molecular graph. We propose a novel probabilistic framework to generate valid and diverse conformations given a molecular graph.
  arXiv  Detail & Related papers  (2021-02-20T03:17:58Z)
• Generating Tertiary Protein Structures via an Interpretative Variational Autoencoder [16.554053012204182]
  This paper proposes and evaluates an alternative approach to generating functionally-relevant three-dimensional structures of a protein. A comprehensive evaluation of several deep architectures shows the promise of generative models in directly revealing the latent space for sampling novel tertiary structures.
  arXiv  Detail & Related papers  (2020-04-08T17:40:21Z)

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.