atlas

Research map / Architectures and efficiency / Attention and transformers

Caduceus: Bi-Directional Equivariant Long-Range DNA Sequence Modeling

ICML 2024 · Architectures and efficiency › Attention and transformers · approach: State space and recurrent sequence models · 54 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Introduces BiMamba and MambaDNA (reverse-complement equivariant Mamba) and uses them to build Caduceus, a family of bi-directional, RC-equivariant DNA foundation models pre-trained with masked language modeling on the human genome.

Problem

Genomic sequence models must capture long-range interactions, bi-directional context, and reverse-complement (RC) symmetry of DNA, which prior long-context DNA LMs (e.g., HyenaDNA, Transformer-based models) handle only partially and at high cost.

Approach

Extends the causal Mamba block to bi-directional BiMamba with shared projection weights, and to MambaDNA by applying the block to a sequence and its reverse complement with shared parameters and channel splitting; Caduceus-PS bakes RC equivariance into the model via parameter sharing and embeddings/LM head, while Caduceus-Ph uses post-hoc conjoining with RC augmentation.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use Caduceus-PS instead of HyenaDNAGenomics benchmark models · Mouse Enhancersaccuracy: 0.780 -> 0.793Helped
Use Caduceus-Ph instead of MambaGenomics benchmark models · Coding vs. Intergenomicaccuracy: 0.904 -> 0.915Helped
Use Caduceus-Ph instead of HyenaDNAGenomics benchmark models · Human vs. Wormaccuracy: 0.964 -> 0.973Helped
Use Caduceus-Ph instead of MambaGenomics benchmark models · Human Enhancers Cohnaccuracy: 0.732 -> 0.747Helped
Use Caduceus-PS instead of HyenaDNAGenomics benchmark models · Human Enhancer Ensemblaccuracy: 0.849 -> 0.900Helped
Use Caduceus-PS instead of HyenaDNAGenomics benchmark models · Human Regulatoryaccuracy: 0.869 -> 0.873Helped
Use Caduceus-Ph instead of HyenaDNAGenomics benchmark models · Human OCR Ensemblaccuracy: 0.783 -> 0.828Helped
Use Caduceus-Ph instead of HyenaDNAGenomics benchmark models · Human NonTATA Promotersaccuracy: 0.944 -> 0.946Helped
Add RC equivariance (ablation)Caduceus · Genomics benchmark tasks overallaccuracyMixed
Use parameter sharing rather than post-hoc conjoining (ablation)Caduceus · Genomics benchmark tasks overallaccuracyMixed
Use Caduceus instead of HyenaDNALong-range variant effect prediction models · SNPs more than 100k from nearest TSSAUROCHelped
Use Caduceus-PS instead of Nucleotide Transformer v2Long-range variant effect prediction models · Variant effect prediction across TSS distancesAUROCHelped

Surprising findings

Limitations

Experimental setup

Datasets: HG38 human reference genome Genomics Benchmarks Nucleotide Transformer benchmark tasks Enformer-derived variant effect prediction dataset (Chromosomes 9 and 10 test)

Benchmarks: Genomics Benchmarks Nucleotide Transformer tasks variant effect prediction on gene expression (Enformer-derived)

Metrics: top-1 accuracy MCC F1 AUROC

Baselines: HyenaDNA Mamba NTP Nucleotide Transformer v2 DNABERT-2 Enformer CNN

Builds on: Mamba HyenaDNA BERT RCPS (reverse complement parameter sharing) post-hoc conjoining

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy