Kai Zhu 「祝凯」

I am a Ph.D. student in Pharmaceutical Sciences at Zhejiang University, advised by Prof. Tingjun Hou. I am also a visiting student at the Italian Institute of Technology, where I work with Luigi Bonati and Prof. Michele Parrinello.

My research focuses on machine-learning methods for molecular dynamics and enhanced sampling, particularly dynamical representation learning for rare-event sampling and mechanistic analysis.

I also contribute to open-source scientific software, including mlcolvar, to make these methods easier to use and reproduce.

Research interests
Machine-learning collective variables Enhanced sampling Rare-event dynamics Dynamical representation learning

Research

Selected publications

† Equal contribution. * Corresponding author.

arXiv SelfTICA framework

SelfTICA: contrastive learning of dynamical representations for rare-event sampling and characterization

Kai Zhu†, Jintu Zhang†, Pietro Novelli, Tingjun Hou*, Luigi Bonati*
arXiv arXiv:2606.15495 (2026)

arXiv Code Dataset
scholar 0
Rare events govern many important molecular processes but remain difficult to characterize within accessible simulation timescales. This challenge has motivated machine-learning methods to construct low-dimensional collective variables for specific objectives, including state discrimination, slow-mode identification, and committor-function approximation. Here we introduce SelfTICA, a self-supervised framework that uses contrastive learning on time-lagged configurations to learn a latent representation of the slow modes governing relevant transitions. Once learned, this representation is frozen and reused across downstream tasks, including collective variables construction, enhanced sampling, free-energy estimation, and committor learning. Compared with direct slow-mode optimization, SelfTICA improves training stability and provides collective variables even from limited and exploratory trajectories, accelerating rare-event sampling and enabling accurate free-energy estimation. The dynamical information encoded during pretraining also accelerates committor learning and enables characterization of transition-state regions. These results show that contrastive learning of slow dynamical representations provides a common foundation for rare-event sampling and mechanistic analysis.
Nat. Commun. AR-NTD enhanced sampling workflow

Targeting the intrinsically disordered AR-NTD through a machine learning-based enhanced sampling workflow

Kai Zhu†, Huating Wang†, Jintu Zhang†, Renling Hu, Linlong Jiang, Hui Zhang, Yu Kang, Tingjun Hou*, Dan Li*
Nat. Commun. 17, 7206 (2026)

Paper Code Dataset
scholar 1
Intrinsically disordered proteins are challenging drug targets because they lack stable three-dimensional structures and instead exist as dynamic conformational ensembles. In this work, we developed a machine learning-based enhanced sampling workflow to target the intrinsically disordered N-terminal domain of the androgen receptor (AR-NTD), a therapeutic target for drug-resistant prostate cancer. By combining enhanced sampling simulations with machine-learning collective variables, we characterized the conformational landscape of the Tau-5 region, identified druggable conformations, and revealed key molecular features involved in ligand recognition. These insights enabled structure-based virtual screening and led to the identification of K53, an AR-NTD antagonist with anti-proliferative activity in enzalutamide-resistant prostate cancer cells.
Chem. Rev. Enhanced sampling in the age of machine learning

Enhanced Sampling in the Age of Machine Learning: Algorithms and Applications

Kai Zhu†, Enrico Trizio†, Jintu Zhang, Renling Hu, Linlong Jiang, Tingjun Hou*, Luigi Bonati*
Chem. Rev. 2026, 126, 1, 671-713

Paper arXiv
scholar 85
Molecular dynamics simulations hold great promise for providing insight into the microscopic behavior of complex molecular systems. However, their effectiveness is often constrained by long timescales associated with rare events. Enhanced sampling methods have been developed to address these challenges, and recent years have seen a growing integration with machine learning techniques. This Review provides a comprehensive overview of how they are reshaping the field, with a particular focus on the data-driven construction of collective variables. Furthermore, these techniques have also improved biasing schemes and unlocked novel strategies via reinforcement learning and generative approaches. In addition to methodological advances, we highlight applications spanning different areas, such as biomolecular processes, ligand binding, catalytic reactions, and phase transitions. We conclude by outlining future directions aimed at enabling more automated strategies for rare-event sampling.
Nat. Commun. LiTEN foundation model

A Scalable and Quantum-Accurate Foundation Model for Biomolecular Force Field via Linearly Tensorized Quadrangle Attention

Qun Su†, Kai Zhu†, Qiaolin Gou†, Jintu Zhang, Renling Hu, Yurong Li, Yongze Wang, Hui Zhang, Ziyi You, Linlong Jiang, Yu Kang, Jike Wang, Chang-Yu Hsieh, Tingjun Hou
Nat. Commun. 17, 3639 (2026)

Paper arXiv Code Dataset
scholar 3
Accurate atomistic biomolecular simulations are vital for disease mechanism understanding, drug discovery, and biomaterial design, but existing simulation methods exhibit significant limitations. Classical force fields are efficient but lack accuracy for transition states and fine conformational details critical in many chemical and biological processes. Quantum Mechanics methods are highly accurate but computationally infeasible for large-scale or long-time simulations. AI-based force fields aim to achieve quantum-level accuracy with efficiency but struggle to balance many-body modeling complexity, accuracy, and speed. To overcome these challenges, we introduce LiTEN, an equivariant neural network with Tensorized Quadrangle Attention. TQA efficiently models three- and four-body interactions with linear complexity by reparameterizing high-order tensor features via vector operations. Building on LiTEN, LiTEN-FF is a robust AI force-field foundation model pre-trained on the nablaDFT dataset and fine-tuned on SPICE. LiTEN achieves strong performance across rMD17, MD22, and Chignolin while enabling downstream biomolecular modeling tasks including conformer searches, geometry optimization, and free-energy-surface construction.