From Retrieval to Typed Decisions: Calibrated System One Models from Biomedical Sentence Encoders
cs.CL, cs.AI
Submitted: 2026-10-01
Updated: 2026-10-01
Code: https://github.com/pritamdeka/sbert2s1
License: http://creativecommons.org/licenses/by/4.0/
The gist: Typed decision models answer schema-constrained questions about a text in one forward pass and return probabilities meant to be thresholded.
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.
Sources
- FrugalGPT: How to Use Large Language Models While Reducing Cost and Improving Performance
- Calibrated Decision Models for Autonomous Penetration-Testing Harnesses: JEV and Laya as System One Decision Layers for LLM-Driven Pentest Agents
- Gemma 4 Technical Report
- Evaluating Decision Models for Text Annotation in Computational Social Science
- Towards General Text Embeddings with Multi-stage Contrastive Learning
- Passage Re-ranking with BERT
- Nomic Embed: Training a Reproducible Long Context Text Embedder
- Calibrated Decisions at Scale: Converting Police Crash Narratives into Probabilistic Crash Variables with a System One Model (Jev)
- DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models
- BioClinical ModernBERT: A State-of-the-Art Long-Context Encoder for Biomedical and Clinical NLP
- JevOut: Natural Context Can Flip Decision Models
- LAVOIR: Teaching a Single-Pass Decision Encoder When and What to Ask with Amortized Value of Information
Related papers
- Exploring Solution Divergence and Its Effect on Large Language Model Problem Solving
- Ishigaki-IDS-Bench: A Benchmark for Generating Information Delivery Specification from BIM Information Requirements
- Subliminal Steering: Stronger Encoding of Hidden Signals
- MedStruct-S: A Benchmark for Key Discovery, Key-Conditioned QA and Semi-Structured Extraction from OCR Clinical Reports
- The End of Transformers? On Challenging Attention and the Rise of Sub-Quadratic Architectures
- Untangling the Mechanisms of Misleading Context in Medical Question Answering