license: cc-by-nc-4.0
Matcha: Multi-Stage Riemannian Flow Matching for Accurate and Physically Valid Molecular Docking
This repository hosts the Matcha pipeline checkpoints (three-stage flow matching models). The model is described in the paper Matcha: Multi-Stage Riemannian Flow Matching for Accurate and Physically Valid Molecular Docking.
Code and full documentation: GitHub – LigandPro/Matcha. The previous release is tagged v1 there.
Download the matcha_pipeline folder from the Files and versions tab and set checkpoints_folder in your paths config to the folder that contains it.
Abstract
Accurate prediction of protein-ligand binding poses is crucial for structure-based drug design, yet existing methods struggle to balance speed, accuracy, and physical plausibility. We introduce Matcha, a novel molecular docking pipeline that combines multi-stage flow matching with learned scoring and physical validity filtering. Our approach consists of three sequential stages applied consecutively to refine docking predictions, each implemented as a flow matching model operating on appropriate geometric spaces (R³, SO(3), and SO(2)). We enhance the prediction quality through a dedicated scoring model and apply unsupervised physical validity filters to eliminate unrealistic poses. Compared to various approaches, Matcha demonstrates superior performance on Astex and PDBbind test sets in terms of docking success rate and physical plausibility. Moreover, our method works approximately 31× faster than modern large-scale co-folding models.
Overview
Matcha is a molecular docking pipeline that combines multi-stage flow matching with learned scoring and physical validity filtering. Our approach consists of three sequential stages applied consecutively to progressively refine docking predictions, each implemented as a flow matching model operating on appropriate geometric spaces (R³, SO(3), and SO(2)). We enhance the prediction quality through a dedicated scoring model and apply unsupervised physical validity filters to eliminate unrealistic poses.
More details, CLI usage, and the full pipeline (including GNINA) are in the GitHub repository.
Compared to various approaches, Matcha demonstrates superior performance on Astex, PDBBind and PoseBusters test sets in terms of docking success rate and physical plausibility. Moreover, our method works approximately 31× faster than modern large-scale co-folding models.
Installation
Clone the Matcha repository and install:
cd Matcha
uv sync
# or: pip install -e .
Download the matcha_pipeline folder from this Hugging Face repo and set checkpoints_folder in configs/paths/paths.yaml to the parent folder of matcha_pipeline.
Sample usage
To run docking, save scored predictions and compute metrics, see the instructions provided in GitHub README.
Citation
If you use Matcha in your work, please cite:
@misc{frolova2025matchamultistageriemannianflow,
title={Matcha: Multi-Stage Riemannian Flow Matching for Accurate and Physically Valid Molecular Docking},
author={Daria Frolova and Talgat Daulbaev and Egor Sevriugov and Sergei A. Nikolenko and Dmitry N. Ivankov and Ivan Oseledets and Marina A. Pak},
year={2025},
eprint={2510.14586},
archivePrefix={arXiv},
primaryClass={cs.LG},
url={https://arxiv.org/abs/2510.14586},
}