Before revision:  However, CHMM enhances the  hidden Markov model 's flexibility and context representation capability by predicting token-wise transition and emission probabilities from the BERT embeddings of the input tokens .
After revision:  However, CHMM enhances the hidden Markov model with the contextual representation power of pre-trained language models. Specifically, CHMM learns token-wise transition and emission probabilities from the BERT embeddings of the input tokens .