Molecular Conformation Generation via Shifting Scores
Zihan Zhou, Ruiying Liu, Chaolong Ying, Ruimao Zhang, Tianshu Yu

TL;DR
This paper introduces a novel diffusion-based method for molecular conformation generation that models inter-atomic distances with shifting distributions, improving the realism and feasibility of generated molecular geometries.
Contribution
The paper proposes a new generative approach using shifting distributions for inter-atomic distances, ensuring feasible and reversible molecular conformations, outperforming existing methods.
Findings
Outperforms state-of-the-art methods in molecular conformation generation
Ensures feasible inter-atomic distance geometries
Demonstrates advantages of shifting distribution modeling
Abstract
Molecular conformation generation, a critical aspect of computational chemistry, involves producing the three-dimensional conformer geometry for a given molecule. Generating molecular conformation via diffusion requires learning to reverse a noising process. Diffusion on inter-atomic distances instead of conformation preserves SE(3)-equivalence and shows superior performance compared to alternative techniques, whereas related generative modelings are predominantly based upon heuristical assumptions. In response to this, we propose a novel molecular conformation generation approach driven by the observation that the disintegration of a molecule can be viewed as casting increasing force fields to its composing atoms, such that the distribution of the change of inter-atomic distance shifts from Gaussian to Maxwell-Boltzmann distribution. The corresponding generative modeling ensures a…
Peer Reviews
Decision·Submitted to ICLR 2024
The connection between gaussian perturbation and inter-atomic distance shifts are quite interesting, and the authors are able to leverage such observation to learn a diffusion model to learn such shifting scores. It gives an interesting likelihood model on top of many diffusion-based conformer generation models.
While the observation is interesting and it's great that the authors are able to demonstrate its superior performance, the GEOM benchmark has been used for quite some time now and probably over-optimized, so it's difficult to argue true superiority marginal gain on one benchmark alone. In addition, the majority of the mathematical framework for score matching involving langevin dynamics are not new to this problem either.
The empirical performance is compelling
- Probably my biggest problem with the paper is the motivation for using the Maxwell- Boltzmann distribution (MBD). The MBD describes the distribution of the length of velocity vectors in an ideal gas, and to some good approximation even in a real molecule. The authors generate molecular conformations based on distances, velocities are never generated nor used (e.g., the math on page 5 after eq 7 only includes distances). It is also unclear how interatomic distances (which are by definition pure
Originality is good but not surprising. The model follows the existing geometric diffusion models but with novel transition kernels, and the paper well explains the mathematical foundation of the diffusion process. Quality and clarity are good. The paper is well-presented and easy to follow. The technical details are clearly explained.
The main weakness from my perspective is the significance of empirical comparison. The improvement over GeoDiff is not significant to me. Could the author provide more ablation study about the $f_\sigma$ function in Eq7, which can help to verify the importance of the proposed MB diffusion distribution.
Videos
No videos yet. Explain this paper in a talk, walkthrough, or lecture? Add one.
Taxonomy
TopicsMachine Learning in Materials Science · Computational Drug Discovery Methods · Protein Structure and Dynamics
MethodsDiffusion
