arXiv:2510.24260cs.CV2025-10被引 1

用序列模型提升去阴影效果,避免颜色失真。

DeshadowMamba: Deshadowing as 1D Sequential Similarity

  • 将去阴影任务视为序列建模,用Mamba捕捉长程依赖
  • 提出CrossGate机制,精准引入阴影相关上下文
  • 引入色彩一致性约束,有效防止颜色污染

近期图像去阴影的深度模型多依赖注意力机制捕捉长距离依赖,但其固定注意力模式常混入无关区域的光照信息,导致结构扭曲与色彩不一致。本文从序列建模视角重新审视去阴影问题,探索使用Mamba——一种通过方向性状态转移传播全局上下文的可选择状态空间模型。该机制在保持位置连续性的同时实现高效全局感受野。然而直接应用于图像数据时,Mamba缺乏对阴影-非阴影语义的感知,仍易受邻近区域颜色干扰。为此,我们提出CrossGate,一种方向性调制机制,将阴影感知的相似性注入Mamba输入门,实现沿转移轴的选择性上下文融合。为进一步保障外观保真度,引入ColorShift正则化,一种基于全局色彩统计的对比学习目标。通过构造结构化的有信息负样本,引导模型抑制色彩污染,实现鲁棒的色彩恢复。上述组件共同使序列建模适配去阴影所需的结构完整性与色相一致性。在多个公开基准上的大量实验表明,DeshadowMamba在视觉质量与定量性能上均达到当前最优水平。

原文摘要 · Abstract (English)

Recent deep models for image shadow removal often rely on attention-based architectures to capture long-range dependencies. However, their fixed attention patterns tend to mix illumination cues from irrelevant regions, leading to distorted structures and inconsistent colors. In this work, we revisit shadow removal from a sequence modeling perspective and explore the use of Mamba, a selective state space model that propagates global context through directional state transitions. These transitions yield an efficient global receptive field while preserving positional continuity. Despite its potential, directly applying Mamba to image data is suboptimal, since it lacks awareness of shadow-non-shadow semantics and remains susceptible to color interference from nearby regions. To address these limitations, we propose CrossGate, a directional modulation mechanism that injects shadow-aware similarity into Mamba's input gate, allowing selective integration of relevant context along transition axes. To further ensure appearance fidelity, we introduce ColorShift regularization, a contrastive learning objective driven by global color statistics. By synthesizing structured informative negatives, it guides the model to suppress color contamination and achieve robust color restoration. Together, these components adapt sequence modeling to the structural integrity and chromatic consistency required for shadow removal. Extensive experiments on public benchmarks demonstrate that DeshadowMamba achieves state-of-the-art visual quality and strong quantitative performance.

去阴影序列建模Mamba色彩一致性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。