GPU-ACCELERATED FORWARD-BACKWARD ALGORITHM WITH APPLICATION TO LATTICE-FREE MMI

被引:3
|
作者
Ondel, Lucas [1 ]
Lam-Yee-Mui, Lea-Marie [1 ,3 ]
Kocour, Martin [2 ]
Corro, Caio Filippo [1 ]
Burget, Lukas [2 ]
机构
[1] Univ Paris Saclay, LISN, CNRS, Orsay, France
[2] Brno Univ Technol, Fac Informat Technol, Brno, Czech Republic
[3] Vocapia Res, Orsay, France
关键词
Lattice-Free MMI; end-to-end ASR; Julia language; forward-backward;
D O I
10.1109/ICASSP43922.2022.9746824
中图分类号
O42 [声学];
学科分类号
070206 ; 082403 ;
摘要
We propose to express the forward-backward algorithm in terms of operations between sparse matrices in a specific semiring. This new perspective naturally leads to a GPU-friendly algorithm which is easy to implement in Julia or any programming languages with native support of semiring algebra. We use this new implementation to train a TDNN with the LF-MMI objective function and we compare the training time of our system with PyChain-a recently introduced C++/CUDA implementation of the LF-MMI loss. Our implementation is about two times faster while not having to use any approximation such as the "leaky-HMM".
引用
收藏
页码:8417 / 8421
页数:5
相关论文
共 50 条
  • [1] DYSARTHRIC SPEECH RECOGNITION WITH LATTICE-FREE MMI
    Hermann, Enno
    Magimai-Doss, Mathew
    2020 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH, AND SIGNAL PROCESSING, 2020, : 6109 - 6113
  • [2] An Accelerated Forward-Backward Algorithm with Applications to Image Restoration Problems
    Janngam, Kobkoon
    Suantai, Suthep
    THAI JOURNAL OF MATHEMATICS, 2021, 19 (02): : 325 - 339
  • [3] GPU-Accelerated Apriori Algorithm
    Jiang, Hao
    Xu, Chen-Wei
    Liu, Zhi-Yong
    Yu, Li-Yan
    4TH ANNUAL INTERNATIONAL CONFERENCE ON INFORMATION TECHNOLOGY AND APPLICATIONS (ITA 2017), 2017, 12
  • [4] ACCELERATED AND INEXACT FORWARD-BACKWARD ALGORITHMS
    Villa, Silvia
    Salzo, Saverio
    Baldassarre, Luca
    Verri, Alessandro
    SIAM JOURNAL ON OPTIMIZATION, 2013, 23 (03) : 1607 - 1633
  • [5] A free boundary problem for interfaces in forward-backward lattice diffusion
    Helmers, Michael
    IFAC PAPERSONLINE, 2015, 48 (01): : 396 - 400
  • [6] WakeWord Detection with Alignment-Free Lattice-Free MMI
    Wang, Yiming
    Lv, Hang
    Povey, Daniel
    Xie, Lei
    Khudanpur, Sanjeev
    INTERSPEECH 2020, 2020, : 4258 - 4262
  • [7] Extended forward-backward algorithm
    Lassonde, Marc
    Nagesseur, Ludovic
    JOURNAL OF MATHEMATICAL ANALYSIS AND APPLICATIONS, 2013, 403 (01) : 167 - 172
  • [8] A generalization of forward-backward algorithm
    Azuma A.
    Matsumoto Y.
    Transactions of the Japanese Society for Artificial Intelligence, 2010, 25 (03) : 494 - 503
  • [9] A Hybrid Forward-Backward Algorithm and Its Optimization Application
    Liu, Liya
    Qin, Xiaolong
    Yao, Jen-Chih
    MATHEMATICS, 2020, 8 (03)
  • [10] On the Locality of the Forward-Backward Algorithm
    Merialdo, Bernard
    IEEE TRANSACTIONS ON SPEECH AND AUDIO PROCESSING, 1993, 1 (02): : 255 - 257