RNAFlow: RNA Structure & Sequence Co-Design via Inverse Folding-Based Flow Matching

Published: 04 Mar 2024, Last Modified: 29 Apr 2024GEM PosterEveryoneRevisionsBibTeXCC BY 4.0
Track: Machine learning: computational method and/or computational results
Cell: I do not want my work to be considered for Cell Systems
Keywords: RNA design, inverse folding, flow matching
TL;DR: We propose RNAFlow, a flow matching model for protein-conditioned RNA sequence-structure co-design, where the denoising network integrates an inverse folding model and fixed folding network.
Abstract: The growing significance of RNA engineering in diverse biological applications has spurred interest in developing AI methods for structure-based RNA design. While diffusion models have excelled in protein design, adapting them for RNA presents new challenges due to RNA's conformational flexibility and the computational cost of fine-tuning large structure prediction models. To this end, we propose RNAFlow, a flow matching model for protein-conditioned RNA sequence-structure co-design. Its denoising network integrates an RNA inverse folding model and a pre-trained RosettaFold2NA network for simultaneous generation of RNA sequences and structures. The integration of inverse folding in the structure denoising process allows us to simplify training by fixing the structure prediction network. We further enhance the inverse folding model by conditioning it on inferred conformational ensembles to model dynamic RNA conformations. Evaluation on protein-conditioned RNA structure and sequence generation tasks demonstrates RNAFlow's advantage over existing RNA design methods.
Submission Number: 48
Loading