Scaling Transferable Implicit Transfer Operators
Title: Scaling Transferable Implicit Transfer Operators
DNr: Berzelius-2026-213
Project Type: LiU Berzelius
Principal Investigator: Simon Olsson <simonols@chalmers.se>
Affiliation: Chalmers tekniska högskola
Duration: 2026-07-09 – 2027-02-01
Classification: 10210
Homepage: https://psolsson.github.io/
Keywords:

Abstract

Computing properties of molecular systems rely on estimating expectations of the (unnormalized) Boltzmann distribution. Molecular dynamics (MD) is a broadly adopted technique to approximate such quantities. However, stable simulations rely on tiny integration time steps (10^(−15) s), whereas convergence of some moments, e.g., binding free energy or rates, might rely on sampling processes on time scales as long as 10^(−1) s, and these simulations must be repeated for every molecular system independently. We recently proposed Implicit Transfer Operator (ITO) Learning [1], a framework to learn surrogates of the simulation process with multiple time resolutions. Our initial work shows that ITO models can simulate challenging molecular systems, such as fast-folding proteins, with time steps at least six orders of magnitude larger than traditional molecular dynamics [1]. We have show that these models can be trained to generalize to small molecules and peptides [2] as well as small coarse-grained single domain proteins [3] in previous allocations, and we have built the first steps towards an all-atom single-domain protein model (unpublished). In this allocation, we plan to scale this to arbitrarily sized single-chain proteins with all-atom resolution using latent generative models coupled with a low-rank attention mechanism. Since we are now reaching a stage where comparison to the ground truth simulation model is intractable, since numerical simulations are costly, we will do a `sim2real' transfer and use our approach [4] to calibrate this new model to all available biophysical data. [1] Mathias Schreiner, Ole Winther and Simon Olsson. Implicit Transfer Operator Learning: Multiple Time-Resolution Surrogates for Molecular Dynamics. 37th Conference on Neural Information Processing Systems (NeurIPS 2023). https://arxiv.org/abs/2305.18046 [2] Viguera Diez J, Schreiner M, Olsson S . "Transferable Generative Models Bridge Femtosecond to Nanosecond Time-Step Molecular Dynamics" Science Advances. [3] Antoniadis P, Pavesi B, Olsson S*, Winther O* "Protein Language Model Embeddings Improve Generalization of Implicit Transfer Operators" ICML 2026 [4] Kolloff C, H\"oppe T, Angelis E, Schreiner M, Bauer S, Dittadi A, Olsson S* "Minimum Excess Work Guidance: Score-Based Sampling with Experimental Data or Sparse Restraints" J. Chem. Theory. Comput. (2026)