Machine Learning-driven Multiscale MD Workflows: The Mini-MuMMI Experience

TL;DR

This paper introduces mini-MuMMI, a lightweight multiscale workflow management system for machine learning-driven molecular dynamics simulations, enabling complex biological modeling on modest computational resources.

Contribution

The paper presents mini-MuMMI, a simplified version of MuMMI, designed for smaller HPC systems, expanding accessibility for multiscale MD simulations.

Findings

Mini-MuMMI successfully models RAS-RAF membrane interactions.

It demonstrates effective orchestration of multiscale simulations on modest hardware.

The approach broadens the application scope of multiscale workflows.

Abstract

Computational models have become one of the prevalent methods to model complex phenomena. To accurately model complex interactions, such as detailed biomolecular interactions, scientists often rely on multiscale models comprised of several internal models operating at difference scales, ranging from microscopic to macroscopic length and time scales. Bridging the gap between different time and length scales has historically been challenging but the advent of newer machine learning (ML) approaches has shown promise for tackling that task. Multiscale models require massive amounts of computational power and a powerful workflow management system. Orchestrating ML-driven multiscale studies on parallel systems with thousands of nodes is challenging, the workflow must schedule, allocate and control thousands of simulations operating at different scales. Here, we discuss the massively parallel Multiscale Machine-Learned Modeling Infrastructure (MuMMI), a multiscale workflow management infrastructure, that can orchestrate thousands of molecular dynamics (MD) simulations operating at different timescales, spanning from millisecond to nanosecond. More specifically, we introduce a novel version of MuMMI called "mini-MuMMI". Mini-MuMMI is a curated version of MuMMI designed to run on modest HPC systems or even laptops whereas MuMMI requires larger HPC systems. We demonstrate mini-MuMMI utility by exploring RAS-RAF membrane interactions and discuss the different challenges behind the generalization of multiscale workflows and how mini-MuMMI can be leveraged to target a broader range of applications outside of MD and RAS-RAF interactions.