Biology

Curriculum-Augmented GFlowNets For mRNA Sequence Generation

How the science connects

Machine learningMessenger RNAProtein synthesis

AI Insight

Researchers developed Curriculum-Augmented GFlowNets (CAGFNs), a machine learning method for designing synthetic mRNA sequences that encode target proteins. The approach uses an adaptive curriculum that trains the model on progressively longer and more complex protein sequences, addressing challenges of sparse rewards and mode collapse that plague traditional methods. Testing on a novel codon-level environment showed CAGFNs converge up to 4 times faster than standard GFlowNets while maintaining complete sequence uniqueness and achieving better coverage of optimal solutions across multiple biological objectives.


This work could accelerate the design of therapeutic mRNAs, such as vaccines and gene therapies, by enabling more efficient optimization of sequences for stability, expression levels, and other critical properties. The curriculum learning approach may also prove valuable for other complex biological sequence design problems.


⚠️ Preprint – Noch nicht peer-reviewed

Dieser Artikel wurde noch nicht von unabhängigen Experten begutachtet. Die Ergebnisse sind vorläufig und sollten mit Vorsicht interpretiert werden.

-cross
Abstract: We develop an approach for the de novo synthesis of messenger RNA (mRNA) sequences encoding target proteins. This requires optimizing multiple biological objectives over large combinatorial spaces defined by synonymous codon sequences, a setting where Reinforcement Learning (RL) methods suffer from mode collapse and Generative Flow Networks (GFlowNets) face sparse rewards over long horizons. To address these challenges, we propose Curriculum-Augmented GFlowNets (CAGFNs), a multi-objective GFlowNet trained with an adaptive length-based curriculum driven by learning progress. In a nutshell, CAGFNs learn from progressively harder tasks corresponding to longer target proteins. We evaluate CAGFNs on CodonDesignEnv, a novel codon-level environment that preserves protein identity via dynamic action masking. Our experiments show that CAGFNs achieve competitive or stronger Pareto-front coverage and diversity while converging up to 4 times faster than non-curriculum GFlowNets, and maintaining 100% sequence uniqueness compared to RL baselines.

Source: Curriculum-Augmented GFlowNets For mRNA Sequence Generation