用轻量点解码器减少提示点,提速三倍同时提升文本检测准确率。
ET-SAM: Efficient Point Prompt Prediction in SAM for Unified Scene Text Detection and Layout Analysis
- 设计轻量点解码器生成词级热图,仅需少量前景点作为提示。
- 在多数据集联合训练下,总文本等三个数据集平均F-score提升11.0%。
- 支持词级、行级等多种标注格式,适合多源异构文本数据训练。
基于分割一切模型(SAM)的先前工作在统一场景文本检测与版面分析上取得了良好效果。然而,依赖像素级文本分割生成数千个前景点作为提示,导致推理延迟高且数据利用率低。为此,我们提出ET-SAM,一种基于SAM的高效统一文本检测与版面分析框架,包含两个解码器。技术上,定制轻量点解码器生成词级热图,实现少量前景点提示,消除冗余点提示并加速推理。脱离对像素级分割的依赖后,进一步设计联合训练策略,融合具有多层级、仅词级、仅行级标注的数据集作为统一训练集。针对这些数据集,在点解码器与分层掩码解码器中引入三组可学习任务提示,以缓解数据差异。大量实验表明,相比以往基于SAM的架构,ET-SAM实现约3倍推理加速,同时在HierText上表现相当;在Total-Text、CTW1500和ICDAR15上平均F-score提升11.0%。
原文摘要 · Abstract (English)
Previous works based on Segment Anything Model (SAM) have achieved promising performance in unified scene text detection and layout analysis. However, the typical reliance on pixel-level text segmentation for sampling thousands of foreground points as prompts leads to unsatisfied inference latency and limited data utilization. To address above issues, we propose ET-SAM, an Efficient framework with two decoders for unified scene Text detection and layout analysis based on SAM. Technically, we customize a lightweight point decoder that produces word heatmaps for achieving a few foreground points, thereby eliminating excessive point prompts and accelerating inference. Without the dependence on pixel-level segmentation, we further design a joint training strategy to leverage existing data with heterogeneous text-level annotations. Specifically, the datasets with multi-level, word-level only, and line-level only annotations are combined in parallel as a unified training set. For these datasets, we introduce three corresponding sets of learnable task prompts in both the point decoder and hierarchical mask decoder to mitigate discrepancies across datasets.Extensive experiments demonstrate that, compared to the previous SAM-based architecture, ET-SAM achieves about 3$\times$ inference acceleration while obtaining competitive performance on HierText, and improves an average of 11.0% F-score on Total-Text, CTW1500, and ICDAR15.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。