A scalable and quantum-accurate foundation model for biomolecular force fields via linearly tensorized quadrangle attention.

Abstract

Accurate atomistic biomolecular simulations are vital for understanding disease mechanisms and drug discovery, yet existing methods struggle to balance quantum-mechanical accuracy with computational scalability. Classical force fields often lack precision, while quantum methods are computationally prohibitive for complex biological systems. Here we show that LiTEN, a scalable equivariant neural network, resolves this dilemma by efficiently modeling complex three- and four-body interactions with linear complexity via Linearly Tensorized Quadrangle Attention. We introduce LiTEN-FF, a foundation model pre-trained on extensive datasets to ensure broad chemical generalization across diverse molecular spaces. We demonstrate that LiTEN achieves state-of-the-art accuracy on standard benchmarks, consistently outperforming leading approaches in both precision and speed. Furthermore, LiTEN-FF enables comprehensive modeling tasks, ranging from geometry optimization to free energy surface construction, with high computational efficiency for large biomolecules. This framework provides a physically grounded, versatile foundation for advanced biomolecular modeling and drug design applications.