Daily Arxiv

This is a page that curates AI-related papers published worldwide.
All content here is summarized using Google Gemini and operated on a non-profit basis.
Copyright for each paper belongs to the authors and their institutions; please make sure to credit the source when sharing.

DEQuify your force field: More efficient simulations using deep equilibrium models

Created by
  • Haebom

Author

Andreas Burger, Luca Thiede, Al an Aspuru-Guzik, Nandita Vijaykumar

Outline

Machine-learning force fields have shown great promise in enabling more accurate molecular dynamics simulations than traditional, manually generated force fields. Recent advances have been achieved by leveraging prior knowledge of the physical system, such as rotational, translational, and reflection symmetries. This paper proposes another crucial priori information, previously unexplored: that simulations of molecular systems are inherently continuous, and therefore, continuous states are highly similar. This study demonstrates that this information can be leveraged by restructuring the state-of-the-art equilibrium base model into a Deep Equilibrium Model (DEQ). This approach reuses intermediate neural network features from previous time steps, resulting in 10-20% accuracy and speedup compared to non-DEQ base models on the MD17, MD22, and OC20 200k datasets. Furthermore, training is significantly more memory-efficient, enabling training of more expressive models on larger systems.

Takeaways, Limitations

Takeaways:
A novel method is presented to improve the accuracy and speed of molecular dynamics simulations (10-20% improvement).
Memory-efficient training enables training more expressive models on larger systems.
A method for effectively utilizing prior information on temporal continuity is presented.
Limitations:
The performance improvement of the proposed method may be limited to specific datasets.
Generalization performance evaluation for various molecular systems is needed.
Potential increase in computational cost due to the complexity of the DEQ model.
👍