AIDB Daily Papers
無制限OCR:人間の作業記憶を模倣した長文文書の効率的な読み取り
※ 日本語タイトル・ポイントはAIによる自動生成です。正確な内容は原論文をご確認ください。
ポイント
- 長文のOCR処理におけるKVキャッシュの増大による効率低下を、人間の作業記憶を模倣することで解決した。
- 提案手法は、デコーダーの全アテンション層をReference Sliding Window Attention (R-SWA)に置き換えることで、計算コストを削減し、KVキャッシュを一定に保つ。
- このモデルは、標準的な32Kの最大長で数十ページを単一フォワードパスで読み取ることができ、ASRや翻訳など他のタスクにも応用可能である。
Abstract
Recently, end-to-end OCR models, exemplified by DeepSeek OCR, have once again thrust OCR into the spotlight. A widely held view is that employing a large language model (LLM) as the decoder allows the model to leverage the prior distribution of language, leading to improved OCR performance. However, the downside is equally evident: as the output sequence lengthens, the accumulated KV cache drives up memory consumption and progressively slows down generation. This stands in stark contrast to humans, who exhibit no such decline in efficiency during long-horizon copying tasks. In this technical report, we propose Unlimited OCR, a model designed to emulate human parsing working memory. Taking DeepSeek OCR as the baseline, we replace all attention layers in the decoder with our proposed Reference Sliding Window Attention (R-SWA), which reduces attention computation costs while maintaining a constant KV cache throughout the entire decoding process. By combining the high compression rate of DeepSeek OCR's encoder with our constant KV cache design, Unlimited OCR can transcribe dozens of pages of documents in a single forward pass under a standard maximum length of 32K. More importantly, R-SWA is a general-purpose parsing attention mechanism - beyond OCR, it is equally applicable to tasks such as ASR, translation, etc. Codes and model weights are publicly available at http://github.com/baidu/Unlimited-OCR.
Paper AI Chat
この論文のPDF全文を対象にAIに質問できます。
質問の例: