Skip to content

Enables computing the gradient of the parameters of Hidden Markov Models (HMMs)

License

Notifications You must be signed in to change notification settings

idiap/HMMGradients.jl

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

32 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

HMMGradients.jl

CI codecov DOI

This package enables computing the gradient of the parameters of Hidden Markov Models (HMMs). This makes it possible to perform HMM training using gradient based methods like stochastic gradient descent, which is necessary for example when neural networks are involved, e.g. in modern automatic speech recognition systems. Check out this TIDIGITS recipe for an example.

Formally, this package extends ChainRulesCore making it possible to train HMM models using the automatic differentiation frameworks of Julia, for example using Zygote and machine learning libraries like Flux. The package also provides numerical stable algorithms to compute forward, backward and posterior probabilities of HMMs.

Installation

To install the package, simply issue the following command in the Julia REPL:

] add HMMGradients

For more information check the documentation.