cs.LGDate pending

When do cheap embeddings beat protein language models? A theoretically-grounded hashing sketch for biological sequence classification

Authors: Sarwan AliTaslim MuradImdadullah KhanSafi Faizullah

Abstract

\textbf{Motivation:} Pre-trained protein language models (PLMs) such as ESM-2 have become the default representation for biological sequence tasks, but they are computationally heavy and require GPUs both for embedding and for fine-tuning. Whether they are actually necessary for sequence \emph{classification}, as opposed to structure prediction, is rarely tested against strong, principled, lightweight alternatives. This question has direct practical stakes for large-scale genomic surveillance, where embedding millions of sequences on commodity hardware is a recurring bottleneck.\ \textbf{Results:} We introduce Murmur2Vec, an alignment-free, training-free embedding that aggregates kk-mer counts into a small hash table via the deterministic MurmurHash function, and we cast it as a randomized sketch of the classical kk-mer spectrum kernel. We provide a complete theoretical treatment: closed-form bias/variance of the inner product, an unbiased signed variant with a Johnson--Lindenstrauss-type concentration bound, an excess-risk bound for downstream linear classifiers that makes the bias--variance trade-off in the hash-table size explicit, and an implicit-regularization mechanism by which collisions damage frequent non-discriminative kk-mers more than rare lineage-defining ones. Across four classification tasks, SARS-CoV-2 spike lineage (22 classes), HIV-1 Env subtype (8 classes), and two protein-family benchmarks (8 and 6 classes), Murmur2Vec matches a LoRA-fine-tuned 650M-parameter ESM-2 model on the two tasks for which LoRA fine-tuning was run to convergence (SARS-CoV-2 and HIV-1) and ties frozen ESM-2 on the two protein-family tasks, and it \emph{outperforms} the fine-tuned model on the hardest task (SARS-CoV-2 lineage: 0.8540.854 vs.\ 0.8070.807 accuracy; macro-F1 0.6840.684 vs.\ 0.4010.401).

Explore similar work

CardsList
  1. Task- and dataset-specific information in protein language models

    Aug 12, 2026Roman Joeres, Ilya Senatorov, Anastasia Kolchina +2

  2. Viral Proteins Reveal Geometry of Protein Language Models

    Jun 10, 2026Arthur Bigot, Harmon Bhasin, Core Francisco Park +2Graph Language ModelsViral Sharing

  3. ProtSent: Protein Sentence Transformers

    May 7, 2026Dan Ofer, Oriel Perets, Michal Linial +1Protein Data BankGlobal Embeddings