English

Near-Optimal Clustering in Mixture of Markov Chains

Machine Learning 2026-03-18 v3 Information Theory Machine Learning math.IT Probability

Abstract

We study the problem of clustering TT trajectories of length HH, each generated by one of K unknown ergodic Markov chains over a finite state space of size SS. 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 TT and HH. 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)