开云体育(中国)官方网站如东说念主脸、LOGO 等-Kaiyun·体育(全站)登陆入口官方网站登录入口
给 AI 生成的作品汲水印,让 AIGC 图像可溯源,一经成为行业共鸣。
问题是,传统水印身手频频把图像当成一个举座处分,全局镶嵌、水印索要一锅端,存在不少"短板":
比如,图像局部区域被变调,就可能导致全局索要失败,也无法定位水印处所具体区域。
又比如,无法只保护某个区域,如东说念主脸、LOGO 等。
针对这个问题,当今,来自南洋理工大学和新加坡 A*STAR 前沿东说念主工智能商榷中心等机构的商榷东说念主员,建议了一种全新的局部鲁棒图像水印身手—— MaskMark。
该身手不仅在多个任务中全面超越 Meta 出品的 SOTA 模子 WAM,况且磨练老本只消它的 1/15。
具体而言,MaskMark 相沿:
多水印镶嵌
可精确定位变调区域
无邪索要局部水印
自适应相沿 32/64/128 比特
中枢念念路:让模子"看得见"水印在那儿
商榷东说念主员引入了一种掩码机制,磨练时告诉模子"水印藏在这里",教它学会精确地镶嵌和索要。
他们给 MaskMark 想象了两个版块:
MaskMark-D(解码掩码)
水印全图镶嵌,但能定位水印位置,相沿局部索要。
即使部分图像被变调,也能奏凯索要水印。
合乎用于举座图像保护、版权声明和实际考据。
MaskMark-ED(编码 + 解码掩码)
水印只镶嵌图像的特定区域(比如东说念主脸或 LOGO)。
对小限制舛误更鲁棒,索要恶果更好。
合乎局部敏锐实际保护、躲闪感知场景等。
其中中枢时刻经由,是磨练和推理的双重优化。
MaskMark 的端到端磨练经由主要包括四个阶段:
掩码生成(Mask Generation):
从四种预界说类型(全掩码、矩形掩码、不规章掩码、分割掩码)中连忙选拔或生成一个掩码 M。
水印镶嵌(Watermark Embedding):
编码器将水印比特镶嵌原始图像(关于 MaskMark-ED 可选拔性地镶嵌掩码 M 来教养镶嵌位置),生成水印图像。此过程愚弄轻量级 CNN 处分水印比特,并谋划 U-Net 结构及 JND 模块优化视觉恶果。
水印掩码操作(Watermark Masking):
使用掩码 M 会通和,生成,即仅在掩码区域保留水印。随后对施加快即失真获取,
并再次用 M 剪辑出仅含水印信号的区域。
水印索要(Watermark Extraction):
解码器从中展望掩码,并从中索要水印比特。解码器包含挑升用于掩码展望的 U2-Net 和用于水印索要的 U-Net 及 CNN 模块。
MaskMark 推理时,解码器领先愚弄定位模块识别含水印区域,将非水印区域置零以减少侵略,然后从保留区域归附水印,这关于小区域水印索要尤为重大。
在多项不同的任务上均进展出 SOTA 的性能
商榷东说念主员在多项不同任务上考据了 MaskMark 的性能。
领先,MaskMark 能结束高索要精度。
在全局水印任务中,即使在高视觉保真度(PSNR > 39.5, SSIM > 0.98)下,MaskMark-D 和 MaskMark-ED 仍能保捏近乎 100% 的比特许确率,尤其在各样值度量和几何失真舛误下进展优异,显贵优于现存基线模子。
在局部水印任务中,实验标明,当水印信号散布在图像的不同大小区域时,MaskMark 仍能保捏近乎 100% 的索要准确率,显贵优于其他全局身手,并特出现时开首进的局部水印模子 WAM。尤其在小面积镶嵌场景中,MaskMark-ED 展现出愈加出色的进展。
其次,MaskMark 能结束精确水印定位。
在不同水印区域比例和失真条目下,其定位性能(以 IoU 预计)均优于 EditGuard 和 WAM 等身手。MaskMark-ED 在小区域定位上更具上风。
MaskMark 还具备多水印镶嵌才智。
尽管并非为此挑升磨练,MaskMark-ED 在单个图像中镶嵌多达 5 个不同水印时,依然保捏强劲的索要和定位性能,优于 WAM。
在高效性和膨大性方面,磨练着力上,MaskMark 仅需在单个 A6000 GPU 上磨练约 20 小时,筹画着力(TFLOPs 预计)比 WAM 提升 15 倍。
在音信长度方面,MaskMark 可简短膨大至不同比特长度(如 32、64、128 位),并保捏较高性能,而 WAM 则受限于 32 位。
另外,MaskMark 还相沿快速微调 *。
通过浅易调度失真层或进行极少微调(如针对 VAE 自适应舛误,仅需 20k 磨练步数),MaskMark 即可适应不同鲁棒性需乞降新兴威迫。
论文相接:
https://arxiv.org/abs/2504.12739
代码相接:
https://github.com/hurunyi/maskmark
一键三连「点赞」「转发」「注重心」
宽宥在批驳区留住你的方针!
— 完 —
� � 点亮星标 � �
科技前沿进展逐日见开云体育(中国)官方网站