Enhancing DETR with Attention-Based Thresholding for Efficient Early Japanese Book Reorganization
Ryuto Ishibashi, Hayata Kaneko, Lin Meng · 2023
Analysis of early Japanese books is an essential clue for researching the history and culture of the period, however, the books are written in pre-modern Japanese called Kuzushiji, which is difficult to read unless one is an expert. Kuzushiji by OCR has been popular in recent years, although recognition of Kuzushiji by end-to-end object detection is difficult with conventional CNN-based models such as YOLO. This paper uses the Transformer-based object detection model DETR, which outperforms CNNs in the field of object detection and improves DETR performance using threshold processing. The threshold processing for Self-Attention can reduce redundant dependencies between each patch and accelerate DETR training. In this experiment, DN-DAB-DETR-R50 with thresholding achieved +2.0%F1in total and up to +7.3%F1per book compared to the vanilla model and prove the efficiency of the thresholding. However, threshold processing increases computational costs slightly, and future work is improving the problem by patch pooling according to attention weights using its sparsity with thresholding.