DR-Mamba: Automatic Inference-Time Domain Adaptation for Document Image Binarization via Sample-Conditioned Detail-Background Suppression
On the map: State-space sequence models (S4, Mamba) → Mamba for image restoration
Abstract (arXiv)
Degraded document image binarization is sensitive to domain shifts caused by paper aging, bleed-through, stains, shadows, and uneven illumination, and the foreground-background separation of recent learning-based methods can become unstable on unseen degradation domains. We propose DR-Mamba, a sample-conditioned detail-background suppression framework that performs automatic inference-time domain adaptation for document image binarization. Unlike test-time adaptation methods that require gradient updates or auxiliary data at inference, DR-Mamba adapts to each input document through input-dependent gates within a single forward pass, requiring no target-domain labels, no fine-tuning, and no test-time parameter updates. Instead of using Mamba-style selective scanning as a single generic feature path, DR-Mamba reinterprets it as fast-slow route modeling: a fast detail route captures local stroke structures, while a slow background route accumulates spatially persistent degradation responses. The two routes are integrated through an input-dependent subtractive gate that explicitly suppresses background interference rather than fusing features by addition or concatenation. We further add full-resolution detail-guided reconstruction and thin-stroke-aware supervision to recover fine strokes lost during downsampling. Evaluated under a leave-one-year-out protocol on DIBCO-style benchmarks, where each held-out year is treated as an unseen degradation domain, DR-Mamba shows that per-document, per-location subtractive suppression improves cross-domain robustness, with particularly strong performance on the most severely degraded held-out fold.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Mamba for image restoration, one of 10 directions in State-space sequence models (S4, Mamba). Builds on 16 of its 32 references that are in the map. First posted on arXiv 21 Jun 2026.
Builds on
16 of 32 references.
References · 16
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
- 2022A ConvNet for the 2020scited by 3,579
- 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,465
- 2017Mixed Precision Trainingcited by 1,156
- 2024Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Modelcited by 1,056
- 2024VMamba: Visual State Space Modelcited by 969
- 2020clDice -- A Novel Topology-Preserving Loss Function for Tubular Structure Segmentationcited by 294
- 2017Tversky loss function for image segmentation using 3D fully convolutional deep networkscited by 228
- 2018Boundary loss for highly unbalanced segmentationcited by 190
- 2023DocDiff: Document Enhancement via Residual Diffusion Modelscited by 37
- 2022DocEnTr: An End-to-End Document Image Enhancement Transformercited by 28
4 more
- 2017Document Image Binarization with Fully Convolutional Neural Networkscited by 25
- 2024NAF-DPM: A Nonlinear Activation-Free Diffusion Probabilistic Model for Document Enhancementcited by 9
- 2023DocBinFormer: A Two-Level Transformer Network for Effective Document Image Binarizationcited by 4
- 2025MFE-GAN: Efficient GAN-based Framework for Document Image Enhancement and Binarization with Multi-scale Feature Extractioncited by 2
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026DeepMine-Mamba: Mitigating Information Dilution in Mamba-Based State Space Models for Document Image Binarization
- 2024Adaptive Multi Scale Document Binarisation Using Vision Mamba
- 2025DvD: Unleashing a Generative Paradigm for Document Dewarping via Coordinates-based Diffusion Model
- 2023DECDM: Document Enhancement using Cycle-Consistent Diffusion Models
- 2024Multimodal Adaptive Inference for Document Image Classification with Anytime Early Exiting
- 2026TOLA: Text-aware One-Step Latent Adaptation for Diffusion-based Text Image Super-Resolution
- 2026MinerU-Diffusion: Rethinking Document OCR as Inverse Rendering via Diffusion Decoding
- 2026Discriminator-Guided Adaptive Diffusion for Source-Free Test-Time Adaptation under Image Corruptions