FoldToken: Learning Protein Language via Vector Quantization and Beyond
Not on the map. Why?
Abstract (arXiv)
Is there a foreign language describing protein sequences and structures simultaneously? Protein structures, represented by continuous 3D points, have long posed a challenge due to the contrasting modeling paradigms of discrete sequences. We introduce FoldTokenizer to represent protein sequence-structure as discrete symbols. This innovative approach involves projecting residue types and structures into a discrete space, guided by a reconstruction loss for information preservation. We refer to the learned discrete symbols as FoldToken, and the sequence of FoldTokens serves as a new protein language, transforming the protein sequence-structure into a unified modality. We apply the created protein language on general backbone inpainting and antibody design tasks, building the first GPT-style model (FoldGPT) for sequence-structure co-generation with promising results. Key to our success is the substantial enhancement of the vector quantization module, Soft Conditional Vector Quantization (SoftCVQ).
Key facts. Cited by 21 papers in the map (data through 7 Oct 2026). Builds on 30 of its 50 references that are in the map. First posted on arXiv 4 Feb 2024.
Builds on
30 of 50 references.
References · 30
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2017Neural Discrete Representation Learningcited by 4,650
- 2020Taming Transformers for High-Resolution Image Synthesiscited by 3,229
- 2021SoundStream: An End-to-End Neural Audio Codeccited by 756
- 2022Autoregressive Image Generation using Residual Quantizationcited by 706
- 2021Vector-quantized Image Modeling with Improved VQGANcited by 656
- 2023Language Model Beats Diffusion -- Tokenizer is Key to Visual Generationcited by 633
- 2023Finite Scalar Quantization: VQ-VAE Made Simplecited by 425
- 2020Learning from Protein Structure with Geometric Vector Perceptronscited by 308
- 2022Diffusion probabilistic modeling of protein backbones in 3D for the motif-scaffolding problemcited by 229
- 2022Protein Representation Learning by Geometric Structure Pretrainingcited by 161
18 more
- 2022Protein structure generation via folding diffusioncited by 113
- 2021Iterative Refinement Graph Neural Network for Antibody Sequence-Structure Co-designcited by 82
- 2022PiFold: Toward effective and efficient protein inverse foldingcited by 82
- 2022Conditional Antibody Design as 3D Equivariant Graph Translationcited by 75
- 2022AlphaDesign: A graph protein design method and benchmark on AlphaFoldDBcited by 39
- 2019Differentiable Product Quantization for End-to-End Embedding Compressioncited by 36
- 2022Image Compression with Product Quantized Masked Image Modelingcited by 32
- 2022Protein Sequence and Structure Co-Design with Equivariant Translationcited by 31
- 2014Stacked Quantizers for Compositional Vector Compressioncited by 30
- 2022Regression as Classification: Influence of Task Formulation on Neural Network Featurescited by 29
- 2023A step towards understanding why classification helps regressioncited by 15
- 2023Knowledge-Design: Pushing the Limit of Protein Design via Knowledge Refinementcited by 11
- 2023DiffSDS: A language diffusion model for protein backbone inpainting under geometric conditions and constraintscited by 10
- 2023Cross-Gate MLP with Protein Complex Invariant Embedding is A One-Shot Antibody Designercited by 8
- 2023Progressive Multi-Modality Learning for Inverse Protein Foldingcited by 5
- 2023Streaming Active Learning for Regression Problems Using Regression via Classificationcited by 2
- 2023Functional Geometry Guided Protein Sequence and Backbone Structure Co-Designcited by 2
- 2023Efficiently Predicting Protein Stability Changes Upon Single-point Mutation with Large Language Modelscited by 1
Built on by
Citing papers per year: 2024: 9; 2025: 3; 2026: 9.
21 papers.
Sort:
By area:
- 2024Addressing Representation Collapse in Vector Quantized Models with One Linear Layercited by 77
- 2024DPLM-2: A Multimodal Diffusion Protein Language Modelcited by 59
- 2025Token Reduction Should Go Beyond Efficiency in Generative Models -- From Vision, Language to Multimodalitycited by 49
- 2024Leveraging Biomolecule and Natural Language through Multi-Modal Learning: A Surveycited by 16
- 2024Structure Language Models for Protein Conformation Generationcited by 16
- 2024Learning the Language of Protein Structurecited by 9
- 2024Bio2Token: All-atom tokenization of any biomolecular structure with Mambacited by 5
- 2025Flow Autoencoders are Effective Protein Tokenizerscited by 4
- 2024Constraint Decoupled Latent Diffusion for Protein Backmappingcited by 2
- 2026Adaptive Protein Tokenizationcited by 2
- 2024FoldToken2: Learning compact, invariant and generative protein structure languagecited by 1
- 2024Tokenizing 3D Molecule Structure with Quantized Spherical Coordinatescited by 1
9 more
- 2026VecFormer: Towards Efficient and Generalizable Graph Transformer with Graph Token Attentioncited by 1
- 2026PaperFit: Vision-in-the-Loop Typesetting Optimization for Scientific Documentscited by 1
- 2025AlphaFold Database Debiasing for Robust Inverse Foldingcited by 0
- 2026Reinforcement-Learned Unequal Error Protection for Quantized Semantic Embeddingscited by 0
- 2026RiboSphere: Learning Unified and Efficient Representations of RNA Structurescited by 0
- 2026Mitigating Premature Discretization with Progressive Quantization for Robust Vector Tokenizationcited by 0
- 2026Yeti: A compact protein structure tokenizer for reconstruction and multi-modal generationcited by 0
- 2026Discrete Diffusion Models: A Unified Framework from Tokenization to Generationcited by 0
- 2026Flow-Matching-Based Protein Structure Tokenizer Made Efficient and Easycited by 0
- 2026Flow-Matching-Based Protein Structure Tokenizer Made Efficient and Easy27 Sep 2026
- 2026Discrete Diffusion Models: A Unified Framework from Tokenization to Generation15 Jul 2026
- 2026PaperFit: Vision-in-the-Loop Typesetting Optimization for Scientific Documents11 May 2026
- 2026Yeti: A compact protein structure tokenizer for reconstruction and multi-modal generation11 May 2026
- 2026RiboSphere: Learning Unified and Efficient Representations of RNA Structures20 Mar 2026
- 2026Mitigating Premature Discretization with Progressive Quantization for Robust Vector Tokenization17 Mar 2026
- 2026VecFormer: Towards Efficient and Generalizable Graph Transformer with Graph Token Attention23 Feb 2026
- 2026Adaptive Protein Tokenization6 Feb 2026
- 2026Reinforcement-Learned Unequal Error Protection for Quantized Semantic Embeddings1 Jan 2026
- 2025Flow Autoencoders are Effective Protein Tokenizers30 Sep 2025
- 2025AlphaFold Database Debiasing for Robust Inverse Folding10 Jun 2025
- 2025Token Reduction Should Go Beyond Efficiency in Generative Models -- From Vision, Language to Multimodality23 May 2025
9 more
- 2024Tokenizing 3D Molecule Structure with Quantized Spherical Coordinates2 Dec 2024
- 2024Addressing Representation Collapse in Vector Quantized Models with One Linear Layer4 Nov 2024
- 2024Bio2Token: All-atom tokenization of any biomolecular structure with Mamba24 Oct 2024
- 2024Structure Language Models for Protein Conformation Generation24 Oct 2024
- 2024DPLM-2: A Multimodal Diffusion Protein Language Model17 Oct 2024
- 2024Constraint Decoupled Latent Diffusion for Protein Backmapping17 Oct 2024
- 2024FoldToken2: Learning compact, invariant and generative protein structure language11 Jun 2024
- 2024Learning the Language of Protein Structure24 May 2024
- 2024Leveraging Biomolecule and Natural Language through Multi-Modal Learning: A Survey3 Mar 2024
- 2025Flow Autoencoders are Effective Protein Tokenizerscited by 4
- 2024Constraint Decoupled Latent Diffusion for Protein Backmappingcited by 2
- 2026Discrete Diffusion Models: A Unified Framework from Tokenization to Generationcited by 0
- 2026Flow-Matching-Based Protein Structure Tokenizer Made Efficient and Easycited by 0
- 2026Flow-Matching-Based Protein Structure Tokenizer Made Efficient and Easy27 Sep 2026
- 2026Discrete Diffusion Models: A Unified Framework from Tokenization to Generation15 Jul 2026
- 2025Flow Autoencoders are Effective Protein Tokenizers30 Sep 2025
- 2024Constraint Decoupled Latent Diffusion for Protein Backmapping17 Oct 2024
- 2024Addressing Representation Collapse in Vector Quantized Models with One Linear Layercited by 77
- 2024DPLM-2: A Multimodal Diffusion Protein Language Modelcited by 59
- 2024Leveraging Biomolecule and Natural Language through Multi-Modal Learning: A Surveycited by 16
- 2024Structure Language Models for Protein Conformation Generationcited by 16
- 2024Learning the Language of Protein Structurecited by 9
- 2026Adaptive Protein Tokenizationcited by 2
- 2024FoldToken2: Learning compact, invariant and generative protein structure languagecited by 1
- 2024Tokenizing 3D Molecule Structure with Quantized Spherical Coordinatescited by 1
- 2026VecFormer: Towards Efficient and Generalizable Graph Transformer with Graph Token Attentioncited by 1
- 2026PaperFit: Vision-in-the-Loop Typesetting Optimization for Scientific Documentscited by 1
- 2025AlphaFold Database Debiasing for Robust Inverse Foldingcited by 0
- 2026Reinforcement-Learned Unequal Error Protection for Quantized Semantic Embeddingscited by 0
12 most cited of 14 All 14 (JSON)
- 2026PaperFit: Vision-in-the-Loop Typesetting Optimization for Scientific Documents11 May 2026
- 2026Yeti: A compact protein structure tokenizer for reconstruction and multi-modal generation11 May 2026
- 2026RiboSphere: Learning Unified and Efficient Representations of RNA Structures20 Mar 2026
- 2026VecFormer: Towards Efficient and Generalizable Graph Transformer with Graph Token Attention23 Feb 2026
- 2026Adaptive Protein Tokenization6 Feb 2026
- 2026Reinforcement-Learned Unequal Error Protection for Quantized Semantic Embeddings1 Jan 2026
- 2025AlphaFold Database Debiasing for Robust Inverse Folding10 Jun 2025
- 2024Tokenizing 3D Molecule Structure with Quantized Spherical Coordinates2 Dec 2024
- 2024Addressing Representation Collapse in Vector Quantized Models with One Linear Layer4 Nov 2024
- 2024Structure Language Models for Protein Conformation Generation24 Oct 2024
- 2024DPLM-2: A Multimodal Diffusion Protein Language Model17 Oct 2024
- 2024FoldToken2: Learning compact, invariant and generative protein structure language11 Jun 2024
12 newest of 14 All 14 (JSON)