yk0/tuna-r-esm_mlp

ESM-MLP Protein-Protein Interaction Predictor

Model Description

ESM-MLP is a sequence-based protein–protein interaction (PPI) predictor that uses a MLP backbone. It produces a probability score for whether two proteins interact. This is an ablation of TUnA.

Intended Use

• Research use for PPI prediction and representation learning.
• These weights can be loaded in from the TUnA-R repository on Github.

Training Data

Trained on the Bernett dataset (https://academic.oup.com/bib/article/25/2/bbae076/7621029) composed of Human PPI interactions.

Model Architecture

Backbone: tuna.models._mlp.MLP

• protein_dim: 640
• hid_dim: 64
• dropout: 0.1
• llgp: False
• use_spectral_norm: False
• out_targets: 1

Evaluation Results (Test Set)

• AUROC: 0.69
• MCC: 0.27
• Accuracy: 0.64
• AUPRC: 0.68
• Precision: 0.63

Citation

If you use this model, please cite:

• TUnA: https://academic.oup.com/bib/article/25/5/bbae359/7720609
• Bernett dataset: https://academic.oup.com/bib/article/25/2/bbae076/7621029

License

MIT