Watermark removal is essential for Optical Character Recognition and digital document reconstruction. Document images differ significantly from natural scene images, necessitating tailored image processing methods. Traditional Transformer-based methods, though effective, face scalability issues due to the quadratic complexity of attention computations. Inspired by Mamba, which scales linearly with context length, we introduce a semantically-guided Mamba approach, MambaDW. This two-stage method first removes most watermarks and then employs semantic-guided 2D selective scanning for precise enhancement and global information fusion. We also incorporate a self-supervised loss to enhance generalizability on unlabeled real data. Our experiments show MambaDW’s effectiveness in document watermark removal.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

MambaDW: Semantic-Aware Mamba for Document Watermark Removal

  • Yifan Liu,
  • Mingfu Yan,
  • He Hua,
  • Jiancheng Huang,
  • Shifeng Chen

摘要

Watermark removal is essential for Optical Character Recognition and digital document reconstruction. Document images differ significantly from natural scene images, necessitating tailored image processing methods. Traditional Transformer-based methods, though effective, face scalability issues due to the quadratic complexity of attention computations. Inspired by Mamba, which scales linearly with context length, we introduce a semantically-guided Mamba approach, MambaDW. This two-stage method first removes most watermarks and then employs semantic-guided 2D selective scanning for precise enhancement and global information fusion. We also incorporate a self-supervised loss to enhance generalizability on unlabeled real data. Our experiments show MambaDW’s effectiveness in document watermark removal.