SpecMER: Fast Protein Generation with K-mer Guided Speculative Decoding

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Walton, Thomas, Tsui, Darin, Musharaf, Aryan, Aghazadeh, Amirali
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914057365225472
author Walton, Thomas
Tsui, Darin
Musharaf, Aryan
Aghazadeh, Amirali
author_facet Walton, Thomas
Tsui, Darin
Musharaf, Aryan
Aghazadeh, Amirali
contents Autoregressive models have transformed protein engineering by enabling the generation of novel protein sequences beyond those found in nature. However, their sequential inference introduces significant latency, limiting their utility in high-throughput protein screening. Speculative decoding accelerates generation by employing a lightweight draft model to sample tokens, which a larger target model then verifies and refines. Yet, in protein sequence generation, draft models are typically agnostic to the structural and functional constraints of the target protein, leading to biologically implausible outputs and a shift in the likelihood distribution of generated sequences. We introduce SpecMER (Speculative Decoding via k-mer Guidance), a novel framework that incorporates biological, structural, and functional priors using k-mer motifs extracted from multiple sequence alignments. By scoring candidate sequences in parallel and selecting those most consistent with known biological patterns, SpecMER significantly improves sequence plausibility while retaining the efficiency of speculative decoding. SpecMER achieves 24-32% speedup over standard autoregressive decoding, along with higher acceptance rates and improved sequence likelihoods.
format Preprint
id arxiv_https___arxiv_org_abs_2509_21689
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle SpecMER: Fast Protein Generation with K-mer Guided Speculative Decoding
Walton, Thomas
Tsui, Darin
Musharaf, Aryan
Aghazadeh, Amirali
Machine Learning
Autoregressive models have transformed protein engineering by enabling the generation of novel protein sequences beyond those found in nature. However, their sequential inference introduces significant latency, limiting their utility in high-throughput protein screening. Speculative decoding accelerates generation by employing a lightweight draft model to sample tokens, which a larger target model then verifies and refines. Yet, in protein sequence generation, draft models are typically agnostic to the structural and functional constraints of the target protein, leading to biologically implausible outputs and a shift in the likelihood distribution of generated sequences. We introduce SpecMER (Speculative Decoding via k-mer Guidance), a novel framework that incorporates biological, structural, and functional priors using k-mer motifs extracted from multiple sequence alignments. By scoring candidate sequences in parallel and selecting those most consistent with known biological patterns, SpecMER significantly improves sequence plausibility while retaining the efficiency of speculative decoding. SpecMER achieves 24-32% speedup over standard autoregressive decoding, along with higher acceptance rates and improved sequence likelihoods.
title SpecMER: Fast Protein Generation with K-mer Guided Speculative Decoding
topic Machine Learning
url https://arxiv.org/abs/2509.21689