HMMGradients.jl

Enables computing the gradient of the parameters of Hidden Markov Models (HMMs)
Author idiap
Popularity
6 Stars
Updated Last
1 Year Ago
Started In
December 2020

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.

Required Packages

Used By Packages

No packages found.