上海大学 | 可迁移的图编码器水印(IEEE WIFS 2024)

文摘   2024-11-29 09:06   北京  



图神经网络在社交网络、推荐系统、分子生物等领域有广泛的应用。作为图神经网络领域的研究热点,自监督预训练图编码器通过自监督任务的预训练,能够在无标注数据的情况下,从图数据中自动学习高效的节点、边和图表示,并迁移到下游任务中。然而,这种迁移性也为攻击者提供了便利,攻击者可窃取图编码器并构建自己的下游应用,为图编码器的版权保护带来挑战。

该成果设计了一种用于保护自监督预训练图编码器的黑盒水印技术。在水印嵌入过程中,该技术在常规图中注入密钥节点以构造触发集,通过训练使待保护的图编码器对触发集的输出在嵌入空间上聚集成一个独特且紧密的“簇”。在提取水印时,版权拥有者可根据下游任务以密钥节点注入的方式构造验证集,并观察下游任务中待验证模型对验证集的预测结果,若预测相似度大于预先指定的阈值,则水印提取成功,否则,水印提取失败。

实验结果表明,通过该技术嵌入的水印可被有效地自适应迁移到节点分类、链接预测、社区检测等不同下游任务中,并且不会影响图模型的原有性能,所嵌入的水印同时具有一定的鲁棒性。

论文信息

论文已发表在会议《IEEE International Workshop on Information Forensics and Security》(主会口头报告),作者来自上海大学,依次是赵翔宇、吴汉舟(通讯作者)和张新鹏。

X. Zhao, H. Wu, and X. Zhang. Transferable watermarking to self-supervised pre-trained graph encoders by trigger embeddings, IEEE International Workshop on Information Forensics and Security, to appear, 2024.

欢迎有志向的同学来读研究生:hanzhou@shu.edu.cn


供稿:赵翔宇、吴汉舟



隐者联盟
本公众号主要推介多媒体、人工智能、信息安全等方面的最新研究进展,愿与同行携手,共同推动科学研究向前发展。
 最新文章