Chem4DLLM: 4D Multimodal LLMs for Chemical Dynamics Understanding
Organizations: Australian Institute for Machine Learning, Adelaide University, Adelaide, Australia · CSIRO’s Data61, Sydney, Australia
Abstract
Existing chemical understanding tasks primarily rely on static molecular representations, limiting their ability to model inherently dynamic phenomena such as bond breaking or conformational changes, which are essential for a chemist to understand chemical reactions. To address this gap, we introduce Chemical Dynamics Understanding (ChemDU), a new task that translates 4D molecular trajectories into interpretable natural-language explanations. ChemDU focuses on fundamental dynamic scenarios, including gas-phase and catalytic reactions, and requires models to reason about key events along molecular trajectories, such as bond formation and dissociation, and to generate coherent, mechanistically grounded narratives. To benchmark this capability, we construct Chem4DBench, the first dataset pairing 4D molecular trajectories with expert-authored explanations across these settings. We further propose Chem4DLLM, a unified model that integrates an equivariant graph encoder with a pretrained large language model to explicitly capture molecular geometry and rotational dynamics. We hope that ChemDU, together with Chem4DBench and Chem4DLLM, will stimulate further research in dynamic chemical understanding and multimodal scientific reasoning.
Figures & tables
| Model | BLEU | EXACT | LEVENSHTEIN | MACCS | RDK | MORGAN | VALIDITY | MAE Barrier | MAE Enthalpy |
|---|---|---|---|---|---|---|---|---|---|
| Transition1x In Distribution | |||||||||
| 3D-MoLM | 0.216 | 0.075 | 10.385 | 0.296 | 0.258 | 0.168 | 0.514 | 3.157 | 3.618 |
| 3D-MolT5 | 0.431 | 0.070 | 4.566 | 0.514 | 0.400 | 0.240 | 0.992 | 0.902 | 1.392 |
| Chem3D LLM | 0.059 | 0.000 | 8.347 | 0.206 | 0.124 | 0.053 | 0.954 | 1.262 | 2.064 |
| 4D-MolT5 | 0.480 | 0.131 | 4.381 | 0.568 | 0.477 | 0.308 | 0.990 | 0.900 | 1.265 |
| 4D Text-based | 0.051 | 0.000 | 8.318 | 0.200 | 0.119 | 0.049 | 0.957 | 1.552 | 2.204 |
| Model | Type Acc | Ads Exact | Ads Morgan | Prod Exact | Prod Morgan | TS Step MAE | Barrier MAE | Enthalpy MAE |
|---|---|---|---|---|---|---|---|---|
| Chem3D | 0.517 | 0.355 | 0.392 | 0.054 | 0.064 | 2.437 | 0.985 | 1.069 |
| 4D text-based | 0.535 | 0.373 | 0.410 | 0.068 | 0.084 | 1.947 | 1.019 | 1.102 |
| Chem4DLLM | 0.774 | 0.762 | 0.776 | 0.432 | 0.454 | 1.348 | 0.848 | 1.019 |
Appendix figures & tables5 assets
Supplementary material from the paper’s appendix.
Appendix
| Chem3DLLLM-style (3D Input) | 4D-Text-based (Trajectory Input) |
| Given the reactant and transition state geometries, predict the product SMILES, reaction barrier and reaction enthalpy. The geometry is as follows: The system is non-periodic. The atom and its position are as follows: C1 -1.6169 0.5542 0.9918 C2 -2.2864 -0.7707 1.3265 C3 -1.4256 0.7574 -0.4801 ... H21 1.0702 -0.4440 -2.0669 The distance between atoms is as follows: atom1_index atom2_index distance(A) 1 7 1.0930 1 6 1.0964 ... 21 19 2.3363 | Given the reactant and transition state geometries, predict the product SMILES, reaction barrier and reaction enthalpy. The geometry is as follows: Frame 1: The system is non-periodic. The atom and its position are as follows: C1 -1.62 0.55 0.99 C2 -2.29 -0.77 1.33 ... The distance between atoms is as follows: atom1_index atom2_index distance(A) 1 7 1.09 1 6 1.10 ... 21 19 2.34 Frame 2: The atom with big movements are listed as follows: C1 0.77 1.22 -0.69 C2 2.26 1.20 -0.37 ... H21 -1.48 -1.01 1.37 New edge distances: atom1_index atom2_index distance(A) 1 7 1.09 1 6 1.09 ... 21 19 2.25 |
Multimodal Large Language Models
Multimodal large language models combine language models with image, video, audio, and other inputs for grounded understanding and generation.
See all 1,091 papers