Large Language-Geometry Model: When LLM meets Equivariance

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Li, Zongzhao, Cen, Jiacheng, Su, Bing, Huang, Wenbing, Xu, Tingyang, Rong, Yu, Zhao, Deli
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912237490274304
author Li, Zongzhao
Cen, Jiacheng
Su, Bing
Huang, Wenbing
Xu, Tingyang
Rong, Yu
Zhao, Deli
author_facet Li, Zongzhao
Cen, Jiacheng
Su, Bing
Huang, Wenbing
Xu, Tingyang
Rong, Yu
Zhao, Deli
contents Accurately predicting 3D structures and dynamics of physical systems is crucial in scientific applications. Existing approaches that rely on geometric Graph Neural Networks (GNNs) effectively enforce $\mathrm{E}(3)$-equivariance, but they often fall in leveraging extensive broader information. While direct application of Large Language Models (LLMs) can incorporate external knowledge, they lack the capability for spatial reasoning with guaranteed equivariance. In this paper, we propose EquiLLM, a novel framework for representing 3D physical systems that seamlessly integrates E(3)-equivariance with LLM capabilities. Specifically, EquiLLM comprises four key components: geometry-aware prompting, an equivariant encoder, an LLM, and an equivariant adaptor. Essentially, the LLM guided by the instructive prompt serves as a sophisticated invariant feature processor, while 3D directional information is exclusively handled by the equivariant encoder and adaptor modules. Experimental results demonstrate that EquiLLM delivers significant improvements over previous methods across molecular dynamics simulation, human motion simulation, and antibody design, highlighting its promising generalizability.
format Preprint
id arxiv_https___arxiv_org_abs_2502_11149
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Large Language-Geometry Model: When LLM meets Equivariance
Li, Zongzhao
Cen, Jiacheng
Su, Bing
Huang, Wenbing
Xu, Tingyang
Rong, Yu
Zhao, Deli
Machine Learning
Artificial Intelligence
Accurately predicting 3D structures and dynamics of physical systems is crucial in scientific applications. Existing approaches that rely on geometric Graph Neural Networks (GNNs) effectively enforce $\mathrm{E}(3)$-equivariance, but they often fall in leveraging extensive broader information. While direct application of Large Language Models (LLMs) can incorporate external knowledge, they lack the capability for spatial reasoning with guaranteed equivariance. In this paper, we propose EquiLLM, a novel framework for representing 3D physical systems that seamlessly integrates E(3)-equivariance with LLM capabilities. Specifically, EquiLLM comprises four key components: geometry-aware prompting, an equivariant encoder, an LLM, and an equivariant adaptor. Essentially, the LLM guided by the instructive prompt serves as a sophisticated invariant feature processor, while 3D directional information is exclusively handled by the equivariant encoder and adaptor modules. Experimental results demonstrate that EquiLLM delivers significant improvements over previous methods across molecular dynamics simulation, human motion simulation, and antibody design, highlighting its promising generalizability.
title Large Language-Geometry Model: When LLM meets Equivariance
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2502.11149