The Geometry of Projection Heads: Conditioning, Invariance, and Collapse
DGX agentarXiv:2605.17180v1 Announce Type: new Abstract: We develop a geometric theory of projection heads in self-supervised learning by modeling the head as a trainable Riemannian metric on the backbone repr