Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

QualiTeacher: Quality-Conditioned Pseudo-Labeling for Real-World Image Restoration

About

Real-world image restoration (RWIR) is a highly challenging task due to the absence of clean ground-truth images. Many recent methods resort to pseudo-label (PL) supervision, often within a Mean-Teacher (MT) framework. However, these methods face a critical paradox: unconditionally trusting the often imperfect, low-quality PLs forces the student model to learn undesirable artifacts, while discarding them severely limits data diversity and impairs model generalization. In this paper, we propose QualiTeacher, a novel framework that transforms pseudo-label quality from a noisy liability into a conditional supervisory signal. Instead of filtering, QualiTeacher explicitly conditions the student model on the quality of the PLs, estimated by an ensemble of complementary non-reference image quality assessment (NR-IQA) models spanning low-level distortion and semantic-level assessment. This strategy teaches the student network to learn a quality-graded restoration manifold, enabling it to understand what constitutes different quality levels. Consequently, it can not only avoid mimicking artifacts from low-quality labels but also extrapolate to generate results of higher quality than the teacher itself. To ensure the robustness and accuracy of this quality-driven learning, we further enhance the process with a multi-augmentation scheme to diversify the PL quality spectrum, a score-based preference optimization strategy inspired by Direct Preference Optimization (DPO) to enforce a monotonically ordered quality separation, and a cropped consistency loss to prevent adversarial over-optimization (reward hacking) of the IQA models. Experiments on standard RWIR benchmarks demonstrate that QualiTeacher can serve as a plug-and-play strategy to improve the quality of the existing pseudo-labeling framework, establishing a new paradigm for learning from imperfect supervision. Code will be released.

Fengyang Xiao, Jingjia Feng, Peng Hu, Dingming Zhang, Lei Xu, Guanyi Qin, Lu Li, Chunming He, Sina Farsiu• 2026

Related benchmarks

TaskDatasetResultRank
Object DetectionExDark
mAP (Mean Average Precision)79.1
58
DesnowingReal 3000
CNNIQA67.5
7
Low-light Image EnhancementReal 3000
CNNIQA0.669
7
DerainingReal 3000
CNNIQA0.694
6
DehazingReal 3000
CNNIQA0.706
5
DehazingRTTS (inference)
Score3.62
4
Image DesnowingRealSnow 10K (test)
Score3.58
4
Underwater Image EnhancementReal 3000
CNNIQA0.515
4
Showing 8 of 8 rows

Other info

Follow for update