Wavelets Meet Transformers: New AI Brings Blurry Underwater Images Into Focus

The ocean is one of the least forgiving environments for a camera. Light is absorbed unevenly as it travels through seawater, with red wavelengths vanishing within the first few meters and blue dominating everything deeper down. Suspended particles scatter what little light remains, blurring edges, washing out textures, and distorting colors until photographs of shipwrecks, pipelines, or marine organisms become ghostly, low-contrast impressions of reality. For engineers who rely on underwater imagery to identify materials, inspect infrastructure, and keep remotely operated vehicles safe, this degradation is not merely an aesthetic nuisance; it is an operational hazard. A new study published in the journal Cluster Computing by Ruolan Chen, Huibo Zhou, Bingyang Wang, and Hui Xie of Harbin Normal University in China proposes a deep learning architecture, called WFGformer, designed specifically to reverse that damage and restore clarity, color, and fine structural detail to images captured beneath the waves.

The core insight behind WFGformer is that underwater degradation behaves differently at different scales and frequencies within an image. Coarse structures such as the silhouette of a wreck or the outline of a reef survive the journey through the water column far better than fine textures like coral polyps, rivets on a hull, or the subtle gradations of a fish’s scales. Rather than forcing a single neural network to treat every pixel identically, the researchers begin by applying Haar wavelet decomposition, a classical mathematical technique that splits an image into a set of sub-bands capturing progressively finer levels of detail. The Haar wavelet, one of the simplest and oldest members of the wavelet family, uses short, box-like filters that separate an image into a low-frequency approximation and three high-frequency components encoding horizontal, vertical, and diagonal edges. By stacking this operation hierarchically, the network obtains a multi-resolution representation in which each frequency band can be processed and refined according to its own characteristics.

Working in the wavelet domain gives the model a principled way to allocate its capacity. Low-frequency bands carry the overall illumination and color information that must be corrected to undo the greenish or bluish cast typical of underwater scenes, while high-frequency bands contain the edges and textures that enhancement algorithms so often smooth away in their pursuit of denoising. Because the decomposition is invertible, whatever corrections the network learns in the frequency domain can be mapped back to a clean, full-resolution output image. This divide-and-conquer strategy echoes a broader trend in modern image restoration, where frequency-domain reasoning has proven especially powerful for tasks in which degradation is spatially varying and scale-dependent, exactly the conditions that prevail in turbid, light-starved seawater.

The second pillar of the architecture is a module the authors call the frequency-spatial local interaction transformer, or FLIT. Transformers, the architecture family that revolutionized natural language processing and now dominates computer vision, owe their power to self-attention, a mechanism that lets every part of an image consult every other part when deciding how to transform itself. The catch is computational cost: standard attention scales quadratically with the number of pixels, which is prohibitive for high-resolution restoration tasks. FLIT sidesteps this bottleneck by transferring the attention computation from the spatial domain into the frequency domain using the fast Fourier transform. In Fourier space, a global operation that would require comparing every pixel pair in the spatial domain collapses into simple element-wise multiplications, allowing the module to model long-range dependencies across the entire image at a fraction of the cost. The result is a transformer block that can reason about global structure, such as the overall color gradient from the bright surface toward the dark deep, without sacrificing efficiency.

FLIT does more than relocate attention into the frequency domain. The module also calibrates the weights of the wavelet features produced by the decomposition stage, deciding dynamically which frequency bands deserve amplification and which should be suppressed for each individual image. A murky estuary photograph with heavy backscatter might benefit from aggressive attenuation of certain high-frequency noise components, whereas a relatively clear shallow-water shot might call for strong reinforcement of fine texture. By learning this calibration end to end from data, the network performs what the authors describe as efficient feature optimization and structural detail enhancement, tailoring its frequency response to the specific degradation profile of every scene it encounters rather than applying a one-size-fits-all correction.

Complementing FLIT is a second transformer branch, the geometric-dilated attention transformer, or GDAT, which the researchers constructed to strengthen the recovery of both global image structures and local details. Dilated attention borrows an idea from dilated convolutions, expanding the receptive field of attention operations by sampling positions at increasing intervals, so that a single layer can aggregate information from a wide neighborhood while still preserving fine local relationships. The dual-branch design means the network processes the image along two parallel pathways, one specialized for the broad geometric scaffolding of the scene and the other for the intricate local texture that gives underwater photographs their sense of material realism. Fusing these complementary streams allows WFGformer to avoid a common failure mode of enhancement networks, which restore pleasing global color while leaving details smeared, or sharpen textures while introducing global color artifacts.

The architecture is completed by a multi-kernel residual feed-forward network, a component that refines the local feature fusion and nonlinear transformation performed within each transformer block. Feed-forward layers in transformers have been characterized in prior research as key-value memories that store and retrieve transformation patterns learned during training, and the multi-kernel design equips these layers with several parallel convolutional kernels of different sizes. Small kernels capture pixel-adjacent relationships, while larger kernels integrate context over broader regions, and residual connections ensure that information flows stably through the deep stack of layers. Together with the wavelet front end and the two transformer branches, this component rounds out a pipeline in which every stage has a clearly defined role: decompose, attend globally in frequency space, attend locally with geometric dilation, and transform nonlinearly with multi-scale kernels.

The authors validated their design through extensive ablation and comparative experiments on multiple synthetic and real-world underwater image datasets. Ablation studies, in which individual modules are removed one at a time, allow researchers to verify that each component contributes measurably to the final performance, and the team reports that the combination of wavelet decomposition, frequency-domain attention, the dual-branch transformer, and the multi-kernel feed-forward network outperformed existing approaches in restoring clarity and retaining detail. The evaluation spans both synthetic data, where degraded images are generated algorithmically from clean references so that ground truth is available for quantitative comparison, and real-world captures, which present the messier, unmodeled distortions that practical deployment demands. Benchmark datasets for underwater enhancement, such as those introduced in earlier landmark work on underwater image restoration, have become standard proving grounds for exactly this kind of head-to-head evaluation.

The significance of the work extends well beyond prettier vacation photos. Reliable underwater vision underpins marine material identification, the inspection of offshore platforms, subsea cables, and pipelines, the navigation of autonomous and remotely operated vehicles, and emerging applications from automotive wading systems that must see through flooded roads to bioinspired soft robots exploring the deep sea. Physics-driven approaches, including polarization-based de-scattering methods, attack the problem from the direction of optics and light transport, while earlier deep learning efforts relied on convolutional networks, generative adversarial frameworks, and, more recently, transformer models such as the U-shape transformer and wavelet-based designs presented at major computer vision conferences. WFGformer sits squarely within this rapidly evolving lineage, pushing the frontier by combining classical signal processing, in the form of Haar wavelets, with state-of-the-art attention machinery.

What makes the approach compelling is its synthesis of old and new. Wavelets have been a cornerstone of signal processing since Ingrid Daubechies formulated compactly supported orthonormal wavelet bases in the late 1980s, prized for their ability to localize information in both space and frequency. Transformers, by contrast, are barely a decade old, yet their capacity for modeling global context has already reshaped image restoration, deblurring, deraining, and super-resolution. By routing attention through the Fourier domain and letting it calibrate wavelet features, WFGformer demonstrates that the most effective tools for seeing clearly underwater may come not from choosing between classical mathematics and modern deep learning, but from wiring them together. As ocean industries expand and autonomous underwater systems multiply, networks of this kind could become the standard optical correction layer through which humanity views the roughly seventy percent of the planet that lies beneath the sea.

Subject of Research: Deep learning-based underwater image enhancement using wavelet frequency-domain decomposition and dual-branch transformer architecture

Article Title: WFGformer: an underwater image enhancement method based on wavelet frequency domain decomposition and dual-branch transformer

Article References: Chen, R., Zhou, H., Wang, B., & Xie, H. (2026). WFGformer: an underwater image enhancement method based on wavelet frequency domain decomposition and dual-branch transformer. Cluster Computing, 29(13), Article 752. https://doi.org/10.1007/s10586-026-06559-y

Image Credits: AI Generated

DOI: 10.1007/s10586-026-06559-y

Keywords: underwater image enhancement, WFGformer, transformer, Haar wavelet decomposition, frequency-domain attention, image restoration, deep learning, computer vision, marine robotics, self-attention, fast Fourier transform, Cluster Computing