Rigid Body Flows for Sampling Molecular Crystal Structures

• URL: http://arxiv.org/abs/2301.11355v4
• Date: Wed, 7 Jun 2023 09:01:03 GMT
• Title: Rigid Body Flows for Sampling Molecular Crystal Structures
• Authors: Jonas K\"ohler, Michele Invernizzi, Pim de Haan, Frank No\'e
• Abstract summary: We introduce a new type of normalizing flow that is tailored for modeling positions and orientations of multiple objects in three-dimensional space. Our approach is based on two key ideas: first, we define smooth and expressive flows on the group of unit quaternions, which allows us to capture the continuous rotational motion of rigid bodies. We evaluate the method by training Boltzmann generators for two molecular examples, namely the multi-modal density of a tetrahedral system in an external field and the ice XI phase in the TIP4P water model.
• Score: 4.368185344922342
• License: http://creativecommons.org/licenses/by/4.0/
• Abstract: Normalizing flows (NF) are a class of powerful generative models that have gained popularity in recent years due to their ability to model complex distributions with high flexibility and expressiveness. In this work, we introduce a new type of normalizing flow that is tailored for modeling positions and orientations of multiple objects in three-dimensional space, such as molecules in a crystal. Our approach is based on two key ideas: first, we define smooth and expressive flows on the group of unit quaternions, which allows us to capture the continuous rotational motion of rigid bodies; second, we use the double cover property of unit quaternions to define a proper density on the rotation group. This ensures that our model can be trained using standard likelihood-based methods or variational inference with respect to a thermodynamic target density. We evaluate the method by training Boltzmann generators for two molecular examples, namely the multi-modal density of a tetrahedral system in an external field and the ice XI phase in the TIP4P water model. Our flows can be combined with flows operating on the internal degrees of freedom of molecules and constitute an important step towards the modeling of distributions of many interacting molecules.

Related papers

• Geometric Trajectory Diffusion Models [58.853975433383326]
  Generative models have shown great promise in generating 3D geometric systems. Existing approaches only operate on static structures, neglecting the fact that physical systems are always dynamic in nature. We propose geometric trajectory diffusion models (GeoTDM), the first diffusion model for modeling the temporal distribution of 3D geometric trajectories.
  arXiv  Detail & Related papers  (2024-10-16T20:36:41Z)
• Iterated Energy-based Flow Matching for Sampling from Boltzmann Densities [11.850515912491657]
  We propose iterated energy-based flow matching (iEFM) to train continuous normalizing flow (CNF) models from unnormalized densities. Our results demonstrate that iEFM outperforms existing methods, showcasing its potential for efficient and scalable probabilistic modeling.
  arXiv  Detail & Related papers  (2024-08-29T04:06:34Z)
• Conditional Normalizing Flows for Active Learning of Coarse-Grained Molecular Representations [0.6906005491572401]
  generative machine learning methods such as normalizing flows have been used to learn the Boltzmann distribution directly, without samples. In this work, we address this challenge by separating the problem into two levels, the fine-grained and coarse-grained degrees of freedom. To explore the configurational space, we employ coarse-grained simulations with active learning which allows us to update the flow and make all-atom potential energy evaluations only when necessary.
  arXiv  Detail & Related papers  (2024-02-02T07:44:26Z)
• Equivariant Flow Matching with Hybrid Probability Transport [69.11915545210393]
  Diffusion Models (DMs) have demonstrated effectiveness in generating feature-rich geometries. DMs typically suffer from unstable probability dynamics with inefficient sampling speed. We introduce geometric flow matching, which enjoys the advantages of both equivariant modeling and stabilized probability dynamics.
  arXiv  Detail & Related papers  (2023-12-12T11:13:13Z)
• 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)
• Manipulating growth and propagation of correlations in dipolar multilayers: From pair production to bosonic Kitaev models [0.0]
  We map the many-body spin dynamics to bosonic models. In a bilayer configuration we show how to engineer the paradigmatic two-mode squeezing Hamiltonian known from quantum optics. In multi-layer configurations we engineer a bosonic variant of the Kitaev model displaying chiral propagation along the layer direction.
  arXiv  Detail & Related papers  (2022-11-22T19:00:01Z)
• GeoDiff: a Geometric Diffusion Model for Molecular Conformation Generation [102.85440102147267]
  We propose a novel generative model named GeoDiff for molecular conformation prediction. We show that GeoDiff is superior or comparable to existing state-of-the-art approaches.
  arXiv  Detail & Related papers  (2022-03-06T09:47:01Z)
• Normalizing flows for atomic solids [67.70049117614325]
  We present a machine-learning approach, based on normalizing flows, for modelling atomic solids. We report Helmholtz free energy estimates for cubic and hexagonal ice modelled as monatomic water as well as for a truncated and shifted Lennard-Jones system. Our results thus demonstrate that normalizing flows can provide high-quality samples and free energy estimates of solids, without the need for multi-staging or for imposing restrictions on the crystal geometry.
  arXiv  Detail & Related papers  (2021-11-16T18:54:49Z)
• Smooth Normalizing Flows [0.0]
  We introduce a class of smooth mixture transformations working on both compact intervals and hypertori. We show that such inverses can be computed from forward evaluations via the inverse function theorem. We demonstrate two advantages of such smooth flows: they allow training by force matching to simulation data and can be used as potentials in molecular dynamics simulations.
  arXiv  Detail & Related papers  (2021-10-01T12:27:14Z)
• Floquet Engineering Ultracold Polar Molecules to Simulate Topological Insulators [5.6888061544906545]
  We present a near-term experimental blueprint for the quantum simulation of topological insulators using lattice-trapped ultracold polar molecules. We focus on the so-called Hopf insulator, which represents a three-dimensional topological state of matter existing outside the conventional way.
  arXiv  Detail & Related papers  (2021-05-21T18:00:00Z)
• 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)

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.