提出联合退化处理网络,提升水下图像增强效果
JDPNet: A Network Based on Joint Degradation Processing for Underwater Image Enhancement
- 设计联合特征挖掘模块与概率引导策略,统一处理耦合退化
- 在6个公开数据集和2个自建数据集上达当前最优性能
- 兼顾色彩、清晰度与对比度,模型参数少、计算成本低
由于水下环境复杂且水介质特性多变,水下图像不可避免地受到多种退化影响。这些退化呈现非线性耦合而非简单叠加,使得有效处理此类耦合退化尤为困难。现有方法多针对特定退化设计独立分支或模块,忽视退化间的潜在关联信息,难以从底层捕捉多重退化的非线性交互。为此,本文提出JDPNet,一种基于联合退化处理的网络,旨在统一框架内挖掘并整合耦合退化中的潜在信息。具体而言,引入联合特征挖掘模块及概率自助分布策略,以实现对耦合退化特征的有效挖掘与统一调节;同时设计新型AquaBalanceLoss,平衡颜色、清晰度与对比度,引导网络学习多退化联合损失。在六个公开水下数据集及两个本研究构建的新数据集上的实验表明,JDPNet在性能、参数量与计算开销之间取得更优权衡,达到当前最优水平。
原文摘要 · Abstract (English)
Given the complexity of underwater environments and the variability of water as a medium, underwater images are inevitably subject to various types of degradation. The degradations present nonlinear coupling rather than simple superposition, which renders the effective processing of such coupled degradations particularly challenging. Most existing methods focus on designing specific branches, modules, or strategies for specific degradations, with little attention paid to the potential information embedded in their coupling. Consequently, they struggle to effectively capture and process the nonlinear interactions of multiple degradations from a bottom-up perspective. To address this issue, we propose JDPNet, a joint degradation processing network, that mines and unifies the potential information inherent in coupled degradations within a unified framework. Specifically, we introduce a joint feature-mining module, along with a probabilistic bootstrap distribution strategy, to facilitate effective mining and unified adjustment of coupled degradation features. Furthermore, to balance color, clarity, and contrast, we design a novel AquaBalanceLoss to guide the network in learning from multiple coupled degradation losses. Experiments on six publicly available underwater datasets, as well as two new datasets constructed in this study, show that JDPNet exhibits state-of-the-art performance while offering a better tradeoff between performance, parameter size, and computational cost.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。