<p>Underwater image restoration is a fundamental technique for marine vision applications. Although supervised learning–based methods have achieved notable progress, their performance heavily relies on large-scale, high-quality paired datasets, which are difficult to acquire in real-world underwater environments. Semi-supervised learning, particularly Mean Teacher–based frameworks, partially alleviates this limitation by leveraging unlabeled data. However, existing methods still suffer from limited supervision diversity caused by single-teacher pseudo labeling and insufficient capability to jointly model global color distortion and local detail degradation. To address these issues, we propose a Dual Collaborative Mean Teacher (DCMT) framework for semi-supervised underwater image restoration. DCMT introduces a homogeneous dual-teacher architecture, in which two teacher models are independently maintained through the exponential moving average (EMA) updates from two separately optimized student networks. This design enhances supervision diversity by coupling each teacher with a different student trajectory, thereby reducing the risk of pseudo-label error accumulation and providing more robust consistency guidance. In addition, a reliability-aware pseudo-label selection mechanism is introduced to conservatively filter teacher predictions according to relative quality assessment. We also employ a SwinCBAM-based hybrid attention design as an effective combination of global context modeling and local feature refinement, enabling joint modeling of global color correction and local detail restoration. Extensive experiments on multiple benchmark datasets show that the proposed method achieves strong quantitative and visual performance, with consistent improvements in color fidelity and detail preservation.</p>

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

Semi-supervised underwater image restoration based on dual-mean teachers and hybrid attention

  • Dinghong Lu,
  • Xiaonan Luo,
  • Wenjie Li,
  • Peining Chen,
  • Yongxin Li

摘要

Underwater image restoration is a fundamental technique for marine vision applications. Although supervised learning–based methods have achieved notable progress, their performance heavily relies on large-scale, high-quality paired datasets, which are difficult to acquire in real-world underwater environments. Semi-supervised learning, particularly Mean Teacher–based frameworks, partially alleviates this limitation by leveraging unlabeled data. However, existing methods still suffer from limited supervision diversity caused by single-teacher pseudo labeling and insufficient capability to jointly model global color distortion and local detail degradation. To address these issues, we propose a Dual Collaborative Mean Teacher (DCMT) framework for semi-supervised underwater image restoration. DCMT introduces a homogeneous dual-teacher architecture, in which two teacher models are independently maintained through the exponential moving average (EMA) updates from two separately optimized student networks. This design enhances supervision diversity by coupling each teacher with a different student trajectory, thereby reducing the risk of pseudo-label error accumulation and providing more robust consistency guidance. In addition, a reliability-aware pseudo-label selection mechanism is introduced to conservatively filter teacher predictions according to relative quality assessment. We also employ a SwinCBAM-based hybrid attention design as an effective combination of global context modeling and local feature refinement, enabling joint modeling of global color correction and local detail restoration. Extensive experiments on multiple benchmark datasets show that the proposed method achieves strong quantitative and visual performance, with consistent improvements in color fidelity and detail preservation.