Dense Feature Learning via Linear Structure Preservation in Medical Data
Not on the map. Why?
Abstract (arXiv)
Deep learning models for medical data are typically trained using task specific objectives that encourage representations to collapse onto a small number of discriminative directions. While effective for individual prediction problems, this paradigm underutilizes the rich structure of clinical data and limits the transferability, stability, and interpretability of learned features. In this work, we propose dense feature learning, a representation centric framework that explicitly shapes the linear structure of medical embeddings. Our approach operates directly on embedding matrices, encouraging spectral balance, subspace consistency, and feature orthogonality through objectives defined entirely in terms of linear algebraic properties. Without relying on labels or generative reconstruction, dense feature learning produces representations with higher effective rank, improved conditioning, and greater stability across time. Empirical evaluations across longitudinal EHR data, clinical text, and multimodal patient representations demonstrate consistent improvements in downstream linear performance, robustness, and subspace alignment compared to supervised and self supervised baselines. These results suggest that learning to span clinical variation may be as important as learning to predict clinical outcomes, and position representation geometry as a first class objective in medical AI.
Key facts. No citing papers in the map yet (data through 13 Oct 2026). Builds on 28 of its 57 references that are in the map. First posted on arXiv 7 Feb 2026.
Builds on
28 of 57 references.
References · 28
- 2019BioBERT: a pre-trained biomedical language representation model for biomedical text miningcited by 1,942
- 2019Publicly Available Clinical BERT Embeddingscited by 999
- 2023Time-LLM: Time Series Forecasting by Reprogramming Large Language Modelscited by 693
- 2020Med-BERT: pre-trained contextualized embeddings on large-scale structured electronic health records for disease predictioncited by 369
- 2023LLM4TS: Aligning Pre-Trained LLMs as Data-Efficient Time-Series Forecasterscited by 128
- 2017Machine Learning for Survival Analysis: A Surveycited by 112
- 2023EHRSHOT: An EHR Benchmark for Few-Shot Evaluation of Foundation Modelscited by 103
- 2020Language Models Are An Effective Patient Representation Learning Technique For Electronic Health Record Datacited by 86
- 2020BioMegatron: Larger Biomedical Domain Language Modelcited by 76
- 2023MOTOR: A Time-To-Event Foundation Model For Structured Medical Recordscited by 76
- 2023Multimodal LLMs for health grounded in individual-specific datacited by 48
- 2022Diagnosing failures of fairness transfer across distribution shift in real-world medical settingscited by 45
16 more
- 2024Context Clues: Evaluating Long Context Models for Clinical Prediction Tasks on EHRscited by 39
- 2025TIMER: Temporal Instruction Modeling and Evaluation for Longitudinal Clinical Recordscited by 34
- 2024EHRMamba: Towards Generalizable and Scalable Foundation Models for Electronic Health Recordscited by 31
- 2025Clinical ModernBERT: An efficient and long context encoder for biomedical textcited by 26
- 2024Emergency Department Decision Support using Clinical Pseudo-notescited by 22
- 2020Extremely Dense Point Correspondences using a Learned Feature Descriptorcited by 20
- 2024Can Large Language Models abstract Medical Coded Language?cited by 20
- 2024CORE-BEHRT: A Carefully Optimized and Rigorously Evaluated BEHRTcited by 20
- 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
- 2025A Case Study Exploring the Current Landscape of Synthetic Medical Record Generation with Commercial LLMscited by 13
- 2025Raptor: Scalable Train-Free Embeddings for 3D Medical Volumes Leveraging Pretrained 2D Foundation Modelscited by 13
- 2024FEET: A Framework for Evaluating Embedding Techniquescited by 8
- 2019Scale-Adaptive Neural Dense Features: Learning via Hierarchical Context Aggregationcited by 5
- 2025LabTOP: A Unified Model for Lab Test Outcome Prediction on Electronic Health Recordscited by 5
- 2025Reflections from Research Roundtables at the Conference on Health, Inference, and Learning (CHIL) 2025cited by 4
Built on by
Not cited yet.