kaldi grammar on-the-fly 解决未登录词识别

背景

场景一:互联网技术大会,主讲人中有不少国外大牛,演讲中经常中英文夹杂。现在需要你们组做现场语音识别支持。如何在中文识别很好的前提下,对中英文夹杂部分也能很好的识别?
场景二:友商要做一次路演,演讲中有很多以前闻所未闻的固定句式。现在需要你们做现场语音识别支持。如何在通用识别效果不降的前提下,对那些不常见的固定内容也能很好的识别?

上面两个场景设计到实际语音识别中经常遇到的两个问题:

  • 临时有未登录词需要识别,比如英文词汇或者通讯录人名识别等,如:这个idea真的很amazing。打电话给鲁智深。
  • 需要对一些常见的句式做增强,比如最近需要做活动,一些特定的句式或者题库部分需要识别100%正确。如:边小溪给我讲个推荐一部电影吧。我想看扫毒之天地对决。

针对第一类未登录词识别需求,常见的做法有两种:

  • 一种是收集相关语料和新词,重新训练语言模型,同时修改词典以及重新生成相关资源,接着重新构图,测试上线。语言模型如果重新训练时间比较久,可以考虑用新增语料训练一个小语言模型,与原先的语言模型插值生成新语言模型。
  • 第二种方式就是on the fly 方式,根据新词构建一个小网络,与原先的HCLG合并生成一个新的HCLG。

针对第二类固定句式强化需求,常见的做法有两种:

  • 一种是同上面一样,重新训练语言模型。
  • 第二种方式就是lattice lmrescore。注意lattice lmrescore是无法识别未登录词的。

未登录词识别两种解决办法中:
第一种方式效果好。缺点是速度慢,资源变动多。演讲稿大纲变了,又新添了很多英文新词汇......
第二种方式效果一般,优点是速度快,主体资源无需变动,这一点对需要对接很多很多业务的工作人员来说是很重要的。这次重点讲第二种方式,也就是kaldi 5.5版本新增的Support for grammars and graphs with on-the-fly parts部分

原理

解码过程中,根据StateId 是norm还是special,走不同的网络,norm情况走Top HCLG,special 根据fst instance走对应的Part HCLG。

解码时

special 的依据是判断final-prob是否是某个特定值(4096.0)。具体来说就是解码过程内部循环ArcIterator初始化时做判断,kaldi代码见下面两张图:


lattice-faster-decoder.cc

decoder/grammar-fst.h

aiter type 是 fst::ArcIterator<fst::GrammarFst>,也就是说online2bin/online2-wav-nnet3-latgen-grammar代码相比较原先的解码代码:主要变化是ConstFst 变为 GrammarFst,对应的decoder模板参数也发生了变化。

解码前

主要是grammar的构造,utils/mkgraph 中会根据情况调用make-grammar-fst,主要做两件事:

  • 设置special state 的 final-cost = 4096.0
  • 插入epsilon arcs确保一个state只能进入一个网络。

流程

具体见local/grammar/extend_vocab_demo.sh脚本

TOP HCLG

需要修改三个地方:

  • NGram,arpa模型中需要有unk,需要对unk做一个替换。
cat traih.lm | sed 's/<unk>/#nonterm:unk/g' | \ 
arpa2fst --disambig-symbol=#0 --read-symbol-table=$lang_base/words.txt - | \
fstrmsymbols --remove-from-output=true "echo $nonterm_unk|" - $lang_base/G.fst
  • words.txt 添加如下部分
#nonterm_begin 125463
#nonterm_end 125464
#nonterm:unk 125465
  • phones.txt
#nonterm_bos 277
#nonterm_begin 278
#nonterm_end 279
#nonterm_reenter 280
#nonterm:unk 281

其他部分见通常的HCLG.fst构建过程

Part HCLG部分(也就是需要增加的新词HCLG部分)

  • 词典准备
dota 1.0    d ao1 t a1
iphone 1.0  aa ai1 f eng4
...
  • 生成G.txt以及G.fst
    构图HCLG1.fst

  • 合并make-grammar-fst
    HCLGo.fst HCLG1.fst → HCLG.fst

一些坑

  • 如果不做特殊处理,指定了词典的中文语言模型是无法生成unk的。
    如 打电话给鲁智深 分词后 打电话 给 鲁 智 深
    解决办法
    • 对人名打标签,让人名作为一个整体:
      打电话 给 鲁智深 指定词典中如果没有鲁智深这个词,就会变成:打电话 给 unk
    • 正则表达式生成一些包含unk的常见句子:
      打电话 给 unk
      给 unk 发短信
      ...
  • This framework is limited to work only with left-biphone models,like chain model

Reference
https://kaldi-asr.org/doc/grammar.html

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 205,386评论 6 479
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,939评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,851评论 0 341
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,953评论 1 278
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,971评论 5 369
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,784评论 1 283
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,126评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,765评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,148评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,744评论 2 323
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,858评论 1 333
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,479评论 4 322
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,080评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,053评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,278评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,245评论 2 352
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,590评论 2 343