ISOKANN

The ISOKANN.jl package implements the ISOKANN algorithm for the identification of macro-states of molecular systems. Its main features comprise of:

• A flexible implementation of the ISOKANN core (Iso2) (supporting 1D and N-D ISOKANN, customizable neural networks on a broad set of SimulationData)
• A battery-included interfaces to OpenMM for automated adaptive sampling of molecular dynamics
• Different adaptive sampling strategies (extapolation, kde and stratified sampling)
• A posteriori analysis tools (plots, reaction path extraction and reaction rate estimation)

Quick start

Install the package via using Pkg; Pkg.add("https://github.com/axsk/ISOKANN.jl").

The usual pipeline consists of the creation of system simulation, generation of training data, training ISOKANN and a posteriori analysis of the results.

using ISOKANN

# Define an OpenMMSimulation. The default molecule is the Alanine-Dipeptide.
sim = OpenMMSimulation()

# Sample the initial data for training of ISOKANN with 100 initial points and 5 koopman samples per point.
data = isodata(sim, 100, 5)

# create the ISOKANN training object
iso = Iso2(data)

# train for 100 episodes
run!(iso, 100)

# plot the training losses and chi values
plot_training(iso) 

# scatter plot of all initial points colored in corresponding chi value
scatter_ramachandran(iso)

# estimate the exit rates, i.e. the metastability
exit_rates(iso)

# extract the reactive path
save_reactive_path(iso, out="path.pdb")

A more comprehensive example simulating the folding of the chicken villin can be found in scripts/villin.jl. For further information consult the docstrings (e.g. ?Iso2).

References

• Rabben, Ray, Weber (2018) - ISOKANN: Invariant subspaces of Koopman operators learned by a neural network.
• Sikorski, Ribera Borrell, Weber (2024) - Learning Koopman eigenfunctions of stochastic diffusions with optimal importance sampling and ISOKANN
• Sikorski, Rabben, Chewle, Weber (2024) - Capturing the Macroscopic Behaviour of Molecular Dynamics with Membership Functions