Near-Optimal Clustering in Mixture of Markov Chains
Abstract
We study the problem of clustering trajectories of length , each generated by one of K unknown ergodic Markov chains over a finite state space of size . We derive an instance-dependent, high-probability lower bound on the clustering error rate, governed by the stationary-weighted KL divergence between transition kernels. We then propose a two-stage algorithm: Stage I applies spectral clustering via a new injective Euclidean embedding for ergodic Markov chains, a contribution of independent interest enabling sharp concentration results; Stage II refines clusters with a single likelihood-based reassignment step. We prove that our algorithm achieves near-optimal clustering error with high probability under reasonable requirements on and . Preliminary experiments support our approach, and we conclude with discussions of its limitations and extensions.
Keywords
Cite
@article{arxiv.2506.01324,
title = {Near-Optimal Clustering in Mixture of Markov Chains},
author = {Junghyun Lee and Yassir Jedra and Alexandre Proutière and Se-Young Yun},
journal= {arXiv preprint arXiv:2506.01324},
year = {2026}
}
Comments
AISTATS 2026 (50 pages, 6 figures) (ver3: camera-ready version, major revisions)