Hugging Face model: ByteDance-Seed/ConfRover-base-20M-v1.0
Author: ByteDance-Seed
Created: 2025-12-01 11:41:17
Last modified: 2025-12-01 12:19:13
Likes: 4 | Downloads: 0
Tags: arxiv:2505.17478, license:apache-2.0, region:us
Card metadata:
license: apache-2.0
variant: base
size: 20M
version: v1.0
model_summary: ConfRover base model trained with forward simulation and iid sampling
model_description: 
ConfRover is a deep generative model for protein 3D conformation and motion dynamics.
It leverages diffusion probability model to learn the distribution of protein 3D conformations and captures the their temporal dependencies between frames through temporal causal transformers.
Models are trained using molecular dynamics (MD) trajectories data and can generate protein conformation ensembles and motion trajectories conditioned on the input protein amino acid sequence.

This version was trained with tasks of forward simulation and iid sampling at 1:1 ratio.
recommend: For forward simulation and iid sampling tasks
model_id: ConfRover-base-20M-v1.0
name: ConfRover
repo: https://github.com/ByteDance-Seed/ConfRover
paper: https://arxiv.org/abs/2505.17478
demo: https://ByteDance-Seed.github.io/ConfRover
get_started_code: 
```python
from confrover import ConfRover

model = ConfRover.from_pretrained(<model_name>)

model.to("cuda")

model.generate(
    case_id=<case_name>,
    seqres=<amino_acid_sequence>,
    output_dir=</path/to/save/output>,
    task_mode=<"forward"|"iid"|"interp">,
    n_replicates=<int>, # number of replicated trajectories (forward and interp) or total number of conformation samples (iid)
    n_frames=<int>, # number of frames in the trajectory, including the conditioning frames.
    stride_in_10ps=256, # time interval between frames in the unit of 10 ps.
    conditions=..., # information for conditioning frames for forward simulation and interp. See `ConfRover.generate` for more details.
)
```

model_specs: 
ConfRover contains encoder, temporal module, and diffusion decoder.
- The encoder maps the input amino acid sequence (through a folding model) and coordinates of context frames to a latent representation.
- The temporal module models the temporal dependencies between frames using an interleaving of causal transformers (across the temporal dimension) and pairformers (to update structures).
- The diffusion model learns the probability distribution of protein conformations and generates samples conditioned on the input sequence and conditioning representation.

bias_risks_limitations: 
ConfRover is trained on limited MD trajectories data and may not generalize well to out-of-distribution data.
The quality of generated conformations is also limited by the quality of the input data and the computational resources.
Currently, ConfRover only supports protein conformation generation and models the coordinates of heavy atoms.

citation_bibtex: 
```text
@article{confrover2025,
  title={Simultaneous Modeling of Protein Conformation and Dynamics via Autoregression},
  author={Shen, Yuning and Wang, Lihao and Yuan, Huizhuo and Wang, Yan and Yang, Bangji and Gu, Quanquan},
  journal={arXiv preprint arXiv:2505.17478},
  year={2025}
}
```