arXiv:2607.21467cs.CV2026-07

针对水下图像不同区域退化差异,提出自适应传播的增强方法。

CLUIE: Clustering-Aware Recurrent Propagation with Local Structural Compensation for Underwater Image Enhancement

论文配图:CLUIE: Clustering-Aware Recurrent Propagation with Local Structural Compensation for Underwater Image Enhancement
图 1 · 摘自论文原文
  • 根据语义相似性动态重排令牌顺序,实现内容自适应传播。
  • 在多个基准上达到最优定量指标和视觉效果。
  • 适合需要精细区域修复的水下图像处理场景。

水下图像增强因波长依赖性光吸收、散射和后向散射而面临挑战,导致颜色失真、对比度下降和细节丢失。由于退化程度随场景深度和成像条件变化,同一图像内不同区域常呈现异质退化模式,需区域自适应恢复。尽管视觉RWKV模型能以线性复杂度建模长程依赖,但其预设扫描顺序与内容无关,难以适配空间非均匀的恢复需求。为此,我们提出聚类感知的RWKV框架CRWKV,将传统固定递归传播路径重构为内容自适应的令牌轨迹。具体地,引入聚类感知语义动态重排序(CSDR),依据语义特征相似性分组令牌,并从跨聚类上下文关系中推导动态遍历顺序,使WKV状态沿语义相关区域累积而非固定空间或光谱顺序。由于动态重排序可能破坏原始空间邻域的局部连续性,我们进一步提出暗响应调制局部传播(DMLP),通过深度可分离卷积提取局部结构响应,并利用邻域感知伪暗响应图自适应调节传播强度。从而在递归聚合前补偿局部结构线索,同时保持内容自适应的长程建模能力。在多个水下图像增强基准上的实验表明,CRWKV在定量性能和视觉质量上均达到当前最优水平。

原文摘要 · Abstract (English)

Underwater image enhancement remains challenging due to wavelength-dependent light absorption, scattering, and backscattering, which jointly cause color distortion, contrast degradation, and detail loss. Since these degradations vary with scene depth and imaging conditions, different regions within the same image often exhibit heterogeneous degradation patterns and thus require region-adaptive restoration. Although visual RWKV models offer an efficient linear-complexity solution for long-range dependency modeling, their predefined scanning orders are content-agnostic and therefore fail to adapt recurrent state propagation to spatially non-uniform restoration demands. To address this limitation, we propose a Clustering-aware RWKV framework, termed CRWKV, which reformulates the fixed recurrent propagation path of conventional RWKV into a content-adaptive token trajectory. Specifically, we introduce Clustering-aware Semantic Dynamic Reordering (CSDR), which groups tokens according to semantic feature similarity and derives a dynamic traversal order from inter-cluster contextual relations. This design enables WKV states to be accumulated along semantically correlated regions rather than fixed spatial or spectral orders. Since dynamic reordering may disrupt the local continuity of original spatial neighborhoods, we further propose Dark-response Modulated Local Propagation (DMLP), which extracts local structural responses via depth-wise convolution and adaptively modulates their propagation strength using a neighborhood-aware pseudo-dark response map. In this way, local structural cues are compensated before recurrent aggregation while preserving content-adaptive long-range modeling. Extensive experiments on multiple underwater image enhancement benchmarks demonstrate that CRWKV achieves state-of-the-art quantitative performance and superior visual quality.

图像增强水下成像动态重排序RWKV

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