CPC G06T 1/0028 (2013.01) [G06N 3/08 (2013.01); G06T 1/005 (2013.01); G06T 5/50 (2013.01); G06T 2201/0065 (2013.01); G06T 2207/20084 (2013.01); G06T 2207/20221 (2013.01)] | 18 Claims |
1. A method implemented by a computing device, the method comprising:
obtaining weight information of a target neural network model;
obtaining target watermark information; and
using the target watermark information and the weight information of the target neural network model to train an embedded neural network model, and updating the weight information of the target neural network model according to target watermark embedded data provided by the embedded neural network model to obtain the target neural network model embedded with the target watermark information, wherein using the target watermark information and the weight information of the target neural network model to train the embedded neural network model comprises:
obtaining random noise information;
using the weight information of the target neural network model as an instance of a first watermark training set, and using the target watermark information as a label of the first watermark training set;
using weight information of a reference neural network model as an instance of a second watermark training set, and using the random noise information as a label of the second watermark training set, the reference neural network model being a neural network model without the target watermark information embedded; and
using the first watermark training set and the second watermark training set as a new training set, and updating the weight information of the embedded neural network model according to a model loss function of the embedded neural network model until the model loss function converges.
|