表示
超・人工知能研究所 Super AI Research Lab トップへ超・人工知能研究所AI Research Lab / ぱそ太郎工事中

since
平成38年9月29日

← 検索結果に戻る

専門用語を音声認識に反映するTurboBias:語句の木で復号時の点数を調整

TurboBias

保存した記事を見るこのブラウザに保存します。無料会員で端末間の同期・フォルダ整理
原論文の図

TurboBiasの原論文から掲載した図です。細部はクリックして拡大できます。

原論文の図の説明を読む

Fig. 1: An example of a boosting tree for words “CAT, CATS, CSV, SIT” with character-level tokenization. Below the tree is an example of how boosting scores are assigned when decoding the ”the cat is sitting” sequence.

出典:TurboBias / arXiv ↗

概要

  • 登録語句を木にまとめ、復号時に認識候補の点数を補正する。認識モデルの追加学習は要しない。
  • 著者らのCSTalksでのRNN-Tビーム探索では、登録語句Fスコアが44.2%から82.9%、WERが12.8%から9.6%になった。
  • 貪欲復号での登録外の語への影響と、語句リストを増やした際の誤受理は導入時の確認事項となる。
補足:編集した模式図で仕組みを確認する

VISUAL EXPLAINER

図でつかむ、TurboBiasの仕組み

左から右へ読む
  1. 01音声と登録語句
  2. 02語句の木を構築
  3. 03候補点数を補正
  4. 04認識文を出力

原論文の説明をもとにした模式図です。処理の細部は省略しています。

研究の背景

TurboBiasは、音声認識の出力に指定した語句を反映しやすくする手法だ。人名、製品名、専門用語など、学習データに少ない語は取り違えられやすい。字幕や議事録では、重要な語の誤りが内容の確認を難しくする。

音声認識には、候補から出力文字列を選ぶ「復号」の段階がある。著者らは対象語句のリストをこの段階で利用し、認識モデルを追加学習せずに語句の選択を助ける方法を検討した。

手法

例えば「GPU」「PyTorch」を登録する場合、まず各語句を認識器が扱う文字列単位に分け、共通する先頭部分をまとめた木を作る。語句の2単位目以降には、木の深さに応じて大きくなる点数を設定する。候補が語句から外れた場合は、途中までの加点を差し引く。

復号中は認識モデルの候補点数に木の点数を重み付きで足し、出力する文字列を選ぶ。この復号時の点数の合算を「浅い融合」という。木をGPUで語彙全体の点数をまとめて取得できる形に変換し、一つの候補を選ぶ貪欲復号と、複数候補を残すビーム探索に組み込む。

新規性

著者らの主な提案は、登録語句の木で、先頭より後の文字列単位ほど大きな点数を与える配分だ。候補を一つだけ選ぶ貪欲復号でも、語句の末尾に達する前から選択に影響させる狙いがある。

木をGPUで語彙全体について照会できる形に変換し、CTC、RNN-T、AEDの各方式で使える復号処理に組み込んだ。

従来手法との違い

CSTalksのCTCビーム探索では、著者らの比較でTurboBiasの語句Fスコアは83.2%、Pyctcdecodeは74.0%、均等に加点するGPU-PBuwは75.0%だった。Pyctcdecodeは語句への加点を使うCTC復号器であり、別の語句検出器を併用する方式ではない。

同じデータのCTC貪欲復号では、語句検出器を使うCTC-WSが79.8%、TurboBiasが64.8%だった。比較結果は探索方式で変わり、CTC-WSには音声全体とCTCの出力が必要となる。

実験結果

著者らは英語音声のCSTalks、Earnings21、MultiMedを、CTC、RNN-T、AEDで評価した。登録語句数は順に200、893、991。開発用データで加点を調整し、テスト用データで登録語句のFスコア、全体の単語誤り率WER、音声長を処理時間で割る速度指標RTFxを測った。

CSTalksのRNN-Tビーム探索では、加点なしとTurboBias使用時でFスコアが44.2%から82.9%、WERが12.8%から9.6%になった。表IIの全データセット平均RTFxは1466から1420。速度は単一のNVIDIA RTX A6000でバッチ32とし、準備運転後の3回平均で測定された。

応用の可能性

編集上の応用案 · 論文が実証した用途とは区別しています。

編集上の応用案:社内の技術講演を文字起こしする際、製品名と略語の認識を補助する。入力は収録済みの講演音声と、字幕担当者が確認した語句リスト。出力は対象語句を優先して復号した字幕の下書きとする。

認識器は既存のものを使い、語句をその認識器の文字列単位へ変換して登録する。方式自体に追加の学習用音声は不要だが、効果を確かめるには正しい字幕を人手で付けた評価用音声が要る。収録済み音声を使うため、試行に新たなセンサーは想定しない。

応用先の候補からNauhに相談できます。

導入時の検証

導入時の検証案 · 対象データでの再評価が必要です。

導入時の検証案:技術講演の収録済み音声を開発用と評価用に分け、製品名・略語のリストと、人手で正しい字幕を付けた評価用音声を用意する。方式自体の追加学習データは不要だが、認識器は別途必要だ。語句を認識器の文字列単位へ変換し、開発用音声で加点を調整する。

同じ評価用音声を加点なしとTurboBias使用時で処理し、語句Fスコア、全体のWER、処理時間を比べる。未登録語と似た音の語の誤りも確認し、語句数を増やして再測定する。収録済み音声を使う試行なので、新たなセンサーは想定しない。

限界と課題

著者らは、貪欲復号では登録外の語の精度が下がる可能性を挙げ、未知の文字列に与える点数の調整を勧めている。AEDでは文末の点数が語句への加点に押されるため、文末の点数を補正する処理を加えた。

編集上の検証事項は、登録外の語や似た音の語の誤認識、語句リスト更新時の処理速度である。2万語句まで広げた試験では誤受理によりFスコアとWERが悪化した。英語音声と指定されたモデルでの結果を、日本語の字幕業務へ直接当てはめることはできない。

応用先の候補
研究内容と応用案に基づく分類です。業界での導入実績を示すものではありません。

出典

元のタイトル・要旨を確認する(英語)

TurboBias: Universal ASR Context-Biasing powered by GPU-accelerated Phrase-Boosting Tree

Recognizing specific key phrases is an essential task for contextualized Automatic Speech Recognition (ASR). However, most existing context-biasing approaches have limitations associated with the necessity of additional model training, significantly slow down the decoding process, or constrain the choice of the ASR system type. This paper proposes a universal ASR context-biasing framework that supports all major types: CTC, Transducers, and Attention Encoder-Decoder models. The framework is based on a GPU-accelerated word boosting tree, which enables it to be used in shallow fusion mode for greedy and beam search decoding without noticeable speed degradation, even with a vast number of key phrases (up to 20K items). The obtained results showed high efficiency of the proposed method, surpassing the considered open-source context-biasing approaches in accuracy and decoding speed. Our context-biasing framework is open-sourced as a part of the NeMo toolkit.

論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新:2026-10-01