[edit]
GrAE ScaLE: Grassmannian AutoEncoder for Scalable Linearly-invariant Embeddings
Proceedings of the 2nd Conference on Topology, Algebra, and Geometry in Data Science(TAG-DS 2026), PMLR 334(2):345-363, 2026.
Abstract
Supervised learning and generation on subspace-valued data remain difficult in practice, since existing techniques for embeddings of points on the Grassmann manifold can be costly and difficult to scale. In this work, we introduce the Grassmannian Autoencoder (GrAE), an autoencoder architecture designed for subspace-valued data, together with a variational extension, the Grassmannian Variational Autoencoder (GrVAE). These architectures produce informative, robust, low-dimensional latent representations with improved computational scalability. We demonstrate our approaches across three novel subspace datasets designed to help disambiguate failure modes and support benchmarking for future subspace-learning architectures. Our GrAE architectures{’} compact latent representations offer an efficient and expressive alternative for supervised learning on the Grassmann manifold and a practical step toward scalable subspace learning.