Journal of Optoelectronics · Laser, Volume. 35, Issue 8, 785(2024)

Image quality assessment based on ViT and multi-task self-supervised learning

WANG Huacheng, SANG Qingbing*, and HU Cong
Author Affiliations
  • School of Artificial Intelligence and Computer, Jiangnan University, Wuxi, Jiangsu 214122, China
  • show less

    An image quality assessment method based on multi-task self-supervised learning is proposed to address the existing deep learning-based image quality assessment methods, which suffer from overfitting and insufficient generalization performance due to insufficient labeled data. First,17 distortion type images are synthesized by the algorithm and the full reference mean deviation similarity index (MDSI) score and distortion type are used as 2 labels for the synthesized distortion images. Subsequently, multi-task self-supervised learning on vision transformer (ViT) for predicting MDSI scores and distortion types. Finally, the trained model is fine-tuned on the downstream task to migrate the semantic features learned from the upstream task to the downstream task. The method in this paper is fully compared with mainstream no reference image quality assessment (NR-IQA) methods on several publicly available image quality assessment datasets, and the test results on LIVE, CSIQ, TID2013, and CID2013 are all improved by 1 to 2 percentage points compared with the best performing algorithms, which indicates that the proposed algorithm outperforms most mainstream unreferenced image quality assessment algorithms.

    Tools

    Get Citation

    Copy Citation Text

    WANG Huacheng, SANG Qingbing, HU Cong. Image quality assessment based on ViT and multi-task self-supervised learning[J]. Journal of Optoelectronics · Laser, 2024, 35(8): 785

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category:

    Received: Dec. 25, 2022

    Accepted: Dec. 13, 2024

    Published Online: Dec. 13, 2024

    The Author Email: SANG Qingbing (qingbings@jiangnan.edu.cn)

    DOI:10.16136/j.joel.2024.08.0858

    Topics