GOLD: Geometric Optimized Latent Diffusion for Structure-Aware RNA Inverse Folding

arXiv:2610.22992 · cs.AI · Submitted 2026-10-16 · Read on arXiv

cs.AI

Submitted: 2026-10-16

Updated: 2026-10-16

Comments: Accept as a Spotlight Paper in NeurIPS 2026

License: http://creativecommons.org/licenses/by/4.0/

The gist: Designing RNA sequences that fold into specific 3D structures is a fundamental challenge in therapeutics.

Abstract

Designing RNA sequences that fold into specific 3D structures is a fundamental challenge in therapeutics. Current RNA inverse folding methods typically condition on 3D backbone coordinates but the generation is limited to 1D discrete sequences. This ignores the continuous side-chain orientations, which are essential for overall structural stability. To address this issue, we explicitly introduce side-chain geometry into the RNA generative process. Unlike co-design frameworks that diffuse across separate raw spaces, we propose Geometric Optimized Latent Diffusion (GOLD), which project sequence and structure into a unified latent manifold. GOLD employs a novel GeoVAE to elegantly integrate discrete RNA sequences with continuous side-chain angles. By mapping these modalities into a single, smooth latent manifold, GOLD naturally facilitates a synergistic co-diffusion process. To enhance structural validity, we further introduce a training-free, gradient-guided sampling mechanism. By decoding the generated sequence probabilities and side-chain geometries into structures during the sampling process, we compute analytical gradients from physical constraints. This dynamic feedback actively guides the generated sequence to a physically stable state. Extensive experiments validate the superiority of GOLD, achieving state-of-the-art performance across sequence recovery, secondary structure validity, and 3D foldability metrics. Our code and data are publicly available at https://github.com/darkflash03/GOLD.

Related papers