Learning Hidden Markov Models Using Conditional Samples
Sham M. Kakade, Akshay Krishnamurthy, Gaurav Mahajan, Cyril Zhang
TL;DR
The paper tackles the cryptographic hardness of learning Hidden Markov Models by introducing interactive conditional sampling. It develops two main results: an exact conditional-probability setting that generalizes Angluin's L* to efficiently learn any rank-$r$ HMM in poly$(r,T,1/\varepsilon,\log(1/\delta))$ time, and a conditional-sampling setting where learnability is guaranteed for high-fidelity distributions with complexity poly$(r,T,O,1/\Delta^*,1/\varepsilon,\log(1/\delta))$. The core ideas are an observable-operator representation with a basis-based efficient encoding, a circulant-structure propagation of coefficients, and a novel perturbation analysis that controls error amplification across long sequences; the fidelity parameter $\Delta^*$ plays a central role in determining tractability. These results extend to broader latent low-rank distributions and connect interactive learning with DFA-learning through a robust L*-style framework, suggesting practical avenues for interactive training of sequential models while highlighting open questions about removing fidelity assumptions.
Abstract
This paper is concerned with the computational complexity of learning the Hidden Markov Model (HMM). Although HMMs are some of the most widely used tools in sequential and time series modeling, they are cryptographically hard to learn in the standard setting where one has access to i.i.d. samples of observation sequences. In this paper, we depart from this setup and consider an interactive access model, in which the algorithm can query for samples from the conditional distributions of the HMMs. We show that interactive access to the HMM enables computationally efficient learning algorithms, thereby bypassing cryptographic hardness. Specifically, we obtain efficient algorithms for learning HMMs in two settings: (a) An easier setting where we have query access to the exact conditional probabilities. Here our algorithm runs in polynomial time and makes polynomially many queries to approximate any HMM in total variation distance. (b) A harder setting where we can only obtain samples from the conditional distributions. Here the performance of the algorithm depends on a new parameter, called the fidelity of the HMM. We show that this captures cryptographically hard instances and previously known positive results. We also show that these results extend to a broader class of distributions with latent low rank structure. Our algorithms can be viewed as generalizations and robustifications of Angluin's $L^*$ algorithm for learning deterministic finite automata from membership queries.
