Saved in:
Bibliographic Details
Main Authors: He, Haoyuan, Zheng, Yu, Zhou, Jie, Lu, Jiwen
Format: Preprint
Published: 2026
Subjects:
Online Access:https://arxiv.org/abs/2602.21508
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915815644725248
author He, Haoyuan
Zheng, Yu
Zhou, Jie
Lu, Jiwen
author_facet He, Haoyuan
Zheng, Yu
Zhou, Jie
Lu, Jiwen
contents Robust watermarking is critical for intellectual property protection, whereas existing methods face a severe vulnerability against regeneration-based AIGC attacks. We identify that existing methods fail because they entangle the watermark with high-frequency cover texture, which is susceptible to being rewritten during generative purification. To address this, we propose WaterVIB, a theoretically grounded framework that reformulates the encoder as an information sieve via the Variational Information Bottleneck. Instead of overfitting to fragile cover details, our approach forces the model to learn a Minimal Sufficient Statistic of the message. This effectively filters out redundant cover nuances prone to generative shifts, retaining only the essential signal invariant to regeneration. We theoretically prove that optimizing this bottleneck is a necessary condition for robustness against distribution-shifting attacks. Extensive experiments demonstrate that WaterVIB significantly outperforms state-of-the-art methods, achieving superior zero-shot resilience against unknown diffusion-based editing.
format Preprint
id arxiv_https___arxiv_org_abs_2602_21508
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle WaterVIB: Learning Minimal Sufficient Watermark Representations via Variational Information Bottleneck
He, Haoyuan
Zheng, Yu
Zhou, Jie
Lu, Jiwen
Machine Learning
Cryptography and Security
Computer Vision and Pattern Recognition
Robust watermarking is critical for intellectual property protection, whereas existing methods face a severe vulnerability against regeneration-based AIGC attacks. We identify that existing methods fail because they entangle the watermark with high-frequency cover texture, which is susceptible to being rewritten during generative purification. To address this, we propose WaterVIB, a theoretically grounded framework that reformulates the encoder as an information sieve via the Variational Information Bottleneck. Instead of overfitting to fragile cover details, our approach forces the model to learn a Minimal Sufficient Statistic of the message. This effectively filters out redundant cover nuances prone to generative shifts, retaining only the essential signal invariant to regeneration. We theoretically prove that optimizing this bottleneck is a necessary condition for robustness against distribution-shifting attacks. Extensive experiments demonstrate that WaterVIB significantly outperforms state-of-the-art methods, achieving superior zero-shot resilience against unknown diffusion-based editing.
title WaterVIB: Learning Minimal Sufficient Watermark Representations via Variational Information Bottleneck
topic Machine Learning
Cryptography and Security
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2602.21508