HyperAIHyperAI

Command Palette

Search for a command to run...

音声認識のためのマルチヘッド状態空間モデル

概要

近年、状態空間モデル(State Space Models: SSMs)は、小規模なシーケンスおよび言語モデリングタスクにおいて、多くのアテンションベースの手法と並ぶ乃至はそれを上回る成果を示しており、注目を集めている。本論文では、特別なゲーティング機構を備えたマルチヘッド状態空間(Multi-Head State Space: MH-SSM)アーキテクチャを提案する。このモデルでは、並列に配置された複数のヘッドが、シーケンスデータ上の局所的およびグローバルな時系列ダイナミクスを学習するよう訓練される。このMH-SSMは、Transformerエンコーダにおけるマルチヘッドアテンションの即座の置き換えとして利用可能であり、LibriSpeech音声認識コーパスにおいて、Transformer Transducerを著しく上回る性能を達成した。さらに、TransformerブロックにMH-SSM層を追加したアーキテクチャ、すなわち「Stateformer」を構築した。このStateformerは、外部言語モデルを用いずに、開発セットにおいて1.76%/4.37%、テストセットにおいて1.91%/4.36%という単語誤り率(WER)を達成し、LibriSpeechタスクにおいて最先端の性能を実現した。


AIでAIを構築

アイデアからローンチまで — 無料のAIコーディング支援、すぐに使える環境、最高のGPU価格でAI開発を加速。

AI コーディング補助
すぐに使える GPU
最適な料金体系

HyperAI Newsletters

最新情報を購読する
北京時間 毎週月曜日の午前9時 に、その週の最新情報をメールでお届けします
メール配信サービスは MailChimp によって提供されています
音声認識のためのマルチヘッド状態空間モデル | 記事 | HyperAI超神経