English

RiboSphere: Learning Unified and Efficient Representations of RNA Structures

Machine Learning 2026-03-23 v1

Abstract

Accurate RNA structure modeling remains difficult because RNA backbones are highly flexible, non-canonical interactions are prevalent, and experimentally determined 3D structures are comparatively scarce. We introduce \emph{RiboSphere}, a framework that learns \emph{discrete} geometric representations of RNA by combining vector quantization with flow matching. Our design is motivated by the modular organization of RNA architecture: complex folds are composed from recurring structural motifs. RiboSphere uses a geometric transformer encoder to produce SE(3)-invariant (rotation/translation-invariant) features, which are discretized with finite scalar quantization (FSQ) into a finite vocabulary of latent codes. Conditioned on these discrete codes, a flow-matching decoder reconstructs atomic coordinates, enabling high-fidelity structure generation. We find that the learned code indices are enriched for specific RNA motifs, suggesting that the model captures motif-level compositional structure rather than acting as a purely compressive bottleneck. Across benchmarks, RiboSphere achieves strong performance in structure reconstruction (RMSD 1.25\,\AA, TM-score 0.84), and its pretrained discrete representations transfer effectively to inverse folding and RNA--ligand binding prediction, with robust generalization in data-scarce regimes.

Keywords

Cite

@article{arxiv.2603.19636,
  title  = {RiboSphere: Learning Unified and Efficient Representations of RNA Structures},
  author = {Zhou Zhang and Hanqun Cao and Cheng Tan and Fang Wu and Pheng Ann Heng and Tianfan Fu},
  journal= {arXiv preprint arXiv:2603.19636},
  year   = {2026}
}
R2 v1 2026-07-01T11:29:18.527Z