Scaling RNA Foundation Models through Iterative Refinement
Organizations: Department of Computer Science and Engineering, The Chinese University of Hong Kong
Abstract
Biological function in RNA is encoded across scales, from local sequence motifs to long-range structural dependencies. Conventional RNA foundation models couple parameter capacity and inference depth within a fixed architecture, prescribing a fixed computation graph for each inference. Here we present Eco-RNA, an 80M-parameter RNA foundation model that makes inference depth adjustable through learned iterative refinement. Its shared, input-conditioned Transformer core refines sequence representations through repeated application. The iteration count is selected after training and controls effective depth at fixed model parameters. We train across recurrent depths with masked reconstruction, state regularization, and a one-sided loss that anchors deeper reconstruction to a shallow reference. On the 70-assay RNAGym fitness benchmark, increasing refinement from one to four iterations raises aggregate absolute Spearman correlation from 0.242 to 0.286, with the gain retained through eight iterations. At four iterations, Eco-RNA surpasses the current SOTA Evo 2 40B on all three aggregate fitness metrics despite 500x fewer parameters. Its iteration configurations advance the compute-performance Pareto frontier among the evaluated models. It also outperforms the evaluated baselines in non-coding RNA family classification, RNA secondary-structure prediction, and chemical probing profile prediction. These results establish internal inference computation as an effective scaling dimension for RNA foundation models.
Figures & tables
| Pretraining updates | ||||
|---|---|---|---|---|
| 50k | 0.9357 | 0.8958 | 0.8955 | 0.040 |
| 100k | 0.9070 | 0.8673 | 0.8667 | 0.040 |
| 200k | 0.8829 | 0.8416 | 0.8408 | 0.041 |
| 300k | 0.8772 | 0.8357 | 0.8348 | 0.042 |
| Model | Params | ncRNA accuracy (%) | SSP F1 (%) |
|---|---|---|---|
| RNA-MSM | 96M | 84.85 | 57.98 |
| 3UTRBERT | 87M | 94.32 | 59.41 |
| RNA-FM | 100M | 96.88 | 68.50 |
| RiNALMo | 150M | 91.89 | 75.02 |
| RNAGenesis | 866M | 97.81 | 73.72 |
| Eco-RNA ( ) | 80M | 98.69 | 75.40 |
| DMS | 2A3 | Mean | |||||||
|---|---|---|---|---|---|---|---|---|---|
| Method | AUC | F1 | MAE | AUC | F1 | MAE | AUC | F1 | MAE |
| RNAstructure | 0.646 | 0.605 | 0.326 | 0.721 | 0.667 | 0.300 | 0.683 | 0.636 | 0.313 |
| RNAfold (ViennaRNA) | 0.645 | 0.604 | 0.324 | 0.725 | 0.670 | 0.296 | 0.685 | 0.637 | 0.310 |
| CONTRAfold | 0.659 | 0.608 | 0.372 | 0.740 | 0.684 | 0.330 | 0.700 | 0.646 | 0.351 |
| EternaFold | 0.647 | 0.598 | 0.360 | 0.760 | 0.702 | 0.309 | 0.703 | 0.650 | 0.335 |
| UFold | 0.841 | 0.766 | 0.182 | 0.727 | 0.668 | 0.244 | 0.784 | 0.717 | 0.213 |
Appendix figures & tables7 assets
Supplementary material from the paper’s appendix.
Appendix
| Setting | Value |
|---|---|
| Trainable backbone parameters | 80,856,192 |
| Prelude / recurrent core / coda | Transformer blocks |
| Distinct / executed blocks | / |
| Hidden / feed-forward width | |
| Query / key–value heads; head width | ; |
| Context length / active vocabulary | tokens / tokens |
| Setting | Value |
|---|---|
| Data | OpenGenome2 genomic buckets and non-coding-RNA partition, filtered RNAcentral |
| Updates / maximum length | / input tokens |
| Position selection | ; exponential span scale |
| Replacement after selection | 80% mask / 10% modality-valid random base / 10% unchanged |
| Depth distribution | |
| Backward window | terminal iterations |
| Setting | Value |
|---|---|
| Initialization | Stage-A checkpoint (300,000 updates) |
| Data | OpenGenome2 non-coding-RNA partition and RNAcentral at increased sampling weight; genomic buckets retained |
| Updates / maximum length | / input tokens |
| Position selection | Span masking; , exponential span scale |
| Replacement after selection | 80% mask / 10% random nucleotide / 10% unchanged |
| Depth distribution / backward window | ; |
| Model | Params | Spearman | AUC | MCC |
|---|---|---|---|---|
| GenSLM ( Zvyagin et al., 2023 ) | 2.5B | 0.122 | 0.558 | 0.083 |
| Evo 1 ( Nguyen et al., 2024 ) | 6.5B | 0.124 | 0.561 | 0.091 |
| RiNALMo ( Penić et al., 2025 ) | 650M | 0.155 | 0.582 | 0.122 |
| Nucleotide Transformer ( Dalla-Torre et al., 2025 ) | 2.5B | 0.163 | 0.577 | 0.115 |
| Evo 1.5 ( Arora et al., 2025 ) | 6.5B | 0.177 | 0.583 | 0.131 |
| Nucleotide Transformer v3 ( Boshar et al., 2025 ) | 650M | 0.189 | 0.593 | 0.133 |
| 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | |
|---|---|---|---|---|---|---|---|---|
| nRC family classification | ||||||||
| Accuracy (%) | 97.96 | 98.46 | 98.62 | 98.69 | 98.65 | 98.65 | 98.69 | 98.69 |
| bpRNA TS0 secondary structure | ||||||||
| (%) | 73.10 | 75.10 | 75.38 | 75.40 | 75.47 | 75.48 | 75.47 | 75.48 |
| Precision (%) | 83.00 | 83.66 | 83.70 | 83.67 | 83.69 | 83.69 | 83.69 | 83.70 |
| Recall (%) | 65.31 | 68.13 | 68.57 | 68.62 | 68.72 | 68.74 | 68.72 | 68.72 |
| DMS | 2A3 | Mean | |||||||
|---|---|---|---|---|---|---|---|---|---|
| AUC | F1 | MAE | AUC | F1 | MAE | AUC | F1 | MAE | |
| 1 | 0.836217 | 0.767908 | 0.179242 | 0.806089 | 0.730326 | 0.209712 | 0.821153 | 0.749117 | 0.194477 |
| 2 | 0.846327 | 0.771044 | 0.162480 | 0.853681 | 0.771873 | 0.174473 | 0.850004 | 0.771458 | 0.168476 |
| 3 | 0.854592 | 0.776349 | 0.156537 | 0.866237 | 0.783168 | 0.165125 | 0.860415 | 0.779758 | 0.160831 |
| 4 | 0.853233 | 0.773936 | 0.156149 | 0.873936 | 0.791049 | 0.159765 | 0.863584 | 0.782492 | 0.157957 |
| 5 | 0.855257 | 0.775581 | 0.154953 | 0.875030 | 0.792345 | 0.158763 | 0.865144 | 0.783963 | 0.156858 |
Iterative Refinement
Iterative refinement improves predictions or generated outputs through repeated updates, feedback, and error correction.
See all 48 papers