• 基于改进生成对抗网络的图像水印检测算法

    An improved generative adversarial network algorithm for image watermark detection

    • 针对现有图像水印检测算法计算效率低、参数量大且真伪判别准确性不足等问题,提出了一种基于改进生成对抗网络(GAN)的图像水印检测算法。该算法包括DSD-GAN和DCG-GAN两个模型,分别用于水印存在性检测和水印真伪判别。DSD-GAN模型通过引入深度可分离卷积和空洞卷积优化判别器,有效降低了模型的参数量,并提升了计算效率,显著提高了水印存在性检测的速度与精度。DCG-GAN模型则结合生成器生成高质量的伪造水印,并通过灰狼优化算法(Grey Wolf Optimization, GWO)优化判别器参数,进一步提升了水印真伪判别的准确性。两个模型的协同工作流程使得水印存在性检测与真伪判别得以高效实现。实验结果表明:DSD-GAN模型在确保水印不可见性和鲁棒性的同时,在水印存在性检测任务中取得了99.58%的准确率,具有优异的检测性能;DCG-GAN模型在水印真伪判别任务中,PSNR和NC指标表现优秀,且在不同嵌入强度下的准确率始终高于0.50,保障了水印的不可见性与鲁棒性。与现有算法相比,该方案在显著降低参数量的同时提升了运行效率,为复杂场景下的图像水印检测提供了一种高效可行的解决方案。

       

      Abstract: This study addresses the issues of low computational efficiency, excessive parameter quantities, and insufficient accuracy in discriminating watermark authenticity observed in existing image watermark detection algorithms. An improved GAN-based image watermark detection algorithm is proposed.The algorithm comprises two models, DSD-GAN and DCG-GAN, which are designed for watermark presence detection and watermark authenticity discrimination, respectively.The discriminator of the DSD-GAN model is optimized by incorporating depthwise separable convolution and dilated convolution, where the parameter quantity is effectively reduced and computational efficiency is enhanced. Significant improvements in both speed and accuracy of watermark presence detection are achieved through this architectural refinement.The DCG-GAN model integrates a generator where high-quality forged watermarks are synthesized, and the discriminator parameters are optimized via the Grey Wolf Optimization (GWO) algorithm, leading to a significant improvement in the accuracy of watermark authenticity discrimination.The watermark presence detection and authenticity discrimination are efficiently achieved through the collaborative workflow of the two models, where task-specific feature extraction and adversarial optimization are seamlessly integrated.Experimental results demonstrate that watermark imperceptibility and robustness are ensured while a detection accuracy of 99.58% is achieved by the DSD-GAN model in watermark presence detection tasks, indicating superior performance. For the DCG-GAN model in watermark authenticity discrimination, excellent PSNR and NC metrics are observed, and the accuracy is consistently maintained above 0.50 across varying embedding strengths, ensuring the imperceptibility and robustness of the watermark are comprehensively preserved.The parameter quantity is effectively reduced and computational efficiency is enhanced in the proposed solution compared to other algorithms, while a more efficient solution is provided for image watermark detection in complex scenarios through architectural optimization and lightweight design strategies.

       

    /

    返回文章
    返回