Evidence Hub

YOLO-WSR: A wavelet feature and structure guided diffusion model for detection and automated restoration of cultural heritage artifact images

Chen Y, Wang X, Shaohua W

PloS one, 2026

Abstract

In response to the complexity and diversity involved in damage detection and automated restoration of cultural heritage artifact images associated with intangible cultural heritage preservation, this study proposes a YOLO-based restoration framework that integrates wavelet features with structure-guided diffusion, termed YOLO-WSR. The proposed method enhances detection accuracy through wavelet feature upsampling and employs a structure-guided diffusion generation mechanism to achieve effective restoration of texture and structural details in damaged artifact images. In addition, a CLIP-IQA-based feedback optimization module is introduced to perform no-reference quality assessment on the restored outputs and iteratively refine the restoration results, thereby forming a closed-loop collaborative framework that integrates detection, restoration, and evaluation. Experiments conducted on multiple publicly available datasets demonstrate the effectiveness of YOLO-WSR. The results show that the proposed method consistently improves performance over existing approaches, achieving approximately 5-6% improvement in mAP@50 for damage detection, while obtaining enhanced PSNR, SSIM, LPIPS, and SRCC performance for restoration quality evaluation. Further ablation studies demonstrate the contribution of each component to the overall framework. The proposed method provides an effective and quantifiable technical solution for the digital preservation of cultural heritage artifact images,…

Read the paper

Related studies

Go deeper in Evidence Hub

Evidence Hub adds a study data sheet (design, sample size, evidence level, key numbers), an AI summary and a chat grounded in the paper's text.

Open this study Sign up free