mlsanityData to 14 Oct 2026

AURORA: Contextual Orthogonalization for Geometric Representation Learning in Healthcare Foundation Models

arXiv 2605.17765 ↗First posted 18 May 2026cs.LGcited by 0

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

  1. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  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. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  6. 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
  7. 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,139
  8. 2023Segment Anythingcited by 8,234
  9. 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
  10. 2023Visual Instruction Tuningcited by 7,869
  11. 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
  12. 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,968
38 more
  1. 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,355
  2. 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,598
  3. 2023A Survey of Large Language Modelscited by 2,576
  4. 2022Large Language Models Encode Clinical Knowledgecited by 1,830
  5. 2019Towards Robust Monocular Depth Estimation: Mixing Datasets for Zero-shot Cross-dataset Transfercited by 1,495
  6. 20194D Spatio-Temporal ConvNets: Minkowski Convolutional Neural Networkscited by 1,208
  7. 2021iBOT: Image BERT Pre-Training with Online Tokenizercited by 903
  8. 2021CrossViT: Cross-Attention Multi-Scale Vision Transformer for Image Classificationcited by 806
  9. 2019Contrastive Representation Distillationcited by 748
  10. 2018CCNet: Criss-Cross Attention for Semantic Segmentationcited by 718
  11. 2018Bag of Tricks for Image Classification with Convolutional Neural Networkscited by 431
  12. 2025HealthBench: Evaluating Large Language Models Towards Improved Human Healthcited by 429
  13. 2020Med-BERT: pre-trained contextualized embeddings on large-scale structured electronic health records for disease predictioncited by 373
  14. 2024SegMamba: Long-range Sequential Modeling Mamba For 3D Medical Image Segmentationcited by 266
  15. 2024Foundation Models for Time Series Analysis: A Tutorial and Surveycited by 239
  16. 2019Cross Attention Network for Few-shot Classificationcited by 219
  17. 2024BetterBench: Assessing AI Benchmarks, Uncovering Issues, and Establishing Best Practicescited by 114
  18. 2022UNETR++: Delving into Efficient and Accurate 3D Medical Image Segmentationcited by 112
  19. 2023Large-scale Training of Foundation Models for Wearable Biosignalscited by 95
  20. 2021MedPerf: Open Benchmarking Platform for Medical Artificial Intelligence using Federated Evaluationcited by 88
  21. 2020Language Models Are An Effective Patient Representation Learning Technique For Electronic Health Record Datacited by 87
  22. 2023MOTOR: A Time-To-Event Foundation Model For Structured Medical Recordscited by 78
  23. 2024Foundation Model for Advancing Healthcare: Challenges, Opportunities, and Future Directionscited by 77
  24. 2025MedHELM: Holistic Evaluation of Large Language Models for Medical Taskscited by 64
  25. 2024Context Clues: Evaluating Long Context Models for Clinical Prediction Tasks on EHRscited by 39
  26. 2024EHRMamba: Towards Generalizable and Scalable Foundation Models for Electronic Health Recordscited by 32
  27. 2025Clinical ModernBERT: An efficient and long context encoder for biomedical textcited by 26
  28. 2019Reproducibility in Machine Learning for Healthcited by 25
  29. 2025GENERator: A Long-Context Generative Genomic Foundation Modelcited by 22
  30. 2024Can Large Language Models abstract Medical Coded Language?cited by 20
  31. 2024CORE-BEHRT: A Carefully Optimized and Rigorously Evaluated BEHRTcited by 20
  32. 2024E3D-GPT: Enhanced 3D Visual Foundation for Medical Vision-Language Modelcited by 19
  33. 2025HiMAE: Hierarchical Masked Autoencoders Discover Resolution-Specific Structure in Wearable Time Seriescited by 19
  34. 2024Text Serialization and Their Relationship with the Conventional Paradigms of Tabular Machine Learningcited by 17
  35. 2023E2ENet: Dynamic Sparse Feature Fusion for Accurate and Efficient 3D Medical Image Segmentationcited by 16
  36. 2025Raptor: Scalable Train-Free Embeddings for 3D Medical Volumes Leveraging Pretrained 2D Foundation Modelscited by 13
  37. 2025ChronoFormer: Time-Aware Transformer Architectures for Structured Clinical Event Modelingcited by 12
  38. 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

  1. 2026DeltaRepresentation: Geometry Supervised Representation Learning of Phenotypes via Counterfactual Reasoning for Medical VLMs
  2. 2025Embeddings to Diagnosis: Latent Fragility under Agentic Perturbations in Clinical LLMs
  3. 2026Representation Before Training: A Practical Benchmark for Generative Medical Event Model Tokenization
  4. 2026The Patient is not a Moving Document: A World Model Training Paradigm for Longitudinal EHR
  5. 2026Marrying Generative Model of Healthcare Events with Digital Twin of Social Determinants of Health for Disease Reasoning
  6. 2025Learning Patient-Specific Disease Dynamics with Latent Flow Matching for Longitudinal Imaging Generation
  7. 2025DualAlign: Generating Clinically Grounded Synthetic Data
  8. 2026A General Multimodal Probing Framework for Data Fusion and Prediction with Frozen Large Language Models: Evidence from Multimodal EHR Data