mlsanityData to 14 Oct 2026

Cohort-Anchored Foundation Models for Electronic Health Records: From Risk Scores to Auditable Peer Cohorts

arXiv 2606.21885 ↗First posted 20 Jun 2026cs.LG cs.AIcited by 0

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

  1. 2017Attention Is All You Needcited by 60,361
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  3. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
  4. 2020Language Models are Few-Shot Learnerscited by 32,419
  5. 2023GPT-4 Technical Reportcited by 18,396
  6. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
  7. 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
  8. 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,297
  9. 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
  10. 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,513
  11. 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,968
  12. 2017A Unified Approach to Interpreting Model Predictionscited by 6,555
38 more
  1. 2016"Why Should I Trust You?": Explaining the Predictions of Any Classifiercited by 5,070
  2. 2021On the Opportunities and Risks of Foundation Modelscited by 4,033
  3. 2015Domain-Adversarial Training of Neural Networkscited by 3,044
  4. 2020Unsupervised Learning of Visual Features by Contrasting Cluster Assignmentscited by 2,857
  5. 2023Retrieval-Augmented Generation for Large Language Models: A Surveycited by 2,212
  6. 2022Survey of Hallucination in Natural Language Generationcited by 2,075
  7. 2019BioBERT: a pre-trained biomedical language representation model for biomedical text miningcited by 1,947
  8. 2022Large Language Models Encode Clinical Knowledgecited by 1,830
  9. 2020WILDS: A Benchmark of in-the-Wild Distribution Shiftscited by 1,375
  10. 2020Gradient Surgery for Multi-Task Learningcited by 1,256
  11. 2019Distributionally Robust Neural Networks for Group Shifts: On the Importance of Regularization for Worst-Case Generalizationcited by 1,247
  12. 2018Sanity Checks for Saliency Mapscited by 1,156
  13. 2020Domain-Specific Language Model Pretraining for Biomedical Natural Language Processingcited by 1,009
  14. 2019Publicly Available Clinical BERT Embeddingscited by 1,002
  15. 2018This Looks Like That: Deep Learning for Interpretable Image Recognitioncited by 737
  16. 2023Capabilities of GPT-4 on Medical Challenge Problemscited by 638
  17. 2020The Future of Digital Health with Federated Learningcited by 633
  18. 2020Contrastive Learning with Hard Negative Samplescited by 628
  19. 2022MedMCQA : A Large-scale Multi-Subject Multi-Choice Dataset for Medical domain Question Answeringcited by 543
  20. 2022BioGPT: Generative Pre-trained Transformer for Biomedical Text Generation and Miningcited by 538
  21. 2023BiomedCLIP: a multimodal biomedical foundation model pretrained from fifteen million scientific image-text pairscited by 520
  22. 2020Neural Controlled Differential Equations for Irregular Time Seriescited by 506
  23. 2017Interpretation of Neural Networks is Fragilecited by 444
  24. 2020Hard Negative Mixing for Contrastive Learningcited by 402
  25. 2023Med-Flamingo: a Multimodal Medical Few-shot Learnercited by 392
  26. 2018Confounding variables can degrade generalization performance of radiological deep learning modelscited by 379
  27. 2020Med-BERT: pre-trained contextualized embeddings on large-scale structured electronic health records for disease predictioncited by 373
  28. 2016RETAIN: An Interpretable Predictive Model for Healthcare using Reverse Time Attention Mechanismcited by 349
  29. 2019What Makes Training Multi-Modal Classification Networks Hard?cited by 344
  30. 2024Benchmarking Retrieval-Augmented Generation for Medicinecited by 331
  31. 2023Towards Generalist Biomedical AIcited by 289
  32. 2018Scalable and accurate deep learning for electronic health recordscited by 280
  33. 2022Balanced Multimodal Learning via On-the-fly Gradient Modulationcited by 264
  34. 2019BEHRT: Transformer for Electronic Health Recordscited by 263
  35. 2021Multi-Time Attention Networks for Irregularly Sampled Time Seriescited by 238
  36. 2022GatorTron: A Large Clinical Language Model to Unlock Patient Information from Unstructured Electronic Health Recordscited by 238
  37. 2024Capabilities of Gemini Models in Medicinecited by 237
  38. 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.