AURORA: Contextual Orthogonalization for Geometric Representation Learning in Healthcare Foundation Models
Not on the map. Why?
Abstract (arXiv)
Recent healthcare foundation models have achieved strong predictive performance through large scale self supervised learning, yet their latent representations frequently entangle physiologic severity, intervention intensity, observational structure, and institutional workflow into shared embedding directions. While effective for downstream prediction, such representations remain semantically opaque and unstable under contextual shift. We introduce AURORA, Adaptive Uncertainty aware Representations through Orthogonalized Relational Alignment, a new framework for healthcare representation learning based on contextual latent geometry. Rather than optimizing a single unified embedding manifold, AURORA decomposes representations into orthogonal semantic subspaces corresponding to distinct contextual factors and learns relational consistency objectives within each subspace. This induces latent spaces that are both semantically disentangled and geometrically interpretable. Across multiple clinical prediction and retrieval tasks, AURORA consistently outperforms reconstruction, contrastive, and self distillation baselines while substantially improving contextual disentanglement, neighborhood purity, and robustness under institutional distribution shift. Our results suggest that latent geometry itself constitutes an important axis of healthcare foundation model design and that explicitly structuring representation space according to contextual semantics provides a complementary direction beyond conventional predictive compression objectives.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 61 of its 86 references that are in the map. First posted on arXiv 18 May 2026.
Builds on
61 of 86 references.
References · 61
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 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
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,139
- 2023Segment Anythingcited by 8,234
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
- 2023Visual Instruction Tuningcited by 7,869
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,968
38 more
- 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,355
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,598
- 2023A Survey of Large Language Modelscited by 2,576
- 2022Large Language Models Encode Clinical Knowledgecited by 1,830
- 2019Towards Robust Monocular Depth Estimation: Mixing Datasets for Zero-shot Cross-dataset Transfercited by 1,495
- 20194D Spatio-Temporal ConvNets: Minkowski Convolutional Neural Networkscited by 1,208
- 2021iBOT: Image BERT Pre-Training with Online Tokenizercited by 903
- 2021CrossViT: Cross-Attention Multi-Scale Vision Transformer for Image Classificationcited by 806
- 2019Contrastive Representation Distillationcited by 748
- 2018CCNet: Criss-Cross Attention for Semantic Segmentationcited by 718
- 2018Bag of Tricks for Image Classification with Convolutional Neural Networkscited by 431
- 2025HealthBench: Evaluating Large Language Models Towards Improved Human Healthcited by 429
- 2020Med-BERT: pre-trained contextualized embeddings on large-scale structured electronic health records for disease predictioncited by 373
- 2024SegMamba: Long-range Sequential Modeling Mamba For 3D Medical Image Segmentationcited by 266
- 2024Foundation Models for Time Series Analysis: A Tutorial and Surveycited by 239
- 2019Cross Attention Network for Few-shot Classificationcited by 219
- 2024BetterBench: Assessing AI Benchmarks, Uncovering Issues, and Establishing Best Practicescited by 114
- 2022UNETR++: Delving into Efficient and Accurate 3D Medical Image Segmentationcited by 112
- 2023Large-scale Training of Foundation Models for Wearable Biosignalscited by 95
- 2021MedPerf: Open Benchmarking Platform for Medical Artificial Intelligence using Federated Evaluationcited by 88
- 2020Language Models Are An Effective Patient Representation Learning Technique For Electronic Health Record Datacited by 87
- 2023MOTOR: A Time-To-Event Foundation Model For Structured Medical Recordscited by 78
- 2024Foundation Model for Advancing Healthcare: Challenges, Opportunities, and Future Directionscited by 77
- 2025MedHELM: Holistic Evaluation of Large Language Models for Medical Taskscited by 64
- 2024Context Clues: Evaluating Long Context Models for Clinical Prediction Tasks on EHRscited by 39
- 2024EHRMamba: Towards Generalizable and Scalable Foundation Models for Electronic Health Recordscited by 32
- 2025Clinical ModernBERT: An efficient and long context encoder for biomedical textcited by 26
- 2019Reproducibility in Machine Learning for Healthcited by 25
- 2025GENERator: A Long-Context Generative Genomic Foundation Modelcited by 22
- 2024Can Large Language Models abstract Medical Coded Language?cited by 20
- 2024CORE-BEHRT: A Carefully Optimized and Rigorously Evaluated BEHRTcited by 20
- 2024E3D-GPT: Enhanced 3D Visual Foundation for Medical Vision-Language Modelcited by 19
- 2025HiMAE: Hierarchical Masked Autoencoders Discover Resolution-Specific Structure in Wearable Time Seriescited by 19
- 2024Text Serialization and Their Relationship with the Conventional Paradigms of Tabular Machine Learningcited by 17
- 2023E2ENet: Dynamic Sparse Feature Fusion for Accurate and Efficient 3D Medical Image Segmentationcited by 16
- 2025Raptor: Scalable Train-Free Embeddings for 3D Medical Volumes Leveraging Pretrained 2D Foundation Modelscited by 13
- 2025ChronoFormer: Time-Aware Transformer Architectures for Structured Clinical Event Modelingcited by 12
- 2025Foundation models for electronic health records: representation dynamics and transferabilitycited by 11
50 most cited of 61 All 61 (JSON)
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026DeltaRepresentation: Geometry Supervised Representation Learning of Phenotypes via Counterfactual Reasoning for Medical VLMs
- 2025Embeddings to Diagnosis: Latent Fragility under Agentic Perturbations in Clinical LLMs
- 2026Representation Before Training: A Practical Benchmark for Generative Medical Event Model Tokenization
- 2026The Patient is not a Moving Document: A World Model Training Paradigm for Longitudinal EHR
- 2026Marrying Generative Model of Healthcare Events with Digital Twin of Social Determinants of Health for Disease Reasoning
- 2025Learning Patient-Specific Disease Dynamics with Latent Flow Matching for Longitudinal Imaging Generation
- 2025DualAlign: Generating Clinically Grounded Synthetic Data
- 2026A General Multimodal Probing Framework for Data Fusion and Prediction with Frozen Large Language Models: Evidence from Multimodal EHR Data