Papers › Omni-Kernel Network for Image Restoration

Omni-Kernel Network for Image Restoration

24 Mar 2024Proceedings of the AAAI Conference on Artificial Intelligence 2024 3archive 2025-07-28

Yuning Cui, Wenqi Ren, Alois Knoll

Image restoration aims to reconstruct a high-quality image from a degraded low-quality observation. Recently, Transformer models have achieved promising performance on image restoration tasks due to their powerful ability to model long-range dependencies. However, the quadratically growing complexity with respect to the input size makes them inapplicable to practical applications. In this paper, we develop an efficient convolutional network for image restoration by enhancing multi-scale representation learning. To this end, we propose an omni-kernel module that consists of three branches, i.e., global, large, and local branches, to learn global-to-local feature representations efficiently. Specifically, the global branch achieves a global perceptive field via the dual-domain channel attention and frequency-gated mechanism. Furthermore, to provide multi-grained receptive fields, the large branch is formulated via different shapes of depth-wise convolutions with unusually large kernel sizes. Moreover, we complement local information using a point-wise depth-wise convolution. Finally, the proposed network, dubbed OKNet, is established by inserting the omni-kernel module into the bottleneck position for efficiency. Extensive experiments demonstrate that our network achieves state-of-the-art performance on 11 benchmark datasets for three representative image restoration tasks, including image dehazing, image desnowing, and image defocus deblurring. The code is available at https://github.com/c-yn/OKNet.

PaperPDFCode

Code

c-yn/OKNet mentioned in paperpytorch report

Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

Not run by Syntology. Nothing on this page verifies that the listed code works.

Tasks

DeblurringImage Defocus DeblurringImage DehazingImage RestorationRepresentation Learning

Results from the paper archive 2025-07-28

TaskDatasetModelMetricValueRank at snapshotLeaderboardReport
Image Dehazing SOTS Indoor OKNet PSNR 40.79 #13 of 34 Archive leaderboard report
Image Dehazing SOTS Indoor OKNet SSIM 0.996 #13 of 34 Archive leaderboard report
Image Dehazing SOTS Outdoor OKNet PSNR 37.68 #12 of 31 Archive leaderboard report
Image Dehazing SOTS Outdoor OKNet SSIM 0.995 #12 of 31 Archive leaderboard report
Image Restoration CDD-11 OKNet Average PSNR (dB) 26.33 #6 of 14 Archive leaderboard report
Image Restoration CDD-11 OKNet SSIM 0.8605 #6 of 14 Archive leaderboard report

Ranks are positions in the archive's leaderboards as they stood at the 2025-07-28 snapshot. Results published since then are not among these rows, so a rank here is not a current standing.

Methods

Absolute Position EncodingsAdamAttentionBPEDense ConnectionsDropoutLabel SmoothingLayer NormalizationLinear LayerMulti-Head AttentionPosition-Wise Feed-Forward LayerResidual ConnectionSoftmaxTransformer

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections