Muon Reduces the Training Cost of Regulatory DNA Transformers
Viraj Doshi, Malhar Bhide, Akshat Singh, Yash Rathod, Aditya SivakumariD
Abstract
Gene-therapy design depends on identifying regulatory sequences that drive the right level, timing, and cell-type specificity of expression. Regulatory DNA models offer a way to prioritize such sequences computationally before committing candidates to biological testing. Biological validation involves DNA synthesis, cloning, cell culture, sequencing, and functional screening, so training compute is part of the same constrained discovery pipeline rather than an isolated modeling expense. Reducing the compute required to reach a target pretraining quality could shift time and budget toward larger candidate screens, additional assays, more cell contexts, and broader follow-up validation. Given that Adam-style optimizers are widely used for training genomic sequence models, we study whether Muon can provide a more compute-efficient alternative for regulatory DNA pretraining. We provide an in-depth analysis by training Transformer models (26M-420M parameters) on ENCODE cis-regulatory sequences with Adam and Muon while holding architecture, data, and non-optimizer hyperparameters fixed and varying optimizer family, norm-control scheme, learning rate, and model width. In the largest-scale matched-target comparison, Muon reaches Adam-matched perplexity targets with a median FLOP reduction of 35.4% and a median wall-clock time reduction of 38.5%. The analysis further shows that optimizer rankings depend on norm control: independent weight decay pairs more favorably with Muon than Hyperball in this setting. These findings indicate that optimizer update structure and norm-control choices are practical levers for reducing the training resources required to reach matched perplexity targets in regulatory DNA pretraining.
02 /Specification & Classification
Click any tag to browse matching papersReproducibility ledger
Nobody has reported trying to reproduce this paper yet. Did you replicate an experiment or run the code? Your report saves the next lab a week.
Sign in to file a reproducibility report.
Related papers
- Fine-Tuning Transformers For Genomic Tasks
Vlastimil Martinek, David Čechák, … Petr Šimeček·bioRxiv·Feb 7, 2022
- Parameter-Efficient Fine-Tuning of a Supervised Regulatory Sequence Model
Han Yuan, Johannes Linder, David R. Kelley·bioRxiv·May 26, 2025
- ARSENAL: Learning Transferable Regulatory DNA Representations with Targeted Short-Context Language Models
Aman Patel, Anshul Bharat Kundaje·bioRxiv·Jul 12, 2026
- Designing DNA With Tunable Regulatory Activity Using Discrete Diffusion
Anirban Sarkar, Alejandra Durán, … Peter K. Koo·bioRxiv·May 23, 2024
- DNABERT: pre-trained Bidirectional Encoder Representations from Transformers model for DNA-language in genome
Yanrong Ji, Zhihan Zhou, … Ramana V. Davuluri·BioinformaticsPeer-reviewed·Feb 3, 2021
- Enigma: An Efficient Model for Deciphering Regulatory Genomics
Andreas Jung, Helen Zhu, … Brendan J. Frey·bioRxiv·Dec 19, 2025