MARRI model checkpoints

Trained checkpoints for MARRI (mechanism-aware multimodal framework for RNA--RNA interaction prediction). All checkpoints share the same architecture (RNA-FM + RNet-2D dual-tower encoder, opening branch, hybridization-gain MLP, axial cross-attention classifier) and the same backbone configuration -- RNA-FM fine-tuned, RNet-2D frozen, epoch-wise dynamic negative resampling -- differing only in random seed and/or training data split.

Code: https://github.com/GainGod-Xu/MARRI Interpretability/attention export for the headline model: https://huggingface.co/datasets/Xu-AI4Science/MARRI-interpretability

Each .pth file is a raw state_dict() for the interaction model (not a full training checkpoint with optimizer state), loadable via model.load_state_dict(torch.load(path, map_location=device)).

Main-matrix checkpoints (N-clean dataset, random 8:1:1 split)

File Seed Test ROC-AUC Best epoch Notes
dyn_ft-rnafm_seed42/best_rna_physics_model.pth 42 0.7530 23 Headline model, reported as the main MARRI result throughout the paper
dyn_ft-rnafm_seed43/best_rna_physics_model.pth 43 0.7409 15 Seed-repeat run, same recipe
dyn_ft-rnafm_seed44/best_rna_physics_model.pth 44 0.7342 18 Seed-repeat run, same recipe

Mean +/- SD over these three seeds: ROC-AUC 0.7427 +/- 0.0095 (Table 1 in the paper).

Leakage-resistant held-out generalization checkpoints (seed 42, own data split each)

Same recipe, each retrained from scratch on its own leakage-resistant held-out split (a separate positive/negative pool per split, not the main N-clean dataset -- see Methods/Supplementary for split construction).

File Held-out split Test ROC-AUC Best epoch
generalization_interaction_pair_seed42/best_rna_physics_model.pth Unseen RNA--RNA pair 0.7422 14
generalization_transcript_seed42/best_rna_physics_model.pth Unseen RNA/gene (most stringent) 0.6630 13
generalization_interaction_locus_seed42/best_rna_physics_model.pth Unseen local interaction locus 0.7502 19

These three correspond to Table 4 ("Generalization under leakage-resistant held-out splits") in the paper.

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support