vadimkantorov/ctc

Primer on CTC implementation in pure Python PyTorch code

A primer on CTC implementation in pure Python PyTorch code. This impl is not suitable for real-world usage, only for experimentation and research on CTC modifications. Features:

• CTC impl is in Python and its only loop is over time steps (parallelizes over batch and symbol dimensions)
• Gradients are computed via PyTorch autograd instead of a separate beta computation
• Viterbi path useful for forced alignment
• Get alignment targets out of any CTC impl, so that label smoothing or reweighting can be applied [1, 2]
• It might support double-backwards (not checked)

Very rough time measurements

Device: cuda
Log-probs shape (time X batch X channels): 128x256x32
Built-in CTC loss fwd 0.002052783966064453 bwd 0.0167086124420166
Custom CTC loss fwd 0.09685754776000977 bwd 0.14192843437194824
Custom loss matches: True
Grad matches: True
CE grad matches: True

Device: cpu
Log-probs shape (time X batch X channels): 128x256x32
Built-in CTC loss fwd 0.017746925354003906 bwd 0.21297860145568848
Custom CTC loss fwd 0.38710451126098633 bwd 5.190514087677002
Custom loss matches: True
Grad matches: True
CE grad matches: True

Very rought time measurements if custom logsumexp is used

Device: cuda
Log-probs shape (time X batch X channels): 128x256x32
Built-in CTC loss fwd 0.009581804275512695 bwd 0.012355327606201172
Custom CTC loss fwd 0.09775996208190918 bwd 0.1494584083557129
Custom loss matches: True
Grad matches: True
CE grad matches: True

Device: cpu
Log-probs shape (time X batch X channels): 128x256x32
Built-in CTC loss fwd 0.017041444778442383 bwd 0.23205327987670898
Custom CTC loss fwd 0.3748452663421631 bwd 4.206061363220215
Custom loss matches: True
Grad matches: True
CE grad matches: True

Alignment image example

References (CTC)

1. A Novel Re-weighting Method for Connectionist Temporal Classification; Li et al; https://arxiv.org/abs/1904.10619
2. Focal CTC Loss for Chinese Optical Character Recognition on Unbalanced Datasets; Feng et al; https://www.hindawi.com/journals/complexity/2019/9345861/
3. Improved training for online end-to-end speech recognition systems; Kim et al; https://arxiv.org/abs/1711.02212
4. Connectionist Temporal Classification: Labelling Unsegmented Sequence Data with Recurrent Neural Networks; Graves et all; https://www.cs.toronto.edu/~graves/icml_2006.pdf
5. Sequence Modeling With CTC, Hannun et al, https://distill.pub/2017/ctc/
6. Other CTC implementations:
   • https://github.com/rakeshvar/rnn_ctc/blob/master/nnet/ctc.py#L96
   • https://github.com/artbataev/end2end/blob/master/pytorch_end2end/src/losses/forward_backward.cpp
   • https://github.com/jamesdanged/LatticeCtc
   • https://github.com/zh217/torch-asg/blob/master/torch_asg/native/force_aligned_lattice.cpp
   • https://github.com/amaas/stanford-ctc/blob/master/ctc/ctc.py
   • https://github.com/skaae/Lasagne-CTC/blob/master/ctc_cost.py
   • https://github.com/pytorch/pytorch/blob/master/aten/src/ATen/native/LossCTC.cpp#L37
   • https://github.com/musyoku/chainer-gram-ctc https://github.com/musyoku/chainer-cuda-ctc
   • https://github.com/1ytic/warp-rnnt

References (beam search)

• https://towardsdatascience.com/beam-search-decoding-in-ctc-trained-neural-networks-5a889a3d85a7

• https://medium.com/corti-ai/ctc-networks-and-language-models-prefix-beam-search-explained-c11d1ee23306

• https://github.com/githubharald/CTCDecoder

• https://github.com/githubharald/CTCWordBeamSearch

• https://gist.github.com/awni/56369a90d03953e370f3964c826ed4b0

• https://github.com/wouterkool/stochastic-beam-search

• https://github.com/mjansche/ctc_sampling

• https://www.aclweb.org/anthology/D19-1331/

• https://arxiv.org/abs/1905.08760

• https://arxiv.org/abs/1804.07915

• http://proceedings.mlr.press/v97/cohen19a/cohen19a.pdf

• https://github.com/corticph/prefix-beam-search/