arXiv · 2605.09160
Objective-Specific Privileged Bases via Full-Prefix Matryoshka Learning
Abstract
Learned representations are often invariant to rotational transformations, leaving individual dimensions non-identifiable and interchangeable. We study how Matryoshka Representation Learning (MRL) induces a task-aligned privileged basis distinct from variance-based or regularizer-induced orderings. In the linear setting, we prove that full-prefix MRL recovers the ordered principal directions, and can be computed efficiently using shared statistics. Empirically, we demonstrate that MRL yields consistent per-dimension structure aligned with task signal, where coordinate magnitude reflects informativeness.
Explore related subjects
Keep this discovery
Arghamitra Talukder, Philippe Chlenski, Itsik Pe'er. 2026-05-09. Objective-Specific Privileged Bases via Full-Prefix Matryoshka Learning. https://arxiv.org/abs/2605.09160
Cite the original work for its findings. Save a collection to share your selection of sources.