dxtb-An efficient and fully differentiable framework for extended tight-binding

Marvin Friede

doi:10.1063/5.0216715

2024-08-09

PRE

DOI：10.1063/5.0216715

原文链接

原文求助

摘要

En 中文

Automatic differentiation (AD) emerged as an integral part of machine learning, accelerating model development by enabling gradient-based optimization without explicit analytical derivatives. Recently, the benefits of AD and computing arbitrary-order derivatives with respect to any variable were also recognized in the field of quantum chemistry. In this work, we present dxtb-an open-source, fully differentiable framework for semiempirical extended tight-binding (xTB) methods. Developed entirely in Python and leveraging PyTorch for array operations, dxtb facilitates extensibility and rapid prototyping while maintaining computational efficiency. Through comprehensive code vectorization and optimization, we essentially reach the speed of compiled xTB programs for high-throughput calculations of small molecules. The excellent performance also scales to large systems, and batch operability yields additional benefits for execution on parallel hardware. In particular, energy evaluations are on par with existing programs, whereas the speed of automatically differentiated nuclear derivatives is only 2 to 5 times slower compared to their analytical counterparts. We showcase the utility of AD in dxtb by calculating various molecular and spectroscopic properties, highlighting its capacity to enhance and simplify such evaluations. Furthermore, the framework streamlines optimization tasks and offers seamless integration of semiempirical quantum chemistry in machine learning, paving the way for physics-inspired end-to-end differentiable models. Ultimately, dxtb aims to further advance the capabilities of semiempirical methods, providing an extensible foundation for future developments and hybrid machine learning applications. The framework is accessible at https://github.com/grimme-lab/dxtb.

Keyword:

DENSITY-FUNCTIONAL THEORY

QUANTUM-CHEMICAL METHOD

SPD-BLOCK ELEMENTS

AUTOMATIC DIFFERENTIATION

DISPERSION INTERACTION

MOLECULAR INTEGRALS

RAMAN INTENSITIES

BASIS-SET

SCC-DFTB

DERIVATIVES