Discriminative Representation Learning for Clinical Prediction
Not on the map. Why?
Abstract (arXiv)
Foundation models in healthcare have largely adopted self supervised pretraining objectives inherited from natural language processing and computer vision, emphasizing reconstruction and large scale representation learning prior to downstream adaptation. We revisit this paradigm in outcome centric clinical prediction settings and argue that, when high quality supervision is available, direct outcome alignment may provide a stronger inductive bias than generative pretraining. We propose a supervised deep learning framework that explicitly shapes representation geometry by maximizing inter class separation relative to within class variance, thereby concentrating model capacity along clinically meaningful axes. Across multiple longitudinal electronic health record tasks, including mortality and readmission prediction, our approach consistently outperforms masked, autoregressive, and contrastive pretraining baselines under matched model capacity. The proposed method improves discrimination, calibration, and sample efficiency, while simplifying the training pipeline to a single stage optimization. These findings suggest that in low entropy, outcome driven healthcare domains, supervision can act as the statistically optimal driver of representation learning, challenging the assumption that large scale self supervised pretraining is a prerequisite for strong clinical performance.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 79 of its 118 references that are in the map. First posted on arXiv 21 Mar 2026.
Builds on
79 of 118 references.
References · 79
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,082
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2023Segment Anythingcited by 8,158
- 2023Visual Instruction Tuningcited by 7,792
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,764
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,339
38 more
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,893
- 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,322
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,551
- 2019Towards Robust Monocular Depth Estimation: Mixing Datasets for Zero-shot Cross-dataset Transfercited by 1,484
- 20194D Spatio-Temporal ConvNets: Minkowski Convolutional Neural Networkscited by 1,201
- 2021iBOT: Image BERT Pre-Training with Online Tokenizercited by 891
- 2022Swin UNETR: Swin Transformers for Semantic Segmentation of Brain Tumors in MRI Imagescited by 823
- 2021CrossViT: Cross-Attention Multi-Scale Vision Transformer for Image Classificationcited by 806
- 2019Contrastive Representation Distillationcited by 734
- 2018CCNet: Criss-Cross Attention for Semantic Segmentationcited by 717
- 2023Time-LLM: Time Series Forecasting by Reprogramming Large Language Modelscited by 680
- 2022TotalSegmentator: robust segmentation of 104 anatomical structures in CT imagescited by 597
- 2018Bag of Tricks for Image Classification with Convolutional Neural Networkscited by 430
- 2020Med-BERT: pre-trained contextualized embeddings on large-scale structured electronic health records for disease predictioncited by 368
- 2024SegMamba: Long-range Sequential Modeling Mamba For 3D Medical Image Segmentationcited by 266
- 2022Self-Supervised Contrastive Pre-Training For Time Series via Time-Frequency Consistencycited by 246
- 2024Foundation Models for Time Series Analysis: A Tutorial and Surveycited by 234
- 2019Cross Attention Network for Few-shot Classificationcited by 218
- 2023LLM4TS: Aligning Pre-Trained LLMs as Data-Efficient Time-Series Forecasterscited by 127
- 2022OpenXAI: Towards a Transparent Evaluation of Model Explanationscited by 125
- 2022UNETR++: Delving into Efficient and Accurate 3D Medical Image Segmentationcited by 112
- 2023Large-scale Training of Foundation Models for Wearable Biosignalscited by 89
- 2020Language Models Are An Effective Patient Representation Learning Technique For Electronic Health Record Datacited by 85
- 2024Foundation Model for Advancing Healthcare: Challenges, Opportunities, and Future Directionscited by 77
- 2021CEHR-BERT: Incorporating temporal information from structured EHR data to improve prediction taskscited by 75
- 2023MOTOR: A Time-To-Event Foundation Model For Structured Medical Recordscited by 75
- 2024VoCo: A Simple-yet-Effective Volume Contrastive Learning Framework for 3D Medical Image Analysiscited by 71
- 2024AbdomenAtlas: A Large-Scale, Detailed-Annotated, & Multi-Center Dataset for Efficient Transfer Learning and Open Algorithmic Benchmarkingcited by 64
- 2023Multimodal LLMs for health grounded in individual-specific datacited by 48
- 2023Event Stream GPT: A Data Pre-processing and Modeling Library for Generative, Pre-trained Transformers over Continuous-time Sequences of Complex Eventscited by 46
- 2024Context Clues: Evaluating Long Context Models for Clinical Prediction Tasks on EHRscited by 39
- 2024SleepFM: Multi-modal Representation Learning for Sleep Across Brain Activity, ECG and Respiratory Signalscited by 38
- 2024BrainSegFounder: Towards 3D Foundation Models for Neuroimage Segmentationcited by 36
- 2024Large-Scale 3D Medical Image Pre-training with Geometric Context Priorscited by 34
- 2024EHRMamba: Towards Generalizable and Scalable Foundation Models for Electronic Health Recordscited by 31
- 2024Enhancing Foundation Models for Time Series Forecasting via Wavelet-based Tokenizationcited by 26
- 2025Clinical ModernBERT: An efficient and long context encoder for biomedical textcited by 26
- 2023MIS-FM: 3D Medical Image Segmentation using Foundation Models Pretrained on a Large-Scale Unannotated Datasetcited by 25
50 most cited of 79 All 79 (JSON)
Built on by
Citing papers per year: 2026: 2.
2 papers.