Hugging Face's logo

fudan-generative-ai
/
dynamicPDB

Model card Files
xet
 Community
5
dynamicPDB
File size: 2,865 Bytes
0f70d74
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
---
pipeline_tag: other
---

# DyneTrion: Spatiotemporally Coherent Generative Emulation of Protein Dynamics Across Timescales

This repository contains the pretrained weights for **DyneTrion**, a generative protein dynamics emulator that jointly enforces geometric symmetry, structural consistency, and temporal coherence within a single framework.

- **Paper:** [DyneTrion: A Spatio-temporally Coherent Generative Emulator for Protein Dynamics Across Timescales](https://huggingface.co/papers/2607.15309)
- **GitHub Repository:** [fudan-generative-vision/DyneTrion](https://github.com/fudan-generative-vision/DyneTrion)

<table class="center">
  <tr>
    <td style="text-align: center"><b>1ail_A</b></td>
    <td style="text-align: center"><b>1ifg_A</b></td>
    <td style="text-align: center"><b>2kxl_A</b></td>
    <td style="text-align: center"><b>2rcs_H</b></td>
  </tr>
  <tr>
    <td style="text-align: center">
      <img src="https://raw.githubusercontent.com/fudan-generative-vision/DyneTrion/main/assets/1ail_A.gif" width="100%">
    </td>
    <td style="text-align: center">
      <img src="https://raw.githubusercontent.com/fudan-generative-vision/DyneTrion/main/assets/1ifg_A.gif" width="100%">
    </td>
    <td style="text-align: center">
      <img src="https://raw.githubusercontent.com/fudan-generative-vision/DyneTrion/main/assets/2kxl_A.gif" width="100%">
    </td>
    <td style="text-align: center">
      <img src="https://raw.githubusercontent.com/fudan-generative-vision/DyneTrion/main/assets/2rcs_H.gif" width="100%">
    </td>
  </tr>
</table>

## Installation

```bash
# Create virtual environment (Python 3.10.12 is recommended)
python -m venv .venv
source .venv/bin/activate

# Install PyTorch (CUDA 12.4)
pip install torch==2.4.0 --index-url https://download.pytorch.org/whl/cu124

# Install other dependencies
pip install -r requirements.txt
```

## Inference

Run inference using:

```bash
bash inference.sh
```

- Model checkpoint: `step_400000.pth`
- Input CSV: `datasets/inference/inference_data.csv`
- Frame number: `n_frame = 16`
- Motion number: `n_motion = 2`
- Frame sampling step: `sample_step = 40`
- Extrapolation time: `extrapolation_time = 16`
- Noise scale: `noise_scale = 1.0`

Inference results will be saved to `save_root` (default: `./test/inference/`).

For more details, please check the [GitHub Repository](https://github.com/fudan-generative-vision/DyneTrion).

## Citation

If you find DyneTrion useful for your research, please cite the paper:

```bibtex
@article{cheng2025dynetrion,
  title={DyneTrion: A Spatio-temporally Coherent Generative Emulator for Protein Dynamics Across Timescales},
  author={Cheng, Kaihui and Cai, Zhiqiang and Tu, Peng and Yao, Yisong and Han, Limei and Wu, Libo and Zhu, Siyu and Yang, Tzuhsiung and Qi, Yuan},
  journal={arXiv preprint arXiv:2607.15309},
  year={2026}
}
```