DeepMiRT: miRNA Target Prediction with RNA Foundation Models
Model Description
DeepMiRT predicts miRNA-target interactions using RNA-FM embeddings and cross-attention. It ranks #1 on eCLIP benchmarks among 12 methods (AUROC 0.75) and achieves AUROC 0.96 on a comprehensive 813K-sample test set.
Architecture
- Encoder: RNA-FM T12 (12-layer Transformer, pre-trained on 23M ncRNAs) β shared for both miRNA and target
- Interaction: Cross-attention (2 layers, 8 heads) β target queries miRNA
- Classifier: MLP (640 β 256 β 64 β 1)
Usage
pip install deepmirt
from deepmirt import predict
probs = predict(
mirna_seqs=["UGAGGUAGUAGGUUGUAUAGUU"],
target_seqs=["ACUGCAGCAUAUCUACUAUUUGCUACUGUAACCAUUGAUCU"],
)
print(f"Interaction probability: {probs[0]:.4f}")
Training
- Data: miRNA-target interactions from multiple databases and literature mining
- Two-phase: Phase 1 (frozen backbone, 27 epochs) β Phase 2 (unfreeze top 3 RNA-FM layers)
- Hardware: 2x NVIDIA L20 GPUs, mixed precision (fp16)
Performance
| Benchmark | AUROC | Rank |
|---|---|---|
| miRBench eCLIP (Klimentova 2022) | 0.7511 | #1/12 |
| miRBench eCLIP (Manakov 2022) | 0.7543 | #1/12 |
| Our test set (813K samples) | 0.9606 | #1/16 |
Files
epoch=27-val_auroc=0.9612.ckptβ Best model checkpoint (495 MB)config.yamlβ Training configuration
Links
- GitHub: zichengll/DeepMiRT
- Demo: Hugging Face Spaces
License
MIT
- Downloads last month
- -