Riemannian Archetypal Analysis: Interpretable non-linear data analysis on deformed star distributions

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Diepeveen, Willem, Needell, Deanna
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914594023276544
author Diepeveen, Willem
Needell, Deanna
author_facet Diepeveen, Willem
Needell, Deanna
contents Classical archetypal analysis is appealing for its interpretability, but its linear geometry can limit performance on data with strongly non-linear structure; at the same time, existing neural extensions improve flexibility while often weakening the geometric meaning of archetypes and interpolations. In this work, we develop a Riemannian version of archetypal analysis based on data-driven pullback geometry for real-valued data, with the goal of combining the interpretability of classical archetypal analysis with the expressive power of modern non-linear models. We introduce a class of deformed star distributions together with associated pullback Riemannian geometry to provide a statistical interpretation of the resulting manifold mappings, define the Riemannian archetypal mapping (RAM) as a projection onto the manifold of geodesically convex combinations of archetypes, and propose a practical optimization scheme based on convex relaxation followed by non-convex refinement. We further propose a learning scheme that yields reasonable, albeit generally suboptimal, deformed star distributions from data. Experiments on synthetic examples and MNIST show that the resulting framework produces meaningful geodesics, useful denoising projections, and geometry-aware classifications, while also clarifying where current optimization limitations remain.
format Preprint
id arxiv_https___arxiv_org_abs_2605_24113
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Riemannian Archetypal Analysis: Interpretable non-linear data analysis on deformed star distributions
Diepeveen, Willem
Needell, Deanna
Machine Learning
Differential Geometry
Optimization and Control
Statistics Theory
Classical archetypal analysis is appealing for its interpretability, but its linear geometry can limit performance on data with strongly non-linear structure; at the same time, existing neural extensions improve flexibility while often weakening the geometric meaning of archetypes and interpolations. In this work, we develop a Riemannian version of archetypal analysis based on data-driven pullback geometry for real-valued data, with the goal of combining the interpretability of classical archetypal analysis with the expressive power of modern non-linear models. We introduce a class of deformed star distributions together with associated pullback Riemannian geometry to provide a statistical interpretation of the resulting manifold mappings, define the Riemannian archetypal mapping (RAM) as a projection onto the manifold of geodesically convex combinations of archetypes, and propose a practical optimization scheme based on convex relaxation followed by non-convex refinement. We further propose a learning scheme that yields reasonable, albeit generally suboptimal, deformed star distributions from data. Experiments on synthetic examples and MNIST show that the resulting framework produces meaningful geodesics, useful denoising projections, and geometry-aware classifications, while also clarifying where current optimization limitations remain.
title Riemannian Archetypal Analysis: Interpretable non-linear data analysis on deformed star distributions
topic Machine Learning
Differential Geometry
Optimization and Control
Statistics Theory
url https://arxiv.org/abs/2605.24113