Cohort-Anchored Foundation Models for Electronic Health Records: From Risk Scores to Auditable Peer Cohorts
Not on the map. Why?
Abstract (arXiv)
Foundation models have achieved remarkable performance across medical question answering, imaging, and electronic health record (EHR) tasks, yet reliable clinical deployment remains challenging due to limited interpretability, vulnerability to distribution shift, and weak alignment with clinician reasoning. We argue that these limitations arise because existing approaches prioritize representation learning while treating patient comparison as an emergent property rather than a primary source of clinical evidence. To address this gap, we propose CAFM, a Cohort-Anchored Foundation Model framework that elevates patient cohorts to a first-class object throughout the learning pipeline. The framework consists of four stages: deviation-aware data curation, cohort-conditioned pretraining, multimodal cohort alignment, and clinician-in-the-loop refinement. Together, these stages improve data quality, organize representations around clinically meaningful cohort structure, preserve modality-specific relationships, and support auditable clinical decision-making. The framework is compositional and can augment existing EHR foundation models without modifying their underlying encoders. We illustrate CAFM through four clinical case studies spanning acute kidney injury prediction, cardiovascular risk stratification from electrocardiograms, optic neuropathy triage from orbital imaging, and electroretinogram-grounded report generation. We further present five empirically testable hypotheses and identify open challenges in data quality, irregular temporality, multimodal learning, distribution shift, and evaluation beyond predictive accuracy. We argue that explicitly anchoring foundation models to patient cohorts provides a principled path toward trustworthy clinical AI.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 83 of its 150 references that are in the map. First posted on arXiv 20 Jun 2026.
Builds on
83 of 150 references.
References · 83
- 2017Attention Is All You Needcited by 60,361
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2023GPT-4 Technical Reportcited by 18,396
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,297
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,513
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,968
- 2017A Unified Approach to Interpreting Model Predictionscited by 6,555
38 more
- 2016"Why Should I Trust You?": Explaining the Predictions of Any Classifiercited by 5,070
- 2021On the Opportunities and Risks of Foundation Modelscited by 4,033
- 2015Domain-Adversarial Training of Neural Networkscited by 3,044
- 2020Unsupervised Learning of Visual Features by Contrasting Cluster Assignmentscited by 2,857
- 2023Retrieval-Augmented Generation for Large Language Models: A Surveycited by 2,212
- 2022Survey of Hallucination in Natural Language Generationcited by 2,075
- 2019BioBERT: a pre-trained biomedical language representation model for biomedical text miningcited by 1,947
- 2022Large Language Models Encode Clinical Knowledgecited by 1,830
- 2020WILDS: A Benchmark of in-the-Wild Distribution Shiftscited by 1,375
- 2020Gradient Surgery for Multi-Task Learningcited by 1,256
- 2019Distributionally Robust Neural Networks for Group Shifts: On the Importance of Regularization for Worst-Case Generalizationcited by 1,247
- 2018Sanity Checks for Saliency Mapscited by 1,156
- 2020Domain-Specific Language Model Pretraining for Biomedical Natural Language Processingcited by 1,009
- 2019Publicly Available Clinical BERT Embeddingscited by 1,002
- 2018This Looks Like That: Deep Learning for Interpretable Image Recognitioncited by 737
- 2023Capabilities of GPT-4 on Medical Challenge Problemscited by 638
- 2020The Future of Digital Health with Federated Learningcited by 633
- 2020Contrastive Learning with Hard Negative Samplescited by 628
- 2022MedMCQA : A Large-scale Multi-Subject Multi-Choice Dataset for Medical domain Question Answeringcited by 543
- 2022BioGPT: Generative Pre-trained Transformer for Biomedical Text Generation and Miningcited by 538
- 2023BiomedCLIP: a multimodal biomedical foundation model pretrained from fifteen million scientific image-text pairscited by 520
- 2020Neural Controlled Differential Equations for Irregular Time Seriescited by 506
- 2017Interpretation of Neural Networks is Fragilecited by 444
- 2020Hard Negative Mixing for Contrastive Learningcited by 402
- 2023Med-Flamingo: a Multimodal Medical Few-shot Learnercited by 392
- 2018Confounding variables can degrade generalization performance of radiological deep learning modelscited by 379
- 2020Med-BERT: pre-trained contextualized embeddings on large-scale structured electronic health records for disease predictioncited by 373
- 2016RETAIN: An Interpretable Predictive Model for Healthcare using Reverse Time Attention Mechanismcited by 349
- 2019What Makes Training Multi-Modal Classification Networks Hard?cited by 344
- 2024Benchmarking Retrieval-Augmented Generation for Medicinecited by 331
- 2023Towards Generalist Biomedical AIcited by 289
- 2018Scalable and accurate deep learning for electronic health recordscited by 280
- 2022Balanced Multimodal Learning via On-the-fly Gradient Modulationcited by 264
- 2019BEHRT: Transformer for Electronic Health Recordscited by 263
- 2021Multi-Time Attention Networks for Irregularly Sampled Time Seriescited by 238
- 2022GatorTron: A Large Clinical Language Model to Unlock Patient Information from Unstructured Electronic Health Recordscited by 238
- 2024Capabilities of Gemini Models in Medicinecited by 237
- 2015Doctor AI: Predicting Clinical Events via Recurrent Neural Networkscited by 208
50 most cited of 83 All 83 (JSON)
Built on by
Not cited yet.