Total runs:
1.1K
Run Growth:
-1.1K
Growth Rate:
-97.46%
Sponsored by
Tripo AI
genbio-ai / GB.StructureDecoder
Model's Last Updated: July 26 2026
Introduction of GB.StructureDecoder
Model Details of GB.StructureDecoder
GB.StructureDecoder
GB.StructureDecoder is the decoder-only component of GB.StructureTokenizer for tokenization of protein structures.
Model Description
GB.StructureTokenizer is built on a Vector Quantized Variational Autoencoder (VQ-VAE) architecture with the following components:
- Equivariant Encoder (6M): Encodes backbone structures into a latent space that maintains rotational and translational symmetries using the Equiformer architecture.
- Discrete Codebook: Maps continuous latent vectors into 512 discrete structural tokens.
- Invariant Decoder (300M): Reconstructs full 3D structures, including side chains, from the structural tokens using an architecture adapted from ESMFold.
This model strikes a balance between reconstruction fidelity and structural locality, optimizing its suitability for downstream tasks such as structure prediction, homology detection, and multimodal protein modeling.
Key Features
- Encoding Structures into Tokens (See genbio-ai/GB.StructureEncoder )
- Decoding Tokens into Structures (See below )
- Reconstructing Structures (See genbio-ai/GB.StructureTokenizer )
- Structure Prediction (See this section in genbio-ai/GB.Protein2StructureToken-16B)
How to Use
Please see
experiments/GB.StructureTokenizer
in
Model Generator
for more details.
Setup
Install Model Generator
Decoding Structure Tokens from GB.StructureEncoder
If you have run the encoding task in
genbio-ai/GB.StructureEncoder
with the default
encode.yaml
, the default
decode.yaml
configuration file is already set up to decode the encoded tokens. You don't need to change anything in the configuration file. You can directly run the decoding task using the following command:
CUDA_VISIBLE_DEVICES=0 mgen predict --config=experiments/GB.StructureTokenizer/decode.yaml
Decoding Customized Structure Tokens
To decode protein structures, you will need the structure tokens in
.pt
format and a corresponding codebook file (
codebook.pt
). For ease of use, we recommend preparing the structure tokens in TSV format and then converting them to
.pt
format using the provided script.
The TSV file should include the following columns (an example file is available at
experiments/GB.StructureTokenizer/decode_example_input.tsv
):
-
uid: A unique identifier for the protein sequence. -
sequences: The amino acid sequence (e.g., "LRTPTT"). -
predictions: The structure tokens to be decoded, provided as a list (e.g., "[164, 287, 119, ...]"). The list length must match the length of the amino acid sequence.
After preparing the TSV file, you need to convert the TSV file to the
.pt
format using the following command:
python experiments/GB.StructureTokenizer/struct_token_format_conversion.py your_tsv_file.tsv your_output_pt_file.pt
You also need to prepare a codebook file (
codebook.pt
) that contains the embedding of each token. The codebook could be extracted using this command:
python experiments/GB.StructureTokenizer/extract_structure_tokenizer_codebook.py --output_path your_output_codebook.pt
Then you need to update the
struct_tokens_path
and
codebook_path
in the
decode.yaml
configuration file to point to your structure tokens and codebook file. Alternatively, you can override these parameters when running the command:
CUDA_VISIBLE_DEVICES=0 mgen predict --config experiments/GB.StructureTokenizer/decode.yaml \
--data.init_args.config.struct_tokens_datasets_configs.name="your_dataset_name" \
--data.init_args.config.struct_tokens_datasets_configs.struct_tokens_path="your_structure_tokens.pt" \
--data.init_args.config.struct_tokens_datasets_configs.codebook_path="your_codebook.pt" \
--trainer.callbacks.dict_kwargs.dirpath="your_output_dir"
Input:
-
The encoded tokens saved in
.ptformat. -
The codebook file (
codebook.pt) that contains the embedding of each token.
Output:
-
The decoded protein structures will be saved in the output directory specified in the configuration file. By default it is saved in
logs/protstruct_decode/. - The decoded structures are saved as PDB files.
Notes:
- Decoding the structures could take a long time even when using a GPU.
- Currently, this function only supports single GPU inference due to the file saving mechanism. We plan to support multi-GPU inference in the future.
-
Currently, we don't support specifying the residue index in TSV format. If you need to specify the residue index, you need to modify the
struct_token_format_conversion.pyscript to include the residue index in the TSV file (we may support this feature in the future), or you could provide the.ptfile directly with the desired residue index.
Citation
Please cite GB.StructureTokenizer using the following BibTex code:
@inproceedings{zhang_balancing_2024,
title = {Balancing Locality and Reconstruction in Protein Structure Tokenizer},
url = {https://www.biorxiv.org/content/10.1101/2024.12.02.626366v2},
doi = {10.1101/2024.12.02.626366},
publisher = {bioRxiv},
author = {Zhang, Jiayou and Meynard-Piganeau, Barthelemy and Gong, James and Cheng, Xingyi and Luo, Yingtao and Ly, Hugo and Song, Le and Xing, Eric},
year = {2024},
booktitle={NeurIPS 2024 Workshop on Machine Learning in Structural Biology (MLSB)},
}
Runs of genbio-ai GB.StructureDecoder on huggingface.co
32
Total runs
0
24-hour runs
0
3-day runs
2
7-day runs
9
30-day runs
More Information About GB.StructureDecoder huggingface.co Model
Url of GB.StructureDecoder
GB.StructureDecoder huggingface.co Url
Provider of GB.StructureDecoder huggingface.co
Other API from genbio-ai
Updated:September 13 2025
Total runs:
748
Run Growth:
665
Growth Rate:
90.48%
Total runs:
712
Run Growth:
640
Growth Rate:
90.65%
Total runs:
312
Run Growth:
-354
Growth Rate:
-113.46%
Total runs:
218
Run Growth:
50
Growth Rate:
22.94%
Total runs:
147
Run Growth:
-97
Growth Rate:
-65.99%
Total runs:
136
Run Growth:
52
Growth Rate:
38.24%
Total runs:
66
Run Growth:
-541
Growth Rate:
-819.70%
Total runs:
55
Run Growth:
17
Growth Rate:
30.91%
Total runs:
29
Run Growth:
17
Growth Rate:
58.62%
Total runs:
29
Run Growth:
12
Growth Rate:
41.38%
Total runs:
24
Run Growth:
6
Growth Rate:
27.27%
Total runs:
24
Run Growth:
-8
Growth Rate:
-33.33%
Total runs:
22
Run Growth:
4
Growth Rate:
18.18%
Total runs:
17
Run Growth:
-2
Growth Rate:
-14.29%
Total runs:
15
Run Growth:
5
Growth Rate:
38.46%
Total runs:
15
Run Growth:
6
Growth Rate:
40.00%
Total runs:
15
Run Growth:
8
Growth Rate:
53.33%
Total runs:
15
Run Growth:
2
Growth Rate:
13.33%
Total runs:
15
Run Growth:
-3
Growth Rate:
-20.00%
Total runs:
14
Run Growth:
0
Growth Rate:
0.00%
Total runs:
14
Run Growth:
5
Growth Rate:
38.46%
Total runs:
13
Run Growth:
4
Growth Rate:
36.36%
Total runs:
12
Run Growth:
6
Growth Rate:
50.00%
Total runs:
11
Run Growth:
-6
Growth Rate:
-54.55%
Total runs:
11
Run Growth:
7
Growth Rate:
63.64%
Total runs:
10
Run Growth:
6
Growth Rate:
60.00%
Total runs:
10
Run Growth:
-10
Growth Rate:
-100.00%
Total runs:
10
Run Growth:
5
Growth Rate:
50.00%
Total runs:
10
Run Growth:
5
Growth Rate:
55.56%
Total runs:
10
Run Growth:
2
Growth Rate:
22.22%
Total runs:
10
Run Growth:
4
Growth Rate:
44.44%
Total runs:
9
Run Growth:
-17
Growth Rate:
-188.89%
Total runs:
9
Run Growth:
6
Growth Rate:
66.67%
Total runs:
9
Run Growth:
-8
Growth Rate:
-88.89%
Total runs:
8
Run Growth:
2
Growth Rate:
33.33%
Total runs:
8
Run Growth:
-266
Growth Rate:
-3325.00%
Total runs:
8
Run Growth:
5
Growth Rate:
62.50%
Total runs:
7
Run Growth:
0
Growth Rate:
0.00%
Total runs:
6
Run Growth:
-7
Growth Rate:
-87.50%
Total runs:
6
Run Growth:
-23
Growth Rate:
-383.33%
Total runs:
6
Run Growth:
4
Growth Rate:
66.67%
Total runs:
5
Run Growth:
4
Growth Rate:
80.00%
Total runs:
5
Run Growth:
3
Growth Rate:
60.00%
Total runs:
5
Run Growth:
5
Growth Rate:
100.00%
Total runs:
5
Run Growth:
1
Growth Rate:
20.00%
Total runs:
5
Run Growth:
0
Growth Rate:
0.00%
Total runs:
4
Run Growth:
-1
Growth Rate:
-25.00%
Total runs:
4
Run Growth:
-2
Growth Rate:
-50.00%
Total runs:
4
Run Growth:
2
Growth Rate:
50.00%
Total runs:
3
Run Growth:
-1
Growth Rate:
-33.33%
Total runs:
3
Run Growth:
-5
Growth Rate:
-166.67%
Total runs:
2
Run Growth:
-4
Growth Rate:
-200.00%
Total runs:
2
Run Growth:
-5
Growth Rate:
-250.00%
Total runs:
2
Run Growth:
-2
Growth Rate:
-100.00%
Total runs:
1
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%