English
Related papers

Related papers: Flow Autoencoders are Effective Protein Tokenizers

200 papers

We investigate the impact of tokeniser pretraining on the accuracy and efficiency of physics emulation. Modern high-resolution simulations produce vast volumes of data spanning diverse physical regimes and scales. Training foundation models…

Machine Learning · Computer Science 2026-03-13 Hadi Sotoudeh , Payel Mukhopadhyay , Ruben Ohana , Michael McCabe , Neil D. Lawrence , Shirley Ho , Miles Cranmer

This work proposes kernel transform learning. The idea of dictionary learning is well known; it is a synthesis formulation where a basis is learnt along with the coefficients so as to generate or synthesize the data. Transform learning is…

Computer Vision and Pattern Recognition · Computer Science 2020-08-10 Jyoti Maggu , Angshul Majumdar

Audio tokenization has emerged as a critical component in end-to-end audio language models, enabling efficient discrete representation learning for both audio understanding and generation tasks. However, existing audio tokenizers face…

Audio and Speech Processing · Electrical Eng. & Systems 2026-04-15 Xiangyu Zhang , Benjamin John Southwell , Siqi Pan , Xinlei Niu , Beena Ahmed , Julien Epps

We propose TRACE, a structure-aware framework leveraging diffusion models for localized character encoding to embed data. Unlike existing methods that rely on edge features or pre-defined codebooks, TRACE exploits character structures that…

Computer Vision and Pattern Recognition · Computer Science 2026-03-16 Jiale Meng , Jie Zhang , Runyi Hu , Zhe-Ming Lu , Tianwei Zhang , Yiming Li

RFdiffusion is a popular and well-established model for generation of protein structures. However, this generative process offers limited insight into its internal representations and how they contribute to the final protein structure.…

Quantitative Methods · Quantitative Biology 2025-12-01 Wojciech Zarzecki , Paulina Szymczak , Ewa Szczurek , Kamil Deja

Machine-learning models have demonstrated a great ability to learn complex patterns and make predictions. In high-dimensional nonlinear problems of fluid dynamics, data representation often greatly affects the performance and…

Fluid Dynamics · Physics 2022-07-29 Runze Li , Yufei Zhang , Haixin Chen

Recent multimodal models for instruction-based face editing enable semantic manipulation but still struggle with precise attribute control and identity preservation. Structural facial representations such as landmarks are effective for…

Computer Vision and Pattern Recognition · Computer Science 2026-01-30 Zhenghao Zhang , Ziying Zhang , Junchao Liao , Xiangyu Meng , Qiang Hu , Siyu Zhu , Xiaoyun Zhang , Long Qin , Weizhi Wang

Can continuous diffusion models bring the same performance breakthrough on natural language they did for image generation? To circumvent the discrete nature of text data, we can simply project tokens in a continuous space of embeddings, as…

Protein structure prediction has reached revolutionary levels of accuracy on single structures, yet distributional modeling paradigms are needed to capture the conformational ensembles and flexibility that underlie biological function.…

Biomolecules · Quantitative Biology 2023-04-06 Bowen Jing , Ezra Erives , Peter Pao-Huang , Gabriele Corso , Bonnie Berger , Tommi Jaakkola

Token representations influence the efficiency and adaptability of language models, yet conventional tokenization strategies impose rigid segmentation boundaries that do not adjust dynamically to evolving contextual relationships. The…

Computation and Language · Computer Science 2025-08-11 Alistair Dombrowski , Beatrix Engelhardt , Dimitri Fairbrother , Henry Evidail

Tokens are discrete representations that allow modern deep learning to scale by transforming high-dimensional data into sequences that can be efficiently learned, generated, and generalized to new tasks. These have become foundational for…

Computer Vision and Pattern Recognition · Computer Science 2026-02-05 Levi Lingsch , Georgios Kissas , Johannes Jakubik , Siddhartha Mishra

The use of deep learning methods for modeling fluid flow has drawn a lot of attention in the past few years. In situations where conventional numerical approaches can be computationally expensive, these techniques have shown promise in…

Current vision systems typically assign fixed-length representations to images, regardless of the information content. This contrasts with human intelligence - and even large language models - which allocate varying representational…

Computer Vision and Pattern Recognition · Computer Science 2024-11-05 Shivam Duggal , Phillip Isola , Antonio Torralba , William T. Freeman

Protein backbone generation plays a central role in de novo protein design and is significant for many biological and medical applications. Although diffusion and flow-based generative models provide potential solutions to this challenging…

Machine Learning · Computer Science 2025-05-29 Angxiao Yue , Zichong Wang , Hongteng Xu

Current tokenization methods process sequential data without accounting for signal quality, limiting their effectiveness on noisy real-world corpora. We present QA-Token (Quality-Aware Tokenization), which incorporates data reliability…

Artificial Intelligence · Computer Science 2026-02-09 Arvid E. Gollwitzer , Paridhi Latawa , David de Gruijl , Deepak A. Subramanian , Adrián Noriega de la Colina

Tokenizer adaptation plays an important role in adapting pre-trained language models to new domains or languages. In this work, we address two complementary aspects of this process: vocabulary extension and pruning. The common approach to…

Computation and Language · Computer Science 2026-03-24 Taido Purason , Pavel Chizhov , Ivan P. Yamshchikov , Mark Fishel

Diffusion and flow-based generative models have shown strong potential for image restoration. However, image denoising under unknown and varying noise conditions remains challenging, because the learned vector fields may become inconsistent…

Computer Vision and Pattern Recognition · Computer Science 2026-04-06 Jigang Duan , Genwei Ma , Xu Jiang , Wenfeng Xu , Ping Yang , Xing Zhao

Tokenization is a foundational step in natural language processing (NLP) tasks, bridging raw text and language models. Existing tokenization approaches like Byte-Pair Encoding (BPE) originate from the field of data compression, and it has…

Computation and Language · Computer Science 2024-10-08 Craig W. Schmidt , Varshini Reddy , Haoran Zhang , Alec Alameddine , Omri Uzan , Yuval Pinter , Chris Tanner

Multimodal protein language models (PLMs) integrate sequence and token-based structural information, serving as a powerful foundation for protein modeling, generation, and design. However, the reliance on tokenizing 3D structures into…

Machine Learning · Computer Science 2025-06-13 Cheng-Yen Hsieh , Xinyou Wang , Daiheng Zhang , Dongyu Xue , Fei Ye , Shujian Huang , Zaixiang Zheng , Quanquan Gu

We introduce Purrception, a variational flow matching approach for vector-quantized image generation that provides explicit categorical supervision while maintaining continuous transport dynamics. Our method adapts Variational Flow Matching…