A codon-level language model for mRNA that encodes synonymous-codon symmetry directly into its architecture, as cyclic subgroups of SO(2), rather than leaving the model to infer that structure from data.
Prior codon models (CodonBERT, HELM) tokenize by codon but leave synonymous relationships implicit. Equi-mRNA enforces them by construction: an auxiliary equivariance loss and symmetry-aware pooling keep codon substitutions that preserve amino-acid identity mapped to consistent rotations throughout training.
25M coding sequences sampled from 56M RefSeq entries (20–512 codons, canonical bases only). Ablations were run on a stratified 1M-sequence subset before scaling the winning configuration to the full corpus.
Six biologically-driven benchmarks spanning expression, stability, and regulatory switching (MLOS, mRFP, E. coli, Tc-Riboswitch, iCodon, and SARS-CoV-2 degradation), evaluated against nucleotide- and codon-based baselines including RNA-FM, CodonBERT, and HELM.
The paper is peer-reviewed and public on arXiv (2508.15103), with the full architecture, training protocol, and hyperparameters documented in its appendix.
It's the representation behind expression prediction, stability assessment, mRNA generation, and therapeutics design inside the co-scientist. See Used in below.
The paper covers the full architecture, training procedure, and evaluation methodology.