From Retrieval to Typed Decisions: Calibrated System One Models from Biomedical Sentence Encoders
Organizations: Queen’s University Belfast Belfast, United Kingdom
Abstract
Typed decision models answer schema-constrained questions about a text in one forward pass and return probabilities meant to be thresholded. We ask whether biomedical sentence encoders trained for retrieval are good starting points for such models. We present SBERT2S1, which converts Sentence-Transformers encoders into bi-encoder, cross-head (C) and prior-fused residual (PFR) decision models, together with BIODECIDE, a biomedical typed-decision suite, and MEDLINE-S1, 243k training decisions derived from NLM indexing. Across six parent-retriever pairs, retrieval training improves zero-shot matching of content-bearing options. After fine-tuning, its effect depends on the head: across five pairs and three training-set sizes, retrieval training significantly helps PFR, which keeps the retrieval prior, in 10 of 15 comparisons, but helps C in one and hurts it in five. A matched grid of two heads and five training objectives shows that C outperforms PFR under every objective, and that the released RLCD recipe of open System One models trails cross-entropy by 2.5-3.0 points. The deficit stems mainly from its reward normalisation, which inflates the noisy score-function term 3.6-15-fold; an unbiased leave-one-out estimator recovers most of the gap. After temperature scaling, no objective is clearly better calibrated than cross-entropy. We release the code, the MEDLINE-S1 labels and a model.
Figures & tables
| Task | Family | Type | #Train | #Test |
|---|---|---|---|---|
| Grounded | ||||
| PubMedQA-L | Evidence QA | choice | 400 | 500 |
| PubMedQA-A § | Evidence QA | noul | 20,000 | – |
| SciFact | Claim verif. | choice | 780 | 321 |
| HealthVer | Claim verif. | choice | 10,073 | 1,823 |
| PUBHEALTH † | Claim verif. | choice | – | 1,231 |
| Seen tasks, acc | Held-out, acc | Clinical | ECE | |||||
| Encoder | Type | C | PFR | C, 10% | C | PFR | C | C |
| PubMedBERT | MLM | 67.3 0.7 | 66.9 0.8 | 59.1 0.6 | 29.5 3.3 | 20.7 0.6 | 20.7 0.8 | 4.8 0.9 |
| S-PubMedBERT-MS-MARCO | contr. | 67.7 0.9 | 65.3 1.5 | 61.0 1.4 | 32.8 1.5 | 23.6 1.5 | 20.5 1.8 | 3.9 0.7 |
| MedCPT-Query | contr. | 17.9 4.3 | 60.4 0.7 | 15.0 10.2 | 3.9 0.5 | 23.6 3.2 | 12.5 3.4 | 14.0 2.3 |
| MPNet | MLM | 66.4 0.9 | 63.3 0.4 | 61.5 0.5 | 30.2 2.7 | 20.8 2.5 | 22.7 5.2 | 4.8 0.4 |
| all-MPNet-base-v2 | contr. | 66.4 0.9 | 67.8 0.8 | 60.5 2.5 | 27.5 0.8 | 24.7 2.3 | 13.7 5.0 | 5.2 0.3 |
| Z = PFR at step 0 | C at step 0 | |||
|---|---|---|---|---|
| Encoder | content | label | content | label |
| PubMedBERT | 1.0 | 13.0 | 1.1 | 24.0 |
| S-PubMedBERT-MS-MARCO | 15.2 | 20.4 | 2.3 | 2.8 |
| MedCPT-Query | 19.8 | 20.2 | 3.5 | 17.6 |
| MPNet | 1.0 | 3.6 | 1.2 | 12.5 |
| all-MPNet-base-v2 | 21.2 | 1.2 | 1.2 | 12.1 |
| Seen acc | Held-out acc | ECE | ||||
|---|---|---|---|---|---|---|
| Objective | PFR | C | PFR | C | PFR | C |
| CE | 68.3 | 70.2 | 20.8 | 31.0 | 4.0 | 3.8 |
| Proper | 68.5 | 70.7 | 22.6 | 28.8 | 4.4 | 3.9 |
| PG (released RLCD) | 65.3 | 67.7 | 23.6 | 32.8 | 3.8 | 3.9 |
| PG w/o CE | 65.5 | – | 20.3 | – | 4.0 | – |
| PG-LOO (unbiased) | 67.8 | 69.2 | 23.8 | 30.0 | 3.9 | 3.7 |
Appendix figures & tables13 assets
Supplementary material from the paper’s appendix.
Appendix
| Task | State | Question(s) and type | Options and target | Splits |
|---|---|---|---|---|
| PubMedQA-L | abstract contexts without the conclusion | the original question ( choice ) | yes / no / maybe with short descriptions; one-hot | 1,000 labelled items: 500 test, 50 dev, 50 calibration, rest train (seeded) |
| PubMedQA-A | abstract contexts | the original question ( noul ) | yes true; one-hot | 20k sampled for training, excluding every evaluation PMID |
| SciFact | title and abstract of the evidence (or first cited) document | “Does this abstract support or refute the claim: c ” ( choice ) | supports / refutes / not_enough_info | official train; validation as test; calibration grouped by claim |
| HealthVer | evidence passage | “Does this evidence support or refute the claim: c ” ( choice ) | as SciFact | official train / dev / test; calibration grouped by topic |
| PUBHEALTH | article text (fact-checkers’ explanation removed) | “Based on this article, what is the veracity of the health claim: c ” ( choice ) | false / mixture / true / unproven | official test only |
| DDI-2013 | sentence with the two drugs marked | “What kind of interaction between @DRUG1 does the sentence describe?” ( choice ) | mechanism / effect / advice / int / none | official dev as test (test is gated); test sentences seen in training removed |
| Slope vs. | Score : CE weight | Cosine | SNR | |||||
|---|---|---|---|---|---|---|---|---|
| Checkpoint (head, objective) | Estimator | |||||||
| PFR, PG | released PG | 3.59 | 5.72 | 14.27 | 4.74 | 18.87 | 0.982 | 0.861 |
| mean baseline, unnormalised | 0.75 | 0.75 | 0.75 | 0.99 | 1.00 | 0.982 | 0.816 | |
| PG-LOO | 1.01 | 1.00 | 1.00 | 1.33 | 1.33 | 0.983 | 0.791 | |
| pathwise (Reparam) | 1.00 | 1.00 | 1.00 | 1.32 | 1.32 | 1.000 | 206 | |
| PFR, PG-LOO | released PG | 3.66 | 5.84 | 14.57 | 4.83 | 19.23 | 0.982 | 0.854 |
| Objective | Head | Seen | Held-out | Clinical | Knowledge | ECE | ECE | NLL | MAE | Perm. flip | |
|---|---|---|---|---|---|---|---|---|---|---|---|
| CE | PFR | 68.3 0.4 | 20.8 1.6 | 21.9 2.9 | 13.1 0.8 | 9.3 0.9 | 4.0 0.4 | 0.510 0.004 | 0.588 0.012 | 0.4 0.1 | 7.1 |
| C | 70.2 0.4 | 31.0 1.7 | 20.0 4.1 | 13.6 0.9 | 7.4 0.3 | 3.8 0.2 | 0.491 0.001 | 0.639 0.020 | 2.9 0.4 | 6.0 | |
| Proper | PFR | 68.5 0.2 | 22.6 1.9 | 23.4 0.7 | 13.6 0.4 | 9.1 0.6 | 4.4 0.4 | 0.510 0.006 | 0.582 0.004 | 0.2 0.1 | 8.7 |
| C | 70.7 0.3 | 28.8 0.4 | 22.6 1.1 | 12.8 0.7 | 7.6 0.5 | 3.9 0.3 | 0.493 0.013 | 0.627 0.007 | 2.6 0.2 | 7.7 | |
| PG (released RLCD) | PFR | 65.3 1.5 | 23.6 1.5 | 18.3 8.2 | 11.4 1.2 | 7.0 0.6 | 3.8 0.2 | 0.548 0.018 | 0.638 0.016 | 0.4 0.1 | 88.1 |
| C | 67.7 0.9 | 32.8 1.5 | 20.5 1.8 | 11.2 0.7 | 6.5 0.0 | 3.9 0.7 | 0.527 0.010 | 0.746 0.031 | 3.2 0.5 | 50.2 |
| Contrast ( ) | Seeds | Seen [95% CI] | Held-out [95% CI] | ||
| C full: S-PubMedBERT PubMedBERT | 3/3 | +0.4 [ 0.4, +1.1] | 1 | +3.2 [+0.3, +6.1] | .956 |
| C full: all-MPNet MPNet | 3/3 | +0.0 [ 0.8, +0.9] | 1 | 2.6 [ 5.6, +0.3] | 1 |
| C full: Nomic-embed ModernBERT | 3/3 | 1.0 [ 1.9, 0.1] | .708 | +2.5 [ 1.2, +6.2] | 1 |
| C full: GTE-ModernBERT ModernBERT | 3/3 | 2.5 [ 3.5, 1.5] | .012 | 2.7 [ 5.9, +0.6] | 1 |
| C full: BioClinical-MB-emb BioClinical-MB | 3/3 | 0.2 [ 1.0, +0.5] | 1 | +0.3 [ 3.2, +3.7] | 1 |
| PFR full: S-PubMedBERT PubMedBERT | 3/3 | 1.6 [ 2.5, 0.8] | .014 | +2.9 [ 0.1, +6.1] | 1 |
| Contrast ( ) | Seeds | Seen [95% CI] | Held-out [95% CI] | ||
|---|---|---|---|---|---|
| C PFR (CE) | 3/3 | +1.9 [+0.8, +3.1] | .011 | +10.2 [+5.8, +14.6] | .003 |
| C PFR (Proper) | 3/3 | +2.2 [+1.1, +3.4] | .004 | +6.2 [+2.0, +10.5] | .041 |
| C PFR (PG) | 3/3 | +2.4 [+1.2, +3.6] | .003 | +9.2 [+5.0, +13.4] | .003 |
| C PFR (LOO) | 3/3 | +1.5 [+0.3, +2.6] | .079 | +6.2 [+2.0, +10.3] | .041 |
| C PFR (Reparam) | 3/3 | +1.3 [+0.1, +2.4] | .136 | +3.8 [ 0.6, +8.2] | .247 |
| B C (PG) | 3/3 | 0.6 [ 1.8, +0.7] | .353 | 9.3 [ 13.5, 5.0] | .003 |
| Encoder | Arch | Seen acc | ECE | ECE | Held-out | Clinical | Knowledge |
|---|---|---|---|---|---|---|---|
| PubMedBERT | C | 67.3 0.7 | 6.7 0.6 | 4.8 0.9 | 29.5 3.3 | 20.7 0.8 | 13.4 0.7 |
| PFR | 66.9 0.8 | 8.3 0.4 | 4.1 0.3 | 20.7 0.6 | 23.9 0.6 | 12.5 0.9 | |
| B | 67.0 0.5 | 8.3 0.4 | 4.1 0.1 | 17.8 6.5 | 23.6 1.4 | 11.5 0.7 | |
| S-PubMedBERT-MS-MARCO | C | 67.7 0.9 | 6.5 0.0 | 3.9 0.7 | 32.8 1.5 | 20.5 1.8 | 11.2 0.7 |
| PFR | 65.3 1.5 | 7.0 0.6 | 3.8 0.2 | 23.6 1.5 | 18.3 8.2 | 11.4 1.2 | |
| B | 67.1 0.5 | 7.8 0.6 | 3.9 0.2 | 23.4 1.4 | 25.9 2.1 | 11.8 0.7 |
| Seen (grounded training tasks) | Held-out | ||||||||||
| Model | PubMedQA | SciFact | HealthVer | DDI | HoC | ADE | Druglib | MEDLINE | PUBHEALTH | BIOSSES | MTSamples |
| PubMedBERT C (PG) | 51.6 | 51.2 | 48.8 | 89.3 | 88.1 | 87.3 | 35.6 | 86.5 | 23.1 | 0.4 | 65.2 |
| S-PubMedBERT C (PG) | 47.6 | 55.3 | 49.1 | 91.0 | 88.7 | 88.0 | 35.4 | 86.4 | 24.2 | 5.4 | 68.7 |
| S-PubMedBERT C (CE) | 47.7 | 60.6 | 55.0 | 92.4 | 89.1 | 88.1 | 41.6 | 87.0 | 20.5 | 1.7 | 70.9 |
| S-PubMedBERT PFR (PG) | 44.9 | 34.9 | 52.9 | 91.5 | 85.8 | 86.9 | 42.8 | 82.7 | 17.0 | 2.5 | 51.3 |
| S-PubMedBERT PFR (CE) | 46.9 | 39.9 | 61.9 | 93.3 | 87.3 | 87.3 | 45.7 | 83.8 | 8.4 | 2.5 | 51.5 |
| Perm. | Trial Qs: answer (acc ) | Trial Qs: answerable (acc ) | Name gate AUROC | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|
| Model | flip % | trunc | win-mean | win-conf | RtD | trunc | win-mean | win-conf | RtD | test | common-word |
| PubMedBERT C | 16.6 | 2.4 | 3.4 | 2.3 | 5.9 | 4.5 | 3.7 | 2.8 | 7.9 | 72.9 | 71.8 |
| S-PubMedBERT C | 15.7 | 0.0 | 3.8 | 0.1 | 1.1 | 7.1 | 1.0 | 4.0 | 20.2 | 72.6 | 71.4 |
| S-PubMedBERT PFR | 0.3 | 8.2 | 15.0 | 16.0 | 7.1 | 4.7 | 0.6 | 1.5 | 5.8 | 74.8 | 74.5 |
| S-PubMedBERT PFR, frozen | 13.1 | 0.5 | 0.2 | 2.9 | 0.4 | 7.8 | 0.4 | 2.3 | 29.9 | 75.0 | 74.1 |
| S-PubMedBERT B | 0.0 | 10.9 | 19.7 | 20.5 | 8.3 | 9.8 | 4.2 | 4.7 | 3.8 | 77.6 | 77.7 |
| Model | Historical | Filtered | Difference |
|---|---|---|---|
| S-PubMedBERT C | 67.69 | 67.32 | 0.37 |
| S-PubMedBERT PFR (PG) | 65.30 | 64.91 | 0.40 |
| S-PubMedBERT PFR (CE) | 68.26 | 67.89 | 0.37 |
| S-PubMedBERT PFR (Reparam) | 68.53 | 68.19 | 0.35 |
| BioClinical-ModernBERT C | 73.10 | 72.81 | 0.29 |
| Laya-large FT | 74.99 | 74.69 | 0.30 |
| nDCG@10 ( ) | Latency | ||
|---|---|---|---|
| Model (S-PubMedBERT) | SciFact | NFCorpus | ms |
| Z (no training) | 67.6 | 31.3 | 10.2 |
| B | 34.3 | 11.0 | 9.7 |
| C | 43.3 | 13.0 | 6.6 |
| C, trained with CE | 24.6 | 10.3 | 6.6 |
| PFR, shared prior | 35.6 | 11.2 | 15.2 |
Text Classification
Text classification assigns text to categories such as topic, sentiment, intent, stance, or risk, supporting search, moderation, and decisions.
See all 223 papers