Cheat sheet

gLM cheat sheet: DNABERT

One reference card for DNABERT: architecture, tokenization, size, context, training data, what it was shown to do and what it does not do. One card in a set that covers one genomic language model at a time.

GenomicsLLMMachine Learning

Genomic language model · Cheat sheet

DNABERT

The model that brought BERT's pretrain-once, fine-tune-often recipe to DNA

First public
bioRxiv, September 2020
Published
Bioinformatics, February 2021
Lab
Northwestern University, Stony Brook University
Architecture
BERT. 12 Transformer layers, 768 hidden units, 12 attention heads. A bidirectional encoder trained by masking contiguous spans of k-mers and predicting them back.
Tokenization
k-mer, overlapping 6-mers
Parameters
~92M
Context
512 tokens, and pretraining sequences up to 510 bases. A separate DNABERT-XL was added for longer inputs.
Training data
The human reference genome. One species.
Used for
Prediction and classification, after task-specific fine-tuning

Headline result

On 690 datasets of measured transcription factor binding (ENCODE ChIP-seq), the only method with mean and median accuracy and F1 all above 0.9 (0.918 and 0.919). Pretrained on human only, it still beat models built from scratch when fine-tuned on mouse.

What it does not do

Human only, not a generator, and 512 tokens of context. Every downstream task needs its own fine-tuning run.

Commonly confused

Often quoted at 110M parameters. That is BERT-base with a 30,000-token text vocabulary. DNABERT's k-mer vocabulary is 4,101 tokens, so its embedding table is far smaller and the count works out near 92M.

Download this card PNG, 1080 x 1350, sized for a post or a slide