# End-To-End Underwater Video Enhancement: Dataset and Model

arXiv: https://arxiv.org/abs/2403.11506 · HTML: https://mlsanity.com/paper/2403.11506/

## Key facts

- Cited by 5 papers in the map (data through 14 Oct 2026).
- Builds on 29 of its 54 references that are in the map.
- First posted on arXiv 18 Mar 2024.

## Builds on (most cited first, 29 of 29)

- [Adam: A Method for Stochastic Optimization](https://mlsanity.com/paper/1412.6980/) (2014)
- [Layer Normalization](https://mlsanity.com/paper/1607.06450/) (2016)
- [Squeeze-and-Excitation Networks](https://mlsanity.com/paper/1709.01507/) (2017)
- [MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications](https://mlsanity.com/paper/1704.04861/) (2017)
- [A ConvNet for the 2020s](https://mlsanity.com/paper/2201.03545/) (2022)
- [Real-Time Single Image and Video Super-Resolution Using an Efficient Sub-Pixel Convolutional Neural Network](https://mlsanity.com/paper/1609.05158/) (2016)
- [Instance Normalization: The Missing Ingredient for Fast Stylization](https://mlsanity.com/paper/1607.08022/) (2016)
- [Video Enhancement with Task-Oriented Flow](https://mlsanity.com/paper/1711.09078/) (2017)
- [EDVR: Video Restoration with Enhanced Deformable Convolutional Networks](https://mlsanity.com/paper/1905.02716/) (2019)
- [BasicVSR: The Search for Essential Components in Video Super-Resolution and Beyond](https://mlsanity.com/paper/2012.02181/) (2020)
- [An Underwater Image Enhancement Benchmark Dataset and Beyond](https://mlsanity.com/paper/1901.05495/) (2019)
- [TDAN: Temporally Deformable Alignment Network for Video Super-Resolution](https://mlsanity.com/paper/1812.02898/) (2018)
- [VRT: A Video Restoration Transformer](https://mlsanity.com/paper/2201.12288/) (2022)
- [Fast Underwater Image Enhancement for Improved Visual Perception](https://mlsanity.com/paper/1903.09766/) (2019)
- [Underwater Image Enhancement via Medium Transmission-Guided Multi-Color Space Embedding](https://mlsanity.com/paper/2104.13015/) (2021)
- [U-shape Transformer for Underwater Image Enhancement](https://mlsanity.com/paper/2111.11843/) (2021)
- [Enhancing Underwater Imagery using Generative Adversarial Networks](https://mlsanity.com/paper/1801.04011/) (2018)
- [A Simple Baseline for Video Restoration with Grouped Spatial-temporal Shift](https://mlsanity.com/paper/2206.10810/) (2022)
- [Underwater Ranker: Learn Which Is Better and How to Be Better](https://mlsanity.com/paper/2208.06857/) (2022)
- [PUGAN: Physical Model-Guided Underwater Image Enhancement Using GAN with Dual-Discriminators](https://mlsanity.com/paper/2306.08918/) (2023)
- [UIEC^2-Net: CNN-based Underwater Image Enhancement Using Two Color Space](https://mlsanity.com/paper/2103.07138/) (2021)
- [Domain Adaptation for Underwater Image Enhancement](https://mlsanity.com/paper/2108.09650/) (2021)
- [Temporally Consistent Video Colorization with Deep Feature Propagation and Self-regularization Learning](https://mlsanity.com/paper/2110.04562/) (2021)
- [Underwater Light Field Retention : Neural Rendering for Underwater Imaging](https://mlsanity.com/paper/2203.11006/) (2022)
- [Video Dehazing via a Multi-Range Temporal Alignment Network with Physical Prior](https://mlsanity.com/paper/2303.09757/) (2023)
- [Five A+ Network: You Only Need 9K Parameters for Underwater Image Enhancement](https://mlsanity.com/paper/2305.08824/) (2023)
- [UWGAN: Underwater GAN for Real-world Underwater Color Restoration and Dehazing](https://mlsanity.com/paper/1912.10269/) (2019)
- [Marine Video Kit: A New Marine Video Dataset for Content-based Analysis and Retrieval](https://mlsanity.com/paper/2209.11518/) (2022)
- [UIEDP:Underwater Image Enhancement with Diffusion Prior](https://mlsanity.com/paper/2312.06240/) (2023)

## Built on by (most cited first, 5 of 5)

- [A Physical Model-Guided Framework for Underwater Image Enhancement and Depth Estimation](https://mlsanity.com/paper/2407.04230/) (2024)
- [Enhancing Underwater Imaging with 4-D Light Fields: Dataset and Method](https://mlsanity.com/paper/2408.17339/) (2024)
- [Unsupervised Methods for Video Quality Improvement: A Survey of Restoration and Enhancement Techniques](https://mlsanity.com/paper/2507.08375/) (2025)
- [Impact of Underwater Image Enhancement on Feature Matching](https://mlsanity.com/paper/2507.21715/) (2025)
- [Enhancing Underwater Light Field Images via Global Geometry-aware Diffusion Process](https://mlsanity.com/paper/2601.21179/) (2026)

## Abstract (arXiv)

Underwater video enhancement (UVE) aims to improve the visibility and frame quality of underwater videos, which has significant implications for marine research and exploration. However, existing methods primarily focus on developing image enhancement algorithms to enhance each frame independently. There is a lack of supervised datasets and models specifically tailored for UVE tasks. To fill this gap, we construct the Synthetic Underwater Video Enhancement (SUVE) dataset, comprising 840 diverse underwater-style videos paired with ground-truth reference videos. Based on this dataset, we train a novel underwater video enhancement model, UVENet, which utilizes inter-frame relationships to achieve better enhancement performance. Through extensive experiments on both synthetic and real underwater videos, we demonstrate the effectiveness of our approach. This study represents the first comprehensive exploration of UVE to our knowledge. The code is available at https://anonymous.4open.science/r/UVENet.