RNAGenScape: Latent RNA Design Framework
- RNAGenScape is a latent-space neural framework that jointly represents RNA sequences and structures to generate and optimize biologically valid molecules.
- It spans two formulations: the 2021 model emphasizes RNA secondary structure generation via variational autoencoding, while the 2025 model focuses on mRNA property optimization using an organized autoencoder and Langevin dynamics.
- Key operations include latent-space navigation with manifold projection to ensure that decoded sequences remain close to biologically feasible configurations.
RNAGenScape denotes latent-space neural frameworks for RNA design in which sequence representations are organized so that generation, optimization, and interpolation remain aligned with biochemical or functional constraints. In the 2021 formulation, the system is presented as a model for neural representation and generation of RNA secondary structures, jointly embedding primary sequence, dot-bracket structure, planar molecular graphs, and junction-tree decompositions within a variational architecture (Yan et al., 2021). In the 2025 formulation, RNAGenScape is presented as a property-guided optimization and interpolation framework for mRNA sequences, combining an organized autoencoder, a denoising manifold projector, and manifold Langevin dynamics to steer sequences toward desired property regimes while remaining close to a learned viable manifold (Liao et al., 14 Oct 2025).
1. Conceptual scope and lineage
A recurring source of confusion is that RNAGenScape does not refer to a single frozen architecture. The term spans two related but distinct formulations. The earlier system focuses on RNA secondary-structure-aware representation learning and variational generation; the later system focuses on mRNA property optimization under scarce and imbalanced data through iterative latent updates with projection back to the manifold. Both formulations center latent-space geometry, but they do so with different inductive biases and optimization operators.
| Aspect | 2021 formulation | 2025 formulation |
|---|---|---|
| Molecular object | RNA sequence + secondary structure | mRNA sequence |
| Core machinery | VAE + CNF prior + decoders | Organized autoencoder + projector + Langevin dynamics |
| Main operations | generation, targeted optimization | property-guided optimization, interpolation |
In the 2021 system, the latent variable is used to jointly represent sequence and structural modalities and to support targeted optimization with respect to RNA-binding protein interaction. In the 2025 system, the latent variable is explicitly organized by a target property such as translation efficiency, stability, or ribosome load, and updates are performed by a discrete Langevin rule followed by a learned projection. This suggests a methodological shift from direct generative modeling of RNA structure toward controlled navigation on a learned mRNA manifold.
2. Multi-modal RNA representation in the 2021 formulation
The 2021 formulation represents an RNA molecule in four coupled forms: the primary sequence with ; the dot-bracket string with ; the planar molecular graph whose nodes are nucleotides and whose edges include backbone edges and base-pair edges; and a junction-tree hypergraph in which each hypernode is labeled as one of four substructure types , corresponding to stem, hairpin, internal loop, and multiloop (Yan et al., 2021).
Three encoders are defined over these representations. The sequence-plus-dot-bracket encoder, denoted LSTMVAE, uses a vocabulary of size consisting of all 0 pairs, followed by one-hot embedding, a stacked Bi-LSTM, multi-head self-attention, and global max-pooling to obtain 1, from which linear heads produce 2 and 3. The graph-based encoder, GraphVAE, builds an edge-feature MPNN with gating. For each directed edge 4, it initializes
5
where 6 is a one-hot nucleotide embedding and 7 encodes backbone versus base-pair status. After 8 message-passing steps, node embeddings are aggregated, processed by a Bi-LSTM over 9, and max-pooled to obtain 0.
The hierarchical encoder, HierVAE, adds an explicit substructure decomposition. It first computes nucleotide embeddings 1 with the G-MPNN, then forms hypernode inputs
2
where 3 is the assignment map from hypernodes to nucleotides. A tree-GRU message-passing procedure on the junction tree generates hypernode states, and a depth-first traversal of 4 through another Bi-LSTM followed by max-pooling yields 5. Linear heads then produce 6 and 7.
The significance of this representational stack is the explicit integration of RNA folding regularity into the encoder. The molecular graph captures local connectivity and base pairing, while the junction-tree hierarchy encodes higher-order structural motifs. A plausible implication is that the model’s latent geometry is shaped not only by sequence similarity but also by topological regularities of nested secondary structure.
3. Variational generation and targeted design of RNA secondary structures
The 2021 system is trained as a variational autoencoder with a learnable prior 8 and decoder 9, using an approximate posterior 0. Its objective is the ELBO
1
with 2 annealed during training (Yan et al., 2021).
The prior is implemented as a continuous normalizing flow. The model transforms 3 to 4, with density
5
At training time, 6 is evaluated exactly by sampling 7 and using the CNF density; at test time, one samples 8 and inverts the flow to obtain 9.
Two decoders are used. The linear string decoder is a single-layer unidirectional LSTM whose initial hidden state is a linear projection of 0 and which emits one of the 12 1 tokens plus an end symbol under teacher forcing and cross-entropy loss. The hierarchical tree-plus-segment decoder interleaves topology prediction, node-label prediction, and segment decoding during a depth-first traversal, with the segment LSTM initialized by
2
Structural regularization is enforced at inference time by hard masks on decoder logits. The rules are: base-pair complementarity restricted to A–U, G–C, or G–U; hairpin loops with at least 3 unpaired nucleotides, so that if an open “(” occurs at position 3 then its matching “)” at 4 requires 5; and pairing at most once per nucleotide, with no crossing or pseudoknots. These constraints are designed to guarantee valid, nested, biochemically feasible secondary structures.
The same formulation also introduces a semi-supervised VAE for RBP-binding properties. A small MLP classifier 6 is placed on top of 7, and the joint loss is
8
Targeted design is then performed by activation maximization:
9
optionally with small Gaussian noise, followed by decoding and evaluation by an external oracle.
4. Organized autoencoder for mRNA property supervision
The 2025 formulation defines RNAGenScape as a property-guided manifold Langevin dynamics framework for mRNA sequences. Its front end is an organized autoencoder (OAE). Inputs are one-hot encoded mRNAs of length 0 over a 7-token alphabet: 1, A, U, T, G, C, N. The encoder 2 consists of three 1D-convolutional blocks with GroupNorm, GELU, and SE-block, followed by adaptive average pooling to length 8 and a linear head that produces 3 with 4. A property predictor 5 is a 3-layer MLP with GELU activations and dropout 6, mapping 7. The decoder 8 is a progressive 1D decoder that linearly projects 9 to a seed map of shape 0, repeatedly upsamples by factor 2 with two residual convolutional blocks until reaching 1 positions, and refines with two final convolutional blocks to produce logits 2 (Liao et al., 14 Oct 2025).
The latent space is explicitly supervised by the target property. In addition to reconstructing 3, the model requires 4 to be predictive of 5, where the target may be translation efficiency, stability, or ribosome load. The loss is
6
with 7 for Zebrafish and 8 for the smaller datasets.
This organization by property is central to the later RNAGenScape formulation. Rather than treating optimization as an external search over a latent space learned only for reconstruction, the OAE makes the latent representation itself predictive of the target variable. This reduces the mismatch between decoding fidelity and optimization direction.
5. Manifold projector, Langevin dynamics, and interpolation
After training the OAE, RNAGenScape performs guided random walks in latent space. The discrete update is
9
where 0 is the step size, 1 is a temperature controlling exploration versus exploitation, and 2 is the utility function: to maximize the property, 3; to minimize it, 4 (Liao et al., 14 Oct 2025). The noise term is isotropic Gaussian noise, and the factor 5 matches overdamped Langevin discretization.
The operator 6 is a denoising manifold projector trained to retract off-manifold points back near the manifold. Training uses clean latents 7 formed from original embeddings together with geometry-aware SUGAR samples, followed by a short Gaussian corruption chain
8
and the projector objective
9
In practice 0, and for Zebrafish the corruption scales are 1.
Optimization proceeds by encoding 2, iterating drift computation, noisy update, and projection for 3, and decoding the final latent to 4. Interpolation between source and target latents replaces the property gradient by a normalized pull toward 5:
6
The decoded intermediates are intended to remain close to the viable mRNA manifold throughout the trajectory.
A notable feature of this formulation is that optimization and interpolation are not separated. Both are instances of latent-space dynamics with projection, differing only in the choice of drift term.
6. Training regimes, empirical results, and technical significance
The 2021 formulation trains on unlabeled human-transcriptome snippets of lengths 32–512 nts, with 1.15M train and 20K test sequences, and on labeled RNAcompete-S data of length 40 nts across seven RBP datasets, each containing 500K positives versus 500K negatives with an 80/20 train/test split (Yan et al., 2021). Its reported hyperparameters include latent dimension 7, hidden units 8 in all encoders and decoders, G-MPNN message-passing 9, T-GRU tree passes 0, learning rate 1 with AMSGrad, and batch size 2. KL annealing uses 5 warm-up epochs with 3, followed by a linear increase over the next 6–10 epochs up to 4, or 5 for HierVAE.
On unsupervised generation, posterior decoding uses 20K test RNAs with 5 draws each, and prior decoding uses 10K latent samples with 10 draws each. With structural constraints, HierVAE attains Validity 6, FE DEV 7, Normed FE DEV 8, and Diversity 9; even unconstrained decoding gives Validity 00 and FE DEV 01. By contrast, LSTMVAE and GraphVAE both yield lower validity, around 02, if unconstrained. In the semi-supervised setting, embedding AUROC is approximately 03–04, posterior-constrained sampling is 05 valid, FE DEV is approximately 06, and reconstruction is approximately 07–08. For targeted design, starting from 10K negative RNAs and optimizing for 15–30 steps yields success rates of approximately 09–10 and mean 11 of approximately 12–13.
The 2025 formulation trains on three real mRNA datasets: Zebrafish 5′ UTR with approximately 55k sequences of length 14 and target translation efficiency; OpenVaccine with approximately 2.4k sequences of length 15 and target mRNA stability; and a ribosome-loading dataset with approximately 260k sequences of length 16 and target mean ribosome load (Liao et al., 14 Oct 2025). OAE training uses AdamW with base learning rate 17, warm-up from 18, cosine decay, batch size 19, up to 200 epochs, and early stopping after 20 epochs; projector training uses AdamW with learning rate 20, batch size 21, up to 200 epochs, and early stopping. The Langevin procedure typically uses 22 steps, 23, and 24. SUGAR upsampling ratios are Zebrafish 25, OpenVaccine 26, and Ribosome-loading 27.
Across 5 seeds, the optimization results are reported as follows: for Zebrafish, 28 with 29 success and 30 with 31 success; for OpenVaccine, 32 with 33 success and 34 with 35 success; for Ribosome-load, 36 with 37 success and 38 with 39 success. Inference cost is approximately 40 ms/sample on A100, compared with 41 ms for classic gradient ascent. Manifold fidelity, measured as average 42 to the test set, is approximately 43, compared with about 44 for VAE and about 45 for MCMC. Ablations show that removing 46 causes property changes to collapse to approximately 47, success to about 48, and manifold distance to triple to about 49. Trajectories in PHATE space are smooth and monotonic in predicted property, decoded intermediates fold into valid 2D and 3D structures by ViennaRNA and RhoFold, and interpolation produces a nearly linear decrease in 50 with a corresponding increase in 51.
Taken together, these results define RNAGenScape as a latent-manifold design paradigm rather than a single architecture. The 2021 system emphasizes structural validity, stability, and diversity in RNA secondary-structure generation, while the 2025 system emphasizes controllable mRNA optimization, interpolation, and manifold fidelity under scarce or undersampled data. A plausible implication is that the common scientific contribution lies in coupling biologically structured latent representations with constrained or projected latent traversal so that decoded sequences remain meaningful throughout generation or optimization.