TensorFlowASR icon indicating copy to clipboard operation
TensorFlowASR copied to clipboard

Language model with Conformer

Open AgaDob opened this issue 4 years ago • 3 comments

Hi! Just wondering if the KenLM is compatible with the Conformer model, and if so if there are any example scripts on how to include it? Thank you!

AgaDob avatar May 19 '21 12:05 AgaDob

@AgaDob Currently KenLM is not compatible with the Conformer or any of Transducer models. LM is not supported for transducer models yet, but I'm working on it and also optimize beam search to use LM.

nglehuy avatar May 19 '21 12:05 nglehuy

I am curious to know about the integration of language model in your ASR Conformer implementation. As original article of Conformer also mentioned improvement of performance with using LM. You can also assist me, so that I can try.

ashishBT avatar Jul 01 '21 07:07 ashishBT

@ashishBT The LM's usually used to rescore the prob in the decoding function of the model (aka beam search). So first we need to optimize the beam search performance and then just connect with LM in the function that calculate the prob (the prob used to get the corresponding class)

nglehuy avatar Jul 23 '21 16:07 nglehuy