arXiv reaDer
CTCプレフィックスを使用したテキスト行認識のためのシーケンス間モデルのスコアリング
Rescoring Sequence-to-Sequence Models for Text Line Recognition with CTC-Prefixes
コネクショニスト時分類(CTC)アプローチとは対照的に、手書きテキスト認識(HTR)のシーケンス間(S2S)モデルには、シーケンスの最後で頻繁に発生する単語のスキップや繰り返しなどのエラーが発生します。このホワイトペーパーでは、両方のアプローチの長所を組み合わせるために、S2Sデコード中にCTC-Prefix-Scoreを使用することを提案します。これにより、ビーム検索中に、CTC信頼行列に従って無効なパスにペナルティが課せられます。私たちのネットワークアーキテクチャは、ビジュアルバックボーンとしての畳み込みニューラルネットワーク(CNN)、エンコーダーとしての双方向長短期記憶セル(LSTM)、および相互注意層が挿入されたトランスフォーマーであるデコーダーで構成されています。 CTCの信頼度はエンコーダーで計算されますが、Transformerは文字ごとのS2Sデコードにのみ使用されます。この設定は、IAM、Rimes、およびStAZHの3つのHTRデータセットで評価されます。 IAMでは、合成データでモデルを事前トレーニングし、現代英語の文字ベースの言語モデルを含めると、2.95%の競争力のある文字エラー率(CER)を達成します。他の最先端のアプローチと比較して、私たちのモデルは約10〜20分の1のパラメーターを必要とします。 GitHubへのこのリンクを介して共有実装にアクセスします:https://github.com/Planet-AI-GmbH/tfaip-hybrid-ctc-s2s。
In contrast to Connectionist Temporal Classification (CTC) approaches, Sequence-To-Sequence (S2S) models for Handwritten Text Recognition (HTR) suffer from errors such as skipped or repeated words which often occur at the end of a sequence. In this paper, to combine the best of both approaches, we propose to use the CTC-Prefix-Score during S2S decoding. Hereby, during beam search, paths that are invalid according to the CTC confidence matrix are penalised. Our network architecture is composed of a Convolutional Neural Network (CNN) as visual backbone, bidirectional Long-Short-Term-Memory-Cells (LSTMs) as encoder, and a decoder which is a Transformer with inserted mutual attention layers. The CTC confidences are computed on the encoder while the Transformer is only used for character-wise S2S decoding. We evaluate this setup on three HTR data sets: IAM, Rimes, and StAZH. On IAM, we achieve a competitive Character Error Rate (CER) of 2.95% when pretraining our model on synthetic data and including a character-based language model for contemporary English. Compared to other state-of-the-art approaches, our model requires about 10-20 times less parameters. Access our shared implementations via this link to GitHub: https://github.com/Planet-AI-GmbH/tfaip-hybrid-ctc-s2s.
updated: Tue Mar 29 2022 15:47:24 GMT+0000 (UTC)
published: Tue Oct 12 2021 11:40:05 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト