Search papers, labs, and topics across Lattice.
This paper introduces N-gram Injection (NGI), a method for dynamically adapting transformer-based handwritten text recognition (HTR) models to target corpora with shifted language distributions. NGI injects n-gram language model information, estimated on a target corpus, directly into the transformer decoder at inference time. Experiments on three handwritten datasets demonstrate that NGI significantly reduces the performance gap between source and target corpora without requiring additional training on target image-text pairs.
Dramatically improve handwritten text recognition on out-of-domain data by injecting n-gram language models into the transformer decoder at inference time, sidestepping costly retraining.
Transformer-based encoder-decoder networks have recently achieved impressive results in handwritten text recognition, partly thanks to their auto-regressive decoder which implicitly learns a language model. However, such networks suffer from a large performance drop when evaluated on a target corpus whose language distribution is shifted from the source text seen during training. To retain recognition accuracy despite this language shift, we propose an external n-gram injection (NGI) for dynamic adaptation of the network's language modeling at inference time. Our method allows switching to an n-gram language model estimated on a corpus close to the target distribution, therefore mitigating bias without any extra training on target image-text pairs. We opt for an early injection of the n-gram into the transformer decoder so that the network learns to fully leverage text-only data at the low additional cost of n-gram inference. Experiments on three handwritten datasets demonstrate that the proposed NGI significantly reduces the performance gap between source and target corpora.
