Publication Details

GPU-Accelerated Forward-Backward Algorithm with Application to Lattice-Free MMI

ONDEL Yang Lucas Antoine Francois, LAM-YEE-MUI L'ea-Marie, KOCOUR Martin, CORRO Caio Filippo and BURGET Lukáš. GPU-Accelerated Forward-Backward Algorithm with Application to Lattice-Free MMI. In: ICASSP, IEEE International Conference on Acoustics, Speech and Signal Processing - Proceedings. Singapore: IEEE Signal Processing Society, 2022, pp. 8417-8421. ISBN 978-1-6654-0540-9. Available from: https://ieeexplore.ieee.org/document/9746824
Czech title
Forward-backward algoritmus akcelerovaný na GPU s aplikací pro Lattice-Free MMI trénování
Type
conference paper
Language
english
Authors
Ondel Yang Lucas Antoine Francois, Mgr., Ph.D. (UPSAC)
Lam-Yee-Mui L'ea-Marie (UPSAC)
Kocour Martin, Ing. (DCGM FIT BUT)
Corro Caio Filippo (UPSAC)
Burget Lukáš, doc. Ing., Ph.D. (DCGM FIT BUT)
URL
Keywords

Lattice-Free MMI, end-to-end ASR, Julia language, forward-backward

Abstract

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 PyChaina 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".

Published
2022
Pages
8417-8421
Proceedings
ICASSP, IEEE International Conference on Acoustics, Speech and Signal Processing - Proceedings
Conference
2022 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING (ICASSP), Singapore, SG
ISBN
978-1-6654-0540-9
Publisher
IEEE Signal Processing Society
Place
Singapore, SG
DOI
10.1109/ICASSP43922.2022.9746824
UT WoS
000864187908145
EID Scopus
2-s2.0-85130631441
BibTeX 
@INPROCEEDINGS{FITPUB12788,
   author = "Francois Antoine Lucas Yang Ondel and L'ea-Marie Lam-Yee-Mui and Martin Kocour and Filippo Caio Corro and Luk\'{a}\v{s} Burget",
   title = "GPU-Accelerated Forward-Backward Algorithm with Application to Lattice-Free MMI",
   pages = "8417--8421",
   booktitle = "ICASSP, IEEE International Conference on Acoustics, Speech and Signal Processing - Proceedings",
   year = 2022,
   location = "Singapore, SG",
   publisher = "IEEE Signal Processing Society",
   ISBN = "978-1-6654-0540-9",
   doi = "10.1109/ICASSP43922.2022.9746824",
   language = "english",
   url = "https://www.fit.vut.cz/research/publication/12788"
}
Back to top