atlas

Research map / Image generation / Autoregressive and latent generation

Transformer-Based No-Reference Image Quality Assessment via Supervised Contrastive Learning

AAAI 2024 · Image generation › Autoregressive and latent generation · approach: Learning-based image quality assessment · 45 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes SaTQA, an NR-IQA model that pre-trains a ResNet-50 with supervised contrastive learning on the synthetic KADIS dataset to learn degradation features, then fuses them with Transformer-CNN multi-stream features via a patch attention block.

Problem

No-reference image quality assessment generalizes poorly to unknown authentic distortions, since existing CNN and Transformer models are trained without distinguishing distortion types and levels.

Approach

Use supervised contrastive learning on a large synthetic dataset (KADIS), with labels given by distortion type and level and no subjective scores, to learn degradation features. Combine these with perceptual features from a ViT-based multi-stream backbone (CNN plus Transformer branches) through cross-attention in a patch attention block.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
use SaTQA instead of the strongest listed baselineNR-IQA · LIVE-FBPLCC: 0.625 -> 0.676Helped
use SaTQA instead of the strongest listed baselineNR-IQA · KONIQSROCC: 0.915 -> 0.930Helped
add supervised contrastive learning (SCL) (ablation)SaTQA · LIVE ChallengePLCC: 0.827 -> 0.871Helped
add supervised contrastive learning (SCL) (ablation)SaTQA · CSIQPLCC: 0.922 -> 0.963Helped
add the Multi-Stream Block (MSB) (ablation)SaTQA with SCL · LIVE ChallengePLCC: 0.871 -> 0.880Helped
add the Multi-Stream Block (MSB) (ablation)SaTQA with SCL · CSIQPLCC: 0.963 -> 0.970Helped
add Patch Attention Block (PAB) (ablation)SaTQA with SCL and MSB · LIVE ChallengePLCC: 0.880 -> 0.903Helped
add Patch Attention Block (PAB) (ablation)SaTQA with SCL and MSB · CSIQPLCC: 0.970 -> 0.972Helped
use MSB rather than MR50 in the top backbone branch (ablation)SaTQA · CSIQPLCC: 0.966 -> 0.972Helped
use MR50 rather than MSB in the bottom pre-training branch (ablation)SaTQA · CSIQPLCC: 0.970 -> 0.972Helped
combine MHSA, DeformConv, and DwConv branches (MA+DF+DW) (ablation)MSB · CSIQPLCC: 0.957 -> 0.972Helped
combine MHSA, DeformConv, and DwConv branches (MA+DF+DW) (ablation)MSB · LIVE ChallengePLCC: 0.878 -> 0.903Helped

Surprising findings

Limitations

Experimental setup

Datasets: KADIS LIVE CSIQ TID2013 KADID-10K LIVE Challenge KONIQ LIVE-FB

Benchmarks: LIVE CSIQ TID2013 KADID-10K LIVE Challenge KONIQ LIVE-FB

Metrics: PLCC SROCC

Baselines: DIIVINE BRISQUE ILNIQE BIECON MEON WaDIQaM DBCNN TIQA MetaIQA P2P-BM HyperIQA TReS

Builds on: supervised contrastive learning ViT CONTRIQUE MANIQA CBAM

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy