English

Archetypal Analysis++: Rethinking the Initialization Strategy

Machine Learning 2025-04-09 v4

Abstract

Archetypal analysis is a matrix factorization method with convexity constraints. Due to local minima, a good initialization is essential, but frequently used initialization methods yield either sub-optimal starting points or are prone to get stuck in poor local minima. In this paper, we propose archetypal analysis++ (AA++), a probabilistic initialization strategy for archetypal analysis that sequentially samples points based on their influence on the objective function, similar to kk-means++. In fact, we argue that kk-means++ already approximates the proposed initialization method. Furthermore, we suggest to adapt an efficient Monte Carlo approximation of kk-means++ to AA++. In an extensive empirical evaluation of 15 real-world data sets of varying sizes and dimensionalities and considering two pre-processing strategies, we show that AA++ almost always outperforms all baselines, including the most frequently used ones.

Keywords

Cite

@article{arxiv.2301.13748,
  title  = {Archetypal Analysis++: Rethinking the Initialization Strategy},
  author = {Sebastian Mair and Jens Sjölund},
  journal= {arXiv preprint arXiv:2301.13748},
  year   = {2025}
}

Comments

27 pages, 17 figures, accepted at the Transactions on Machine Learning Research

R2 v1 2026-06-28T08:28:12.340Z