Dense Feature Learning via Linear Structure Preservation in Medical Data

TL;DR

This paper introduces dense feature learning, a linear structure-preserving framework for medical data embeddings that enhances stability, transferability, and interpretability without relying on labels.

Contribution

It proposes a novel, label-free method that explicitly shapes the linear structure of medical embeddings to improve their quality and robustness.

Findings

Higher effective rank and better conditioning of embeddings.

Improved downstream linear task performance.

Enhanced robustness and subspace alignment across data modalities.

Abstract

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.