Cite the posted bioRxiv preprint (doi 10.64898/2026.09.15.751749), and sync the card with the canonical copy
Browse files
README.md
CHANGED
|
@@ -47,9 +47,10 @@ fifteen — the one with the lowest validation loss, the value in the table abov
|
|
| 47 |
quantity took part in the choice. For a benchmark comparison, use the paper's means, not one
|
| 48 |
checkpoint.
|
| 49 |
|
| 50 |
-
Quote the validation losses from `analysis/val_metrics.csv` in the repository
|
| 51 |
-
|
| 52 |
-
|
|
|
|
| 53 |
|
| 54 |
Held-out test sets, never used for selection: 48,352 rows (qualitative), 33,490 (ms), 14,150 (both
|
| 55 |
IC50 arms).
|
|
@@ -66,12 +67,9 @@ From the IEDB Export v3, split into arms by measurement type. The rebuild is
|
|
| 66 |
| ic50_500 | `data/dataset/ic50/` | 33,004 | 14,150 | `Target_500` | 36.7 % |
|
| 67 |
| ic50_1000 | `data/dataset/ic50/` | 33,004 | 14,150 | `Target_1000` | 46.0 % |
|
| 68 |
|
| 69 |
-
The qualitative arm's training config names `dataset/full_bal/`, a directory since renamed to
|
| 70 |
-
`full`. Same CSVs.
|
| 71 |
-
|
| 72 |
Training: AdamW, lr 1e-5, betas (0.9, 0.999), eps 1e-8, weight decay 0.01, five-fold CV, best epoch
|
| 73 |
-
by validation loss (20, 27, 28, 28 for the four files above — the epoch each
|
| 74 |
-
corrected argmin of its run). Both sides were fed from precomputed
|
| 75 |
ESMC 300M embeddings, `emb_hla_esmc_small_0430.h5` for the HLA chains and the matching epitope
|
| 76 |
store.
|
| 77 |
|
|
@@ -118,15 +116,15 @@ Two limits worth knowing before feeding it anything:
|
|
| 118 |
|
| 119 |
```bash
|
| 120 |
git clone https://github.com/daylight-00/PREpiBind && cd PREpiBind
|
| 121 |
-
pip install --extra-index-url https://download.pytorch.org/whl/cu126 -e .
|
| 122 |
|
| 123 |
hf download daylight-00/prepibind prepibind_qualitative_s100_f0.pt --local-dir models
|
| 124 |
hf download daylight-00/esmc-300m-2024-12 esmc_300m_2024_12_v0_fp16.pth --local-dir models
|
| 125 |
hf download daylight-00/prepibind-embeddings emb_hla_esmc_small_0430.h5 --repo-type dataset --local-dir emb
|
| 126 |
```
|
| 127 |
|
| 128 |
-
The ESMC encoder is vendored into the package (`prepibind/esmc/`)
|
| 129 |
-
|
| 130 |
|
| 131 |
```python
|
| 132 |
from prepibind.inference import load_config, main
|
|
@@ -155,8 +153,9 @@ The CLI does the same thing for the configs' own settings:
|
|
| 155 |
python -m prepibind.inference configs/predict/config_ms.py --chkp_path models/prepibind_ms_s128_f3.pt --plot
|
| 156 |
```
|
| 157 |
|
| 158 |
-
No GPU is required
|
| 159 |
-
|
|
|
|
| 160 |
|
| 161 |
## Limitations and known caveats
|
| 162 |
|
|
@@ -191,8 +190,10 @@ licence. Training data derives from IEDB (CC BY 4.0) and allele sequences from I
|
|
| 191 |
title = {PREpiBind: Protein Representation-integrated Epitope-MHC Class II Binding Prediction},
|
| 192 |
author = {Jang, David Hyunyoo and Kim, Dongwoo and Park, Byungho and Hwang, Untaek and Choi, Yoonjoo and Lee, Juyong},
|
| 193 |
journal = {bioRxiv},
|
| 194 |
-
year = {2026}
|
|
|
|
|
|
|
| 195 |
}
|
| 196 |
```
|
| 197 |
|
| 198 |
-
|
|
|
|
| 47 |
quantity took part in the choice. For a benchmark comparison, use the paper's means, not one
|
| 48 |
checkpoint.
|
| 49 |
|
| 50 |
+
Quote the validation losses from `analysis/val_metrics.csv` in the repository; the `val_loss`
|
| 51 |
+
column of `analysis/scoring/*/per_run_results.csv` is joined in from that file and carries the same
|
| 52 |
+
numbers. The `val_auc` column in either is not usable: it comes from a `roc_auc_score` call with its
|
| 53 |
+
arguments swapped.
|
| 54 |
|
| 55 |
Held-out test sets, never used for selection: 48,352 rows (qualitative), 33,490 (ms), 14,150 (both
|
| 56 |
IC50 arms).
|
|
|
|
| 67 |
| ic50_500 | `data/dataset/ic50/` | 33,004 | 14,150 | `Target_500` | 36.7 % |
|
| 68 |
| ic50_1000 | `data/dataset/ic50/` | 33,004 | 14,150 | `Target_1000` | 46.0 % |
|
| 69 |
|
|
|
|
|
|
|
|
|
|
| 70 |
Training: AdamW, lr 1e-5, betas (0.9, 0.999), eps 1e-8, weight decay 0.01, five-fold CV, best epoch
|
| 71 |
+
by validation loss (20, 27, 28, 28 for the four files above — the epoch each source checkpoint
|
| 72 |
+
recorded, and the corrected argmin of its run). Both sides were fed from precomputed
|
| 73 |
ESMC 300M embeddings, `emb_hla_esmc_small_0430.h5` for the HLA chains and the matching epitope
|
| 74 |
store.
|
| 75 |
|
|
|
|
| 116 |
|
| 117 |
```bash
|
| 118 |
git clone https://github.com/daylight-00/PREpiBind && cd PREpiBind
|
| 119 |
+
pip install --extra-index-url https://download.pytorch.org/whl/cu126 -e ".[demo]"
|
| 120 |
|
| 121 |
hf download daylight-00/prepibind prepibind_qualitative_s100_f0.pt --local-dir models
|
| 122 |
hf download daylight-00/esmc-300m-2024-12 esmc_300m_2024_12_v0_fp16.pth --local-dir models
|
| 123 |
hf download daylight-00/prepibind-embeddings emb_hla_esmc_small_0430.h5 --repo-type dataset --local-dir emb
|
| 124 |
```
|
| 125 |
|
| 126 |
+
The ESMC encoder is vendored into the package (`prepibind/esmc/`) and runs on `torch` alone, so
|
| 127 |
+
inference needs no `pip install esm`.
|
| 128 |
|
| 129 |
```python
|
| 130 |
from prepibind.inference import load_config, main
|
|
|
|
| 153 |
python -m prepibind.inference configs/predict/config_ms.py --chkp_path models/prepibind_ms_s128_f3.pt --plot
|
| 154 |
```
|
| 155 |
|
| 156 |
+
No GPU is required, and the runtime says whatever it had to drop: without CUDA, float32 throughout
|
| 157 |
+
and no flash-attn; on a pre-Ampere card, flash-attn off and the bfloat16 ESMC demoted to float16,
|
| 158 |
+
head unchanged.
|
| 159 |
|
| 160 |
## Limitations and known caveats
|
| 161 |
|
|
|
|
| 190 |
title = {PREpiBind: Protein Representation-integrated Epitope-MHC Class II Binding Prediction},
|
| 191 |
author = {Jang, David Hyunyoo and Kim, Dongwoo and Park, Byungho and Hwang, Untaek and Choi, Yoonjoo and Lee, Juyong},
|
| 192 |
journal = {bioRxiv},
|
| 193 |
+
year = {2026},
|
| 194 |
+
doi = {10.64898/2026.09.15.751749},
|
| 195 |
+
url = {https://www.biorxiv.org/content/10.64898/2026.09.15.751749v1}
|
| 196 |
}
|
| 197 |
```
|
| 198 |
|
| 199 |
+
Posted on bioRxiv 2026-09-21 as [doi:10.64898/2026.09.15.751749](https://doi.org/10.64898/2026.09.15.751749), CC BY 4.0.
|