Abstract:The accurate prediction of geometric state evolution in complex systems is critical for advancing scientific domains such as quantum chemistry and material modeling. Traditional experimental and computational methods face challenges in terms of environmental constraints and computational demands, while current deep learning approaches still fall short in terms of precision and generality. In this work, we introduce the Geometric Diffusion Bridge (GDB), a novel generative modeling framework that accurately bridges initial and target geometric states. GDB leverages a probabilistic approach to evolve geometric state distributions, employing an equivariant diffusion bridge derived by a modified version of Doob's $h$-transform for connecting geometric states. This tailored diffusion process is anchored by initial and target geometric states as fixed endpoints and governed by equivariant transition kernels. Moreover, trajectory data can be seamlessly leveraged in our GDB framework by using a chain of equivariant diffusion bridges, providing a more detailed and accurate characterization of evolution dynamics. Theoretically, we conduct a thorough examination to confirm our framework's ability to preserve joint distributions of geometric states and capability to completely model the underlying dynamics inducing trajectory distributions with negligible error. Experimental evaluations across various real-world scenarios show that GDB surpasses existing state-of-the-art approaches, opening up a new pathway for accurately bridging geometric states and tackling crucial scientific challenges with improved accuracy and applicability.

Diffusion bridges for stochastic Hamiltonian systems and shape evolutions

Diffusion Bridge Mixture Transports, Schrödinger Bridge Problems and Generative Modeling

Transport with Support: Data-Conditional Diffusion Bridges

Shape Splines and Stochastic Shape Evolutions: A Second Order Point of View

Unbalanced Diffusion Schrödinger Bridge

Bridging Geometric States via Geometric Diffusion Bridge

Simulating diffusion bridges using the Wiener chaos expansion

Exploring the Design Space of Diffusion Bridge Models via Stochasticity Control

Simulating infinite-dimensional nonlinear diffusion bridges

Conditional Simulation Using Diffusion Schrödinger Bridges

Aligned Diffusion Schrödinger Bridges

Discrete Diffusion Schrödinger Bridge Matching for Graph Transformation

Reflected Schrödinger Bridge for Constrained Generative Modeling

Diffusion Schrödinger Bridge Matching

Generative Modeling with Phase Stochastic Bridges

Stochastic Optimal Control for Diffusion Bridges in Function Spaces

Deformable 3D Shape Diffusion Model

On the Exact Simulation of (Jump) Diffusion Bridges

Geometric Constraints in Probabilistic Manifolds: A Bridge from Molecular Dynamics to Structured Diffusion Processes

Stochastic metamorphosis with template uncertainties

Diffusion Geometry