cs.CVNov 3, 2026

Wonder 4: A Mixture-of-Experts Recipe for Image Quality Enhancement

Authors: Yunan Wu, Bin Sun, Partha Acharjee, Xiaoyu Wang

Organizations: Topaz Labs Research

Abstract

A real photograph is often blurry, grainy, compressed, scratched, and low-resolution at the same time. Each defect has its own severity and its own place in the image. The corrections for these defects conflict with each other. Exist- ing models describe the whole image with one description, so they cannot say where each correction belongs. To solve this problem, we present Wonder 4, a mixture-of-experts (MoE) model for image quality enhancement. It corrects several defects of a real photograph at the same time, each in the region where it occurs. Every block of the model holds a mixture of N = 13 experts. A router, trained with the model, sends every image patch to k = 5 of them. No degradation classifier, mask, prompt, detector, or task label is used. Another issue is that standard MoE training, taken from language models, pushes every expert toward equal use. Equal use applies every correction everywhere, even where it is not needed. We replace this rule with a bias that only sets a minimum share of use. Across all patches, an expert used less than this minimum gets higher router scores until it is selected again. An expert above the minimum is not changed, so no expert is forced to be used as often as the others. The bias needs no extra loss. The method adds no extra module and no extra processing step. We evaluate Wonder 4 on DIV2K and Flickr2K, against five published methods and three open-weight pipelines. It reaches the best LPIPS, DISTS, and NIQE on all three test settings (DIV2K x4, DIV2K x8, and Flickr2K x4).

Figures & tables

Explore similar work

CardsList
  1. ImIR: Image-Instruction Tuning for All-in-One Image Restoration

    Sep 21, 2026Süleyman Aslan, Görkay Aydemir, Mısra Yavuz +5Visual Instruction TuningAll-in-One Image Restoration

  2. MiM-DiT: MoE in MoE with Diffusion Transformers for All-in-One Image Restoration

    Mar 3, 2026Lingshun Kong, Jiawei Zhang, Zhengpeng Duan +6All-in-One Image RestorationDiffusion Models for Image Restoration

  3. Restore, Assess, Repeat: A Unified Framework for Iterative Image Restoration

    Mar 27, 2026I-Hsiang Chen, Isma Hadji, Enrique Sanchez +5Image RestorationAll-in-One Image Restoration