Search papers, labs, and topics across Lattice.
We introduce a novel likelihood-based framework for Archetypal Analysis (AA) that efficiently handles discrete data by employing local quadratic approximations and sequential minimal optimization with an active-set method. This approach overcomes the limitations of traditional least-squares AA for Bernoulli, Poisson, and multinomial distributions, enabling interpretable low-dimensional representations of complex discrete datasets. The framework is validated by its computational efficiency, accurate model recovery on synthetic data, and successful application to diverse biological datasets, achieving competitive fits and stable archetypes.
Interpretable archetypes emerge from discrete data using a principled likelihood framework, unlocking insights in genomics and microbiome analysis previously inaccessible to standard AA.
Archetypal Analysis (AA) represents observations as convex combinations of extremal data-driven profiles, yielding interpretable low-dimensional descriptions of complex datasets. Classical AA relies on a least-squares objective, which is poorly suited to discrete observations such as binary, count, and categorical data. We introduce an efficient likelihood-based framework for AA supporting Bernoulli, Poisson, and multinomial observation models. Our optimization scheme employs local quadratic approximations of the negative log-likelihood, enabling constrained updates through sequential minimal optimization (SMO) and an active-set method. Scalability is improved by bounding the active set while preserving simplex feasibility. We further introduce a cross-validated predictive likelihood criterion for selecting the number of archetypes, providing a principled alternative to reconstruction-error heuristics and stability-based diagnostics. Synthetic experiments demonstrate computational efficiency and accurate recovery of model complexity. Applications to single-cell RNA sequencing, microbiome composition, and somatic mutation data show that the learned archetypes capture interpretable domain-specific structures while achieving competitive likelihood fits and stable solutions. Overall, the proposed framework enables efficient likelihood-based archetypal analysis of discrete data, complemented by predictive likelihood-based model selection.