CoolFace
Modelpublic

raghvendramall/pearl-benchmark-models

sourceHugging Faceupdated 27d agoView on Hugging Face
0likes
Model Card

PEARL Benchmark -- Top-5 Models per Dataset

Parameter-Efficient Adaptation with Retrieval-Augmented Learning for Molecular Property Prediction. Each dataset folder below holds its top-5 best-performing methods, ranked by the dataset's primary metric.

Code: https://github.com/raghvendra5688/PEARL

BACE (classification, ranked by MCC)

RankMethodMCCConfigurationPath
1E2E LoRA -- Uni-Mol0.6234Weighted LossBACE/rank1_e2e-lora-uni-mol
2E2E LoRA -- HF0.5660ibm_MoLFormerXLboth_10pct, flBACE/rank2_e2e-lora-hf
3FT Embed0.5570MolFormerFinetunedFLBACE/rank3_ft-embed
4RAFE0.5570MolFormerFinetunedFLBACE/rank4_rafe
5FT Embed+PC0.5500MolformerFinetunedWL_embeddingsBACE/rank5_ft-embed-pc

BBBP (classification, ranked by MCC)

RankMethodMCCConfigurationPath
1RAFE0.7980MolFormerFinetunedFLBBBP/rank1_rafe
2FT Embed+PC0.7890MolFormerFinetunedFL_embeddingsBBBP/rank2_ft-embed-pc
3FT Embed0.7870MolFormerFinetunedFLBBBP/rank3_ft-embed
4E2E LoRA -- Uni-Mol0.7740Weighted LossBBBP/rank4_e2e-lora-uni-mol
5E2E LoRA -- HF0.7670ibm_MoLFormerXLboth_10pct, flBBBP/rank5_e2e-lora-hf

FLAVOR (classification, ranked by MCC)

RankMethodMCCConfigurationPath
1Chemprop0.8090D-MPNNFLAVOR/rank1_chemprop
2PC-only0.8080XGBoostFLAVOR/rank2_pc-only
3E2E LoRA -- HF0.8020DeepChem_ChemBERTa77M_MTR, wlFLAVOR/rank3_e2e-lora-hf
4FT Embed+PC0.8010MolformerFinetunedWL_embeddingsFLAVOR/rank4_ft-embed-pc
5RAFE0.8000ChemBERTa77MMTR_FLFLAVOR/rank5_rafe

HERG (classification, ranked by MCC)

RankMethodMCCConfigurationPath
1E2E LoRA -- Uni-Mol0.6093Weighted LossHERG/rank1_e2e-lora-uni-mol
2PC-only0.6040XGBoostHERG/rank2_pc-only
3FT Embed+PC0.6015UniMol_FLHERG/rank3_ft-embed-pc
4RAFE0.5889UniMol_FLHERG/rank4_rafe
5FT Embed0.5798UniMol_FLHERG/rank5_ft-embed

DILI (classification, ranked by MCC)

RankMethodMCCConfigurationPath
1RAFE0.7897UniMol_FLDILI/rank1_rafe
2E2E LoRA -- HF0.7704DeepChem_ChemBERTa77M_MTR, wlDILI/rank2_e2e-lora-hf
3PC-only0.7650CatBoostDILI/rank3_pc-only
4GCN0.7280GINConvDILI/rank4_gcn
5E2E LoRA -- Uni-Mol0.7079Focal LossDILI/rank5_e2e-lora-uni-mol

CACO2 (regression, ranked by Spearman)

RankMethodSpearmanConfigurationPath
1PC-only0.8534CatBoostCACO2/rank1_pc-only
2RAFE0.8347ChemBERTa77MMTR_HuberCACO2/rank2_rafe
3FT Embed+PC0.8271ChemBERTa77MMTR_HuberCACO2/rank3_ft-embed-pc
4FT Embed0.8211ChemBERTa77MMTR_HuberCACO2/rank4_ft-embed
5Chemprop0.8022D-MPNNCACO2/rank5_chemprop

HALF_LIFE (regression, ranked by Spearman)

RankMethodSpearmanConfigurationPath
1FT Embed0.6226MolFormerFinetunedHuberHALF_LIFE/rank1_ft-embed
2FT Embed+PC0.5927MolFormerFinetunedHuberHALF_LIFE/rank2_ft-embed-pc
3RAFE0.5684MolFormerFinetunedHuberHALF_LIFE/rank3_rafe
4E2E LoRA -- HF0.5406DeepChem_ChemBERTa77M_MTR, huberHALF_LIFE/rank4_e2e-lora-hf
5PC-only0.5211CatBoostHALF_LIFE/rank5_pc-only