I Introduction
The fourth industrial revolution empowered by machine learning algorithms is underway. The popular class of deep learning models and other contemporary machine learning methods are enabling this revolution by providing a significant leap in accuracy and functionality of the underlying model. Several applications are already undergoing serious transformative changes due to the integration of intelligence, including (but not limited to) social networks, autonomous transportation, automated manufacturing, natural language processing, intelligence warfare and smart health
[1, 2, 3, 4].Deep learning is an empirical field in which training a highly accurate model requires: (i) Having access to a massive collection of mostly labeled data that furnishes comprehensive coverage of potential scenarios that might appear in the target application. (ii) Allocating substantial computing resources to finetune the underlying model topology (i.e., type and number of hidden layers), hyperparameters (i.e., learning rate, batch size, etc.), and DL weights in order to obtain the most accurate model. Given the costly process of designing and training a deep neural network, DL models are typically considered to be the intellectual property of the model builder. Protection of the models against IP infringement is particularly important for deep neural networks to preserve the competitive advantage of the DL model owner and ensure the receipt of continuous query requests by clients if the model is deployed in the cloud as a service.
Embedding digital watermarks into deep neural networks is a key enabler for reliable technology transfer. A digital watermark is a type of marker covertly embedded in a signal or IP, including audio, videos, images, or functional designs. Digital watermarks are commonly adopted to identify ownership of the copyright of such a signal or function. Watermarking has been immensely leveraged over the past decade to protect the ownership of multimedia and video content, as well as functional artifacts such as digital integrated circuits [5, 6, 7, 8, 9]. Extension of watermarking techniques to deep learning models, however, is still in its infancy.
DL models can be used in either a whitebox or a blackbox setting. In a whitebox setting, the model parameters are public and shared with a thirdparty. Model sharing is a common approach in the machine learning field (e.g., the Model Zoo by Caffe Developers, and Alexa Skills by Amazon). Note that even though models are voluntarily shared with the public, it is important to protect pertinent IP and preserve the copyright of the original owner. In the blackbox setting, the model details are not publicly shared and the model is only available to execute as a remote blackbox Application Programming Interface (API). Most of the DL APIs deployed in cloud servers fall within the blackbox category.
Authors in [10, 11] propose a watermarking approach for embedding the IP information in the static content of convolutional neural networks (i.e., weight matrices). Although this work provides a significant leap as the first attempt to watermark neural networks, it poses (at least) three limitations as we shall discuss in Section VII: (i) It incurs a bounded watermarking capacity due to the use of static properties of a model (weights) as opposed to using dynamic content (activations). Note that the weights of a neural network are invariable (static) during the execution phase, regardless of the data passing through the model. The activations, however, are dynamic and both data and modeldependent. As such, we argue that using activations (instead of static weights) provides more flexibility for watermarking purposes. (ii) It is not robust against overwriting the original embedded watermark by a thirdparty. (iii) It targets whitebox settings and is inapplicable to blackbox scenarios.
Requirements  Description 

Fidelity  The functionality (e.g., accuracy) of the target neural network shall not be degraded as a result of watermark embedding. 
Reliability  The watermarking methodology shall yield minimal false negatives; the watermarked model shall be effectively detected using the pertinent keys. 
Robustness  The watermarking methodology shall be resilient against model modifications such as compression/pruning, finetuning, and/or watermark overwriting. 
Integrity  The watermarking methodology shall yield minimal false alarms (a.k.a., false positives); the watermarked model should be uniquely identified using the pertinent keys. 
Capacity  The watermarking methodology shall be capable of embedding a large amount of information in the target neural network while satisfying other requirements (e.g., fidelity, reliability, etc.). 
Efficiency  The communication and computational overhead of watermark embedding and extraction/detection shall be negligible. 
Security  The watermark shall leave no tangible footprints in the target neural network; thus, an unauthorized individual cannot detect the presence of a watermark in the model. 
Generalizability  The watermarking methodology shall be applicable in both whitebox and blackbox settings. 
More recent studies in [12, 13] propose 1bit watermarking methodologies that are applicable to blackbox models.^{1}^{1}11bit and 0bit watermarking are used interchangeably in the literature.
These approaches are built upon model boundary modification and the use of random adversarial samples that lie near decision boundaries. Adversarial samples are known to be statistically unstable, meaning that the adversarial samples crafted for a model are not necessarily misclassified by another network
[14, 15]. Therefore, even though the proposed approaches in [12, 13] yield a high watermark detection rate (a.k.a. true positive rate), they are also too sensitive to hyperparameter tuning and usually lead to a high false alarm rate. Note that false ownership proofs based upon watermark extraction, in turn, jeopardize the integrity of the proposed watermarking methodology and render the use of watermarks for IP protection ineffective.This paper proposes DeepSigns, a novel endtoend framework that empowers coherent integration of robust digital watermarks in contemporary deep learning models with no drop in overall prediction accuracy. The embedded watermarks can be triggered by a set of corresponding input keys to remotely detect the existence of the pertinent neural network in a thirdparty DL service. DeepSigns, for the first time, introduces a generic functional watermarking methodology that is applicable to both whitebox and blackbox settings. Unlike prior works that directly embed the watermark information in the static content (weights) of the pertinent model, DeepSigns works by embedding an arbitrary Nbit string into the probability density function (pdf) of the activation sets in various layers of a deep neural network. The proposed methodology is simultaneously data and modeldependent, meaning that the watermark information is embedded in the dynamic content of the DL network and can only be triggered by passing specific input data to the model. Our suggested method leaves no visible impacts on the static properties of the DL model, such as the histogram of the weight matrices.
We provide a comprehensive set of quantitative and qualitative metrics that shall be evaluated to corroborate the effectiveness of current and pending watermarking methodologies for deep neural networks (Section II). We demonstrate the robustness of our proposed framework with respect to stateoftheart removal and transformative attacks, including model compression/pruning, model finetuning, and watermark overwriting. Extensive evaluation across various DL model topologies  including residual networks, convolutional neural networks, and multilayer perceptrons  confirms the applicability of the proposed watermarking framework in different settings without requiring excessive hyperparameter tuning to avoid false alarms and/or accuracy drop. The explicit contributions of this paper are as follows:

Proposing DeepSigns, the first endtoend framework for systematic deep learning IP protection that works in both whitebox and blackbox settings. A novel watermarking methodology is introduced to encode the pdf of the DL model and effectively trace the IP ownership. DeepSigns is significantly more robust against removal and transformation attacks compared to prior works.

Providing a comprehensive set of metrics to assess the performance of watermark embedding methods for DL models. These metrics enable effective quantitative and qualitative comparison of current and pending DL model protection methods that might be proposed in the future.

Devising an application programming interface to facilitate the adoption of DeepSigns watermarking methodology for training various DL models, including convolutional, residual, and fullyconnected networks.

Performing extensive proofofconcept evaluations on various benchmarks. Our evaluations demonstrate DeepSigns’ effectiveness to protect the IP of an arbitrary DL model and establish the ownership of the model builder.
Ii Watermarking Requirements
There are a set of minimal requirements that should be addressed to design a robust digital watermark. Table I details the requirements for an effective watermarking methodology for DL models. In addition to previously suggested requirements in [10, 12], we believe reliability, integrity, and generalizability are three other major factors that need to be considered when designing a practical DL watermarking methodology.
Reliability is important because the embedded watermark should be accurately extracted using the pertinent keys; the model owner is thereby able to detect any misuse of her model with a high probability. Integrity ensures that the IP infringement detection policy yields a minimal number of false alarms, meaning that there is a very low chance of falsely proving the ownership of the model used by a thirdparty. Generalizability is another main factor in developing an effective watermarking methodology. Generalizability is particularly important since the model owner does not know beforehand whether her model will be misused in a blackbox or whitebox setting by a thirdparty. Nevertheless, the model owner should be able to detect IP infringement in both settings. DeepSigns satisfies all the requirements listed in Table I as shown by our experiments in Section VI.
Potential Attack Scenarios. To validate the robustness of a potential DL watermarking approach, one should evaluate the performance of the proposed methodology against (at least) three types of contemporary attacks: (i) Model finetuning. This type of attack involves retraining of the original model to alter the model parameters and find a new local minimum while preserving the accuracy. (ii) Model pruning. Model pruning is a commonly used approach for efficient execution of neural networks, particularly on embedded devices. We consider model pruning as another attack approach that might affect the watermark extraction/detection. (iii) Watermark overwriting. A thirdparty user who is aware of the methodology used to embed the watermark in the model (but not the owner’s private watermark information) may try to embed a new watermark in the DL network and overwrite the original one. The objective of an overwriting attack is to insert an additional watermark in the model and render the original watermark unreadable. A watermarking methodology should be robust against finetuning, pruning, and overwriting for effective IP protection.
Iii Global Flow
Figure 1 demonstrates the highlevel block diagram of the DeepSigns framework. To protect the IP of a particular neural network, the model owner (a.k.a. Alice) first must locally embed the watermark (WM) information into her neural network. Embedding the watermark involves three main steps: (i) Generating a set of Nbit binary random strings to be embedded in the pdf distribution of different layers in the target neural network. (ii) Creating specific input keys to later trigger the corresponding WM strings after watermark embedding. (iii) Training (finetuning) the neural network with particular constraints enforced by the WM information within intermediate activation maps of the target DL model. Details of each step are discussed in Section IV. Note that local model watermarking is a onetime task only performed by the owner before model distribution.
Once the neural network is locally trained by Alice to include the pertinent watermark information, the model is ready to be deployed by a thirdparty DL service provider (a.k.a. Bob). Bob can either leverage Alice’s model as a blackbox API or a whitebox model. To prove the ownership of the model, Alice queries the remote service provider using her specific input keys that she has initially selected to trigger the WM information. She then obtains the corresponding intermediate activations (in the whitebox setting) or the final model prediction (in the blackbox setting). The acquired activations/predictions are then used to extract the embedded watermarks and detect whether Alice’s model is used by Bob within the underlying DL service or not. The details of watermark extraction are outlined in Section V.
Iv Functional Watermarking
Deep learning models possess nonconvex loss surfaces with many local minima that are likely to yield an accuracy (on test data) very close to another approximate model [16, 17]. The DeepSigns framework is built upon the fact that there is not a unique solution for modern nonconvex optimization problems used in deep neural networks. DeepSigns works by iteratively massaging the corresponding pdf of data abstractions to incorporate the desired watermarking information within each layer of the neural network. This watermarking information can later be used to claim ownership of the neural network or detect IP infringement.
In many realworld DL applications, the activation maps obtained in the intermediate (a.k.a. hidden) layers roughly follow a Gaussian distribution
[18, 19, 20]. In this paper, we consider a Gaussian Mixture Model (GMM) as the prior probability to characterize the data distribution at each hidden layer.
^{2}^{2}2We emphasize that our proposed approach is rather generic and is not restricted to the GMM distribution; the GMM distribution can be replaced with any other prior distribution depending on the application. The last layer (a.k.a. output layer) is an exception since the output can be a discrete variable (e.g., class label) in a large category of DL applications. As such, DeepSigns governs the hidden (Section IVA) and output (Section IVB) layers differently.Iva Watermarking Intermediate Layers
To accommodate for our GMM prior distribution assumption, we suggest adding the following term to the conventional crossentropy loss function (
) used for training deep neural networks:(1) 
Here, is a tradeoff hyperparameter that specifies the contribution of the additive loss term. The additive loss function () aims to minimize the entanglement between data features (activations) belonging to different classes while decreasing the innerclass diversity. This loss function, in turn, helps to augment data features so that they approximately fit a GMM distribution. On one hand, a large value of makes the data activations follow a strict GMM distribution, but large values might also impact the final accuracy of the model due to limited contribution of the accuracyspecific crossentropy loss function (). On the other hand, a very small value of is not adequate to make the activations adhere to a GMM distribution. Note that the default distribution of activations may not strictly follow a GMM. We set the value to in all our experiments.
In Equation (1), is the set of model parameters (i.e., weights and biases), is the activation map corresponding to input sample at the layer, is the groundtruth label, and denotes the mean value of the Gaussian distribution at layer that best fits the data abstractions belonging to class . In DeepSigns framework, the watermark information is embedded in the mean value of the pertinent Gaussian mixture distribution. The mean values
and intermediate feature vectors
in Equation 1 are trainable variables that are iteratively learned and finetuned during the training process of the target deep neural network.Watermark embedding. To watermark the target neural network, the model owner (Alice) first needs to generate the designated WM information for each intermediate layer of her model. Algorithm 1 summarizes the process of watermark embedding for intermediate layers. In the following passages, we explicitly discuss each of the steps outlined in Algorithm 1. The model owner shall repeat Steps 1 through 3 for each layer that she wants to eventually watermark and sum up the corresponding loss functions for each layer in Step 4 to train the pertinent DL model.
Choosing one (or more) random indices between and with no replacement: Each index corresponds to one of the Gaussian distributions in the target mixture model that contains a total of Gaussians. For classification tasks, we set the value equal to the number of classes in the target application. The mean values of the selected distributions are then used to carry the watermark information generated in Steps 2 and 3 as discussed below.
Designating an arbitrary binary string to be embedded in the target model: The elements (a.k.a., bits) of the binary string are independently and identically distributed (i.i.d). Henceforth, we refer to this binary string as the vector where is the number of selected distributions (Step 1) to carry the watermarking information, and is a ownerdefined parameter indicating the desired length of the digital watermark embedded at the mean value of each selected Gaussian distribution.
Specifying a random projection matrix (): The projection matrix is used to map the selected centers in Step 1 into the binary vector chosen in Step 2. The transformation is denoted as the following:

(2) 
Here, is the size of the feature space in the pertinent layer, and
denotes the concatenated mean values of the selected distributions. In our experiments, we use a standard normal distribution
to generate the WM projection matrix (). Using i.i.d. samples drawn from a normal distribution ensures that each bit of the binary string is embedded into all the features associated with the selected centers (mean values). The notation in Equation 2is used as a subscript to indicate the deployment of the Sigmoid function. The output of Sigmoid has a value between 0 and 1. Given the random nature of the binary string, we decide to set the threshold in Equation
2 to , which is the expected value of Sigmoid. The Hard_Thresholding function denoted in Equation 2 maps the values in that are greater than to ones and the values less than to zeros. This threshold value can be easily changed in our API if the user decides to change it for their application. A value greater the means that the binary string has a higher probability to include more zeros than ones. This setting, in turn, is useful for users who use biased random number generators.Training the DL model to embed the pertinent watermark information: The process of computing the vector is differentiable. Thereby, for a selected set of projection matrice () and binary strings (
), the selected centers (Gaussian mean values) can be adjusted/trained via backpropagation such that the Hamming distance between the binarized projected centers and the actual WM vectors
is minimized (ideally zero). To do so, one needs to add the following term to the overall loss function for each specific layer of the underlying deep neural network:(3) 
Here, the variable is a hyperparameter that determines the contribution of in the process of training the neural network. All three loss functions (, , and
) are simultaneously used to train/finetuned the underlying neural network. We used Stochastic Gradient Descent (SGD) in all our experiments to optimize the DL model parameters with the explicit constraints outlined in Equations
1 and 3. This optimization aims to find the best distribution of the activation sets by iterative data subspace alignment in order to obtain the highest accuracy while embedding the WM vectors. We set the variable to in all our experiments, unless mentioned otherwise. As shown in Section VI, our method is robust on various benchmarks even though the hyperparameters are not explicitly tuned for each application.IvB Watermarking The Output Layer
Neural network prediction in the very last layer of a DL model needs to closely match the groundtruth data (e.g., training labels in a classification task) in order to have the maximum possible accuracy. As such, instead of directly regularizing the activation set of the output layer, we choose to adjust the tails of the decision boundaries to incorporate a desired statistical bias in the network as a 1bit watermark. We focus, in particular, on classification tasks using deep neural networks. Watermarking the output layer is a postprocessing step that shall be performed after training the model as discussed in Section IVA. Figure 2 illustrates the highlevel blockdiagram of the DeepSigns framework used for watermarking the output layer of the underlying DL model.
The workflow of embedding watermark in the output layer is summarized in Algorithm 2. In the following, we explicitly discuss each of the steps outlined in Algorithm 2.
Learning the pdf distribution of the activations in each intermediate layer as discussed in Section IVA: The acquired probability density function, in turn, gives us an insight into both the regions of latent space that are thoroughly occupied by the training data and the regions that are only covered by the tail of the GMM distribution, which we refer to as rarely explored regions. Figure 3 illustrates a simple example of two clustered Gaussian distribution spreading in a twodimensional subspace.
Generating a set of unique random input samples to be used as the watermarking keys in step 3: Each selected random sample should be passed through the pretrained neural network in order to make sure its latent features lie within the unused regions (Step 1). If the number of training data within a ball of the random sample is fewer than a threshold, we accept that sample as one of the watermark keys. Otherwise, a new random sample is generated to replace the previous sample. A corresponding random groundtruth vector is generated and assigned to each selected input key sample. For instance, in a classification application, each random input is associated with a randomly selected class.
On one hand, it is desirable to have a high watermark detection rate after WM embedding. On the other hand, one needs to ensure a low false positive rate to address the integrity requirement. As such, we start off by setting the initial key size to be larger than the owner’s desired value and generate accordingly. The target model is then finetuned (Step 3) using a mixture of the generated keys and a subset of original training data. After finetuning, only keys that are simultaneously correctly classified by the marked model and incorrectly predicted by the unmarked model are appropriate candidates that satisfy both a high detection rate and a low false positive (Step 4). In our experiments, we set where is the desired key length selected by the model owner (Alice).
Finetuning the pretrained neural network with the selected random watermarks in Step 2: The model shall be retrained such that the neural network has exact predictions (e.g., an accuracy greater than 99%) for selected key samples. In our experiments, we use the same optimizer setting originally used for training the neural network, except that the learning rate is reduced by a factor of 10 to prevent accuracy drop in the prediction of legitimate input data.
Selecting the final input key set to trigger the embedded watermark in the output layer: To do so, we first find out the indices of initial input keys that are correctly classified by the marked model. Next, we identify the indices of input key samples that are not classified correctly by the original DL model before finetuning in Step 3. The common indices between these two sets are proper candidates to be considered as the final key. A random subset of applicable input key samples is then selected based on the required key size () that is defined by the model owner.
It is worth noting that the watermark information embedded in the output layer can be extracted even in settings where the DL model is used as a blackbox API by a thirdparty (Bob). Recently a method called frontier stitching is proposed to perform 1bit watermarking in blackbox settings [12]. Our proposed approach is different in the sense that we use random samples that lie within the tail regions of the probability density function spanned by the model, as opposed to relying on adversarial samples that lie close to the boundaries [21, 22, 14, 15]. Adversarial samples are known to be statistically unstable, meaning that the adversarial samples carefully crafted for a model are not necessarily misclassified by another network. As shown previously in [12], frontier stitching is highly vulnerable to the hyperparameter selection of the watermarking detection policy and may lead to a high false positive if it is not precisely tuned; thus jeopardizing the integrity requirement. As we empirically verify in Section VI, DeepSigns overcomes this integrity concern by selecting random samples within the unused space of the model. This is due to the fact that the unused regions in the space spanned by a model are specific to that model, whereas the decision boundaries for a given task are often highly correlated among various models.
V Watermarking Extraction
For watermark extraction, the model owner (Alice) needs to send a set of queries to the DL service provider (Bob). The queries include the input keys discussed in Sections IVA and IVB. In the case of blackbox usage of a neural network, Alice can only retrieve model predictions for the queried samples, whereas in the whitebox setting, the intermediate activations can also be recovered. In the rest of this section, we explicitly discuss the decision policy for IP infringement detection in both whitebox and blackbox scenarios. Given the high dimensionality of neural networks, the probability of collision for two honest model owners is very unlikely. Two honest owners should have the exact same weight initialization, projection matrix, binary string, and input keys to end up with the same watermark. In case of a malicious user, the misdetection probability is evaluated for the various attacks discussed in Section VI.
Va Decision Policy: Whitebox Setting
To extract watermark information from intermediate (hidden) layers, Alice must follow five main steps. Algorithm 3 outlines the decision policy for the whitebox scenarios.
(I) Submitting queries to the remote DL service provider using the selected input keys. To do so, Alice first collects a subset of the input training data belonging to the selected watermarked classes ( in Algorithm 1). In our experiments, we use a subset of 1% of training data as the input key. (II) Acquiring the activation features corresponding to the input keys. (III) Computing statistical mean value of the activation features obtained by passing the selected input keys in Step I. The acquired mean values are used as an approximation of the Gaussian centers that are supposed to carry the watermark information. (IV) Using the mean values obtained in Step III and her private projection matrix to extract the pertinent binary string following the protocol outlined in Equation 2. (V) Measuring the Bit Error Rate (BER) between the original watermark string and the extracted string from Step IV. Note that in case of a mismatch between Alice and Bob models, a random watermark will be extracted which, in turn, yields a very high BER.
Computation and communication overheads. From Bob’s point of view, the computation cost is equivalent to the cost of one forward pass in the pertinent DL model for each query from Alice. From Alice’s point of view, the computation cost is divided into two terms. The first term is proportional to to compute the statistical mean in Step 3 outlined in the Algorithm 3. Here, denotes the feature space size in the target hidden layer. The second term corresponds to the computation of matrix multiplication in Step 4 of Algorithm 3, which incurs a cost of . The communication cost for Bob is equivalent to the input key length multiplied by the feature size of intermediate activations (), and the communication cost for Alice is the input key size multiplied by the input feature size for each sample.
VB Decision Policy: Blackbox Setting
To verify the presence of the watermark in the output layer, Alice needs to statistically analyze Bob’s responses to a set of input keys. To do so, she must follow four main steps: (I) Submitting queries to the remote DL service provider using the randomly selected input keys () as discussed in Section IVB. (II) Acquiring the output labels corresponding to the input keys. (III) Computing the number of mismatches between the model predictions and Alice’s groundtruth labels. (IV) Thresholding the number of mismatches to derive the final decision. If the number of mismatches is less than a threshold, it means that the model used by Bob possesses a high similarity to the network owned by Alice. Otherwise, the two models are not replicas. When the two models are the exact duplicate of one another, the number of mismatches will be zero and Alice can safely claim the ownership of the neural network used by the thirdparty.
Dataset  Baseline Accuracy  Accuracy of Marked Model  DL Model Type  DL Model Architecture  

MNIST  98.54%  K = 20  N = 4  MLP  784512FC512FC10FC 
98.59%  98.13%  
CIFAR10  78.47%  K = 20  N = 4  CNN  3*32*3232C3(1)32C3(1)MP2(1) 
81.46%  80.7%  64C3(1)64C3(1)MP2(1)512FC10FC  
CIFAR10  91.42%  K = 20  N = 128  WideResNet  Please refer to [23]. 
91.48%  92.02% 
filters applied with a stride of 2,
denotes a maxpooling layer over regions of size
and stride of 1, and is a fullyconnected layer withoutput neurons. ReLU is used as the activation function in all benchmarks.
In realworld settings, the target DL model might be slightly modified by Bob in both malicious or nonmalicious ways. Examples of such modifications are model finetuning, model pruning, or WM overwriting. As such, the threshold used for WM detection should be greater than zero to withstand DL model modifications. The probability of a network (not owned by Alice) to make at least correct decision according to the Alice private keys is as follows:
, 
(4) 
where is the oracle DL model used by Bob,
is a random variable indicating the number of matched predictions of the two models compared against one another,
is the input key length according to Section IVB, and is the number of classes in the pertinent deep learning application.Algorithm 4 summarizes the decision policy for the blackbox scenarios. Throughout our experiments, we use the decision policy for watermark detection, where is defined in Equation 4. The threshold value used in Step 5 of Algorithm 4 determines the tradeoff between a low false positive rate and a high detection rate. As we empirically corroborate in Section VI, DeepSigns satisfies the reliability, integrity, and robustness requirements in various benchmarks and attack scenarios without demanding that the model owner explicitly finetune her decision policy hyperparameters (e.g., the threshold in Equation 4).
Computation and communication overheads. For Bob, the computation cost is equivalent to the cost of one forward pass through the underlying neural network per queried input key. For Alice, the computation cost is the cost of performing a simple counting to measure the number of mismatches between the output labels by Bob and the actual labels owned by Alice. The communication cost for Alice is equivalent to the key length multiplied by the size of the input layer in the target neural network. From Bob’s point of view, the communication cost to send back the corresponding predicted output is the key length multiplied by the output layer size.
Vi Evaluations
We evaluate the performance of the DeepSigns framework on various datasets including MNIST [24] and CIFAR10 [25] with three different neural network architectures. Table II summarizes the neural network topologies used in each benchmark. In Table II, denotes the key size for watermarking the output layer and
is the length of the WM used for watermarking the hidden layers. In all whitebox related experiments, we use the secondtolast layer for watermarking. However, DeepSigns is generic and the extension of watermarking multiple layers is also supported by our framework. In the blockbox scenario, we use the very last layer for watermark embedding/detection. To facilitate watermark embedding and extraction in various DL models, we provide an accompanying TensorFlowbased
[26] API in which model owners can easily define their specific model topology, watermark information, training data, and pertinent hyperparameters including decision policies’ thresholds, WM embedding strength (), and selected layers to carry the WM information.Metrics  Whitebox  Blackbox  

MNIST  CIFAR10CNN  CIFAR10WRN  MNIST  CIFAR10CNN  CIFAR10WRN  
Number of epochs 
50  100  200  50  100  200  50  100  200  50  100  200  50  100  200  50  100  200 
Accuracy  98.21  98.20  98.18  70.11  62.74  59.86  91.79  91.74  91.8  98.57  98.57  98.59  98.61  98.63  98.60  87.65  89.74  88.35 
BER  0  0  0  0  0  0  0  0  0                   
Detection success  1  1  1  1  1  1  1  1  1  1  1  1  1  1  1  1  1  1 
DeepSigns satisfies all the requirements listed in Table I. In the rest of this section, we explicitly evaluate DeepSigns’ performance with respect to each requirement on three common DL benchmarks.
Via Fidelity
The accuracy of the target neural network shall not be degraded after embedding the watermark information. Matching the accuracy of the unmarked model is referred to as fidelity. Table II summarizes the baseline DL model accuracy (Column 2) and the accuracy of marked models (Column 3) after embedding the WM information. As demonstrated, DeepSigns respects the fidelity requirement by simultaneously optimizing for the accuracy of the underlying model (e.g., crossentropy loss function), as well as the additive WMspecific loss functions ( and ) as discussed in Section IV. In some cases (e.g. wideResNet benchmark), we even observe a slight accuracy improvement compared to the baseline. This improvement is mainly due to the fact that the additive loss functions (Equations 1 and 3) and/or exploiting rarely observed regions act as a form of a regularizer during the training phase of the target DL model. Regularization, in turn, helps the model to avoid overfitting by inducing a small amount of noise to the DL model [3].
ViB Reliability and Robustness
We evaluate the robustness of the DeepSigns framework against three contemporary removal attacks as discussed in Section III. The potential attacks include parameter pruning [27, 28, 29], model finetuning [30, 31], and watermark overwriting [10, 32].
Parameter pruning. We use the pruning approach proposed in [27] to compress the neural network. For pruning each layer of a neural network, we first set of the parameters that possess the smallest weight values to zero. The obtained mask is then used to sparsely finetune the model to compensate for the accuracy drop induced by pruning using conventional crossentropy loss function (). Figure 4 illustrates the impact of pruning on watermark extraction/detection in both blackbox and whitebox settings. In the blackbox experiments, DeepSigns can tolerate up to and parameter pruning for the MNIST and CIFAR10 benchmarks, respectively, and up to and in the whitebox related experiments. The WM lengths to perform watermarking in each benchmark are listed in Table II.
As demonstrated in Figure 4, in occasions where pruning the neural network yields a substantial bit error rate (BER) value, we observe that the sparse model suffers from a large accuracy loss compared to the baseline. As such, one cannot remove the embedded watermark in a neural network by excessive pruning of the parameters while attaining a comparable accuracy with the baseline.
Model finetuning. Finetuning is another form of transformation attack that a thirdparty user might use to remove WM information. To perform this type of attack, one needs to retrain the target model using the original training data with the conventional crossentropy loss function (excluding the watermarking specific loss functions). Table III summarizes the impact of finetuning on the watermark detection rate across all three benchmarks.
There is a tradeoff between the model accuracy and the success rate of watermark removal. If a thirdparty tries to finetune the DL model using a high learning rate with the goal of disrupting the underlying pdf of the activation maps and eventually remove WMs, he will face a large degradation in model accuracy. In our experiments, we use the same learning rate as the one in the final stage of DL training to perform the model finetuning attack. As demonstrated in Table III, DeepSigns can successfully detect the watermark information even after finetuning the deep neural network for many epochs. Note that finetuning deep learning models makes the underlying neural network converge to another local minimum that is not necessarily equivalent to the original one in terms of the ultimate prediction accuracy.
Watermark overwriting. Assuming the attacker is aware of the watermarking technique, he may attempt to damage the original watermark by embedding a new WM in the DL model. In practice, the attacker does not have any knowledge about the location of the watermarked layers. However, in our experiments, we consider the worstcase scenario in which the attacker knows where the WM is embedded but does not know the original watermark information. To perform the overwriting attack, the attacker follows the protocol discussed in Section IV to embed a new set of watermark information (using a different projection matrix, binary vector, and input keys). Table IV summarizes the results of watermark overwriting for all three benchmarks in the blackbox setting. As shown, DeepSigns is robust against the overwriting attack and can successfully detect the original embedded WM in the overwritten model. The decision thresholds shown in Table IV for different key lengths are computed based on Equation 4 as discussed in Section VB. A bit error rate of zero is also observed in the whitebox setting for all the three benchmarks after the overwriting attack. This further confirms the reliability and robustness of the DeepSigns’ watermarking approach against malicious attacks.
Average # of mismatches  Decision threshold 


K = 20  K = 30  K = 20  K = 30  
MNIST  8.3  15.4  13  21  1  
CIFAR10CNN  9.2  16.7  13  21  1  
CIFAR10WRN  8.5  10.2  13  21  1 
ViC Integrity
Figure 5 illustrates the results of integrity evaluation in the blackbox setting where unmarked models with the same (models 1 to 3) and different (models 4 to 6) topologies are queried by Alice’s keys. As shown in Figure 5, DeepSigns satisfies the integrity criterion and has no false positives, which means the ownership of unmarked models will not be falsely proved. Note that unlike the blackbox setting, in the whitebox scenario, different topologies can be distinguished by onetoone comparison of the architectures belonging to Alice and Bob. For the unmarked model with the same topology in the whitebox setting, the integrity analysis is equivalent to model finetuning, for which the results are summarized in Table III.
ViD Capacity and Efficiency
The capacity of the whitebox activation watermarking is assessed by embedding binary strings of different lengths in the intermediate layers. As shown in Figure 6, DeepSigns allows up to 64, 128, and 128 bits capacity for MNIST, CIFAR10CNN, and CIFAR10WRN benchmarks, respectively. Note that there is a tradeoff between the capacity and accuracy which can be used by the IP owner (Alice) to embed a larger watermark in her neural network model if desired. For IP protection purposes, capacity is not an impediment criterion as long as the capacity is sufficient to contain the necessary WM information (). Nevertheless, we have included this property in Table I to have a comprehensive list of requirements.
In scenarios where the accuracy might be jeopardized due to excessive regularization of the intermediate activation features (e.g., when using a large watermarking strength for and , a large WM length , or in cases where the DL model is so compact that there are few free variables to carry the WM information), one can mitigate the accuracy drop by expanding each layer to include more free variables. Although the probability of finding a poor local minimum is nonzero for smallsize networks, this probability decreases quickly with the expansion of network size [16, 35], resulting in the accuracy compensation. Note that, in all our experiments, we do not expand the network to meet the baseline accuracy. The layer expansion technique is simply a suggestion for data scientists and DL model designers who are working on developing new architectures that might require a higher WM embedding capacity.
The computation and communication overhead in the DeepSigns framework is a function of the network topology (i.e., the number of parameters/weights in the pertinent DL model), the selected input key length (), and the size of the desired embedded watermark . In Section V, we provide a detailed discussion on the computation and communication overhead of DeepSigns in both whitebox and blackbox settings for watermark extraction. Note that watermark embedding is performed locally by Alice; therefore, there is no communication overhead for WM embedding. In all our experiments, training the DL models with WMspecific loss functions takes the same number of epochs compared to training the unmarked model (with only conventional crossentropy loss) to obtain a certain accuracy. As such, there is no tangible extra computation overhead for WM embedding.
ViE Security
As mentioned in Table I
, the embedding of the watermark should not leave noticeable changes in the probability distribution spanned by the target neural network. DeepSigns satisfies the security requirement by preserving the intrinsic distribution of weights/activations. For instance, Figure
7 illustrates the histogram of the activations in the embedded layer in the marked model and the same layer in the unmarked model for the CIFAR10WRN benchmark.Vii Comparison With PriorArt
Figure 8 provides a highlevel overview of general capabilities of existing DL WM frameworks. DeepSigns satisfies the generalizability criterion and is applicable to both whitebox and blackbox settings. One may speculate that whitebox and blackbox scenarios can be treated equivalently by embedding the watermark information only in the output layer. In other words, one can simply ignore the whitebox related information (intermediate layers) and simply treat the whitebox setting as another blackbox setting. We would like to emphasize that the output layer can only potentially contain a 1bit watermark; whereas an Nbit () can be used for watermarking the pdf distribution of the intermediate layers, thus providing a higher capacity for IP protection. In this paper, by whitebox scenario, we refer to using the whole capacity of a DL model for watermark embedding, including both hidden and output layers.
Unlike prior works, DeepSigns uses the dynamic statistical properties of DL models for watermark embedding. DeepSigns incorporates the watermark information within the pdf distribution of the activation maps. Note that even though the weights of a DL model are static during the inference phase, activation maps are dynamic features that are both dependent on the input keys and the DL model parameters/weights. As we illustrate in Table V, our data and modelaware watermarking approach is significantly more robust against overwriting and pruning attacks compared with the priorart whitebox methods. None of the previous blackbox deep learning WM frameworks consider overwriting attacks in their experiments. As such, no quantitative comparison is feasible in this context. Nevertheless, DeepSigns performance against overwriting attacks for blackbox settings is summarized in Table IV.
In the rest of this section, we explicitly compare DeepSigns performance against three stateoftheart DL watermarking frameworks existing in the literature.
Viia Whitebox Setting
To the best of our knowledge, [10, 11] are the only existing works that target watermarking hidden layers. These works use the weights of the convolution layers for the purpose of watermarking, as opposed to the activation sets used by DeepSigns. As shown in [10], watermarking weights is not robust against overwriting attacks. Table V provides a sidebyside robustness comparison between our approach and these prior works for different dimensionality ratio of the attacker’s WM vector to the target weights/activations. As demonstrated, DeepSigns’ dynamic data and modelaware approach is significantly more robust compared to priorart [10, 11]. As for its robustness against pruning attack, our approach is tolerant of higher pruning rates. As an example consider the CIFAR10WRN benchmark, in which DeepSigns is robust up to pruning rate, whereas the works in [10, 11] are only robust up to pruning rate.
N to M Ratio  Bit Error Rate (BER)  

Uchida et.al  DeepSigns  
1  0.309  0 
2  0.41  0 
3  0.511  0 
4  0.527  0 
ViiB Blackbox Setting
To the best of our knowledge, there are two prior works that target watermarking the output layer for blackbox scenarios [13, 12]. Even though the works by [13, 12] provide a high WM detection rate (reliability), they do not address the integrity requirement, meaning that these approaches can lead to a high false positive rate in practice. For instance, the work in [13] uses accuracy on the test set as the decision policy to detect WM information. It is well known that there is no unique solution to highdimensional machine learning problems [16, 2, 3]. In other words, there are various models with even different topologies that yield approximately the same test accuracy for a particular data application. Besides high false positive rate, another drawback of using test accuracy for WM detection is the high overhead of communication and computation [13]; therefore, their watermarking approach incurs low efficiency. DeepSigns uses a small input key size () to trigger the WM information, whereas a typical test set in DL problems can be two to three orders of magnitude larger.
Viii Conclusion
In this paper, we introduce DeepSigns, the first endtoend framework that enables reliable and robust integration of watermark information in deep neural networks for IP protection. DeepSigns is applicable to both whitebox and blackbox model disclosure settings. It works by embedding the WM information in the probability density distribution of the activation sets corresponding to different layers of a neural network. Unlike prior DL watermarking frameworks, DeepSigns is robust against overwriting attacks and satisfies the integrity criteria by minimizing the number of potential false alarms raised by the framework. We provide a comprehensive list of requirements that empowers quantitative and qualitative assessment of current and pending DL watermarking approaches. Extensive evaluations using three contemporary benchmarks corroborate the practicability and effectiveness of the DeepSigns framework in the face of malicious attacks, including parameter pruning/compression, model finetuning, and watermark overwriting. We devise an accompanying TensorFlowbased API that can be used by data scientists and engineers for watermarking of different neural networks. Our API provides support for various DL model topologies, including (but not limited to) multilayer perceptrons, convolution neural networks, and wide residual models.
References
 [1] Y. LeCun, Y. Bengio, and G. Hinton, “Deep learning,” nature, vol. 521, no. 7553, 2015.
 [2] L. Deng, D. Yu et al., “Deep learning: methods and applications,” Foundations and Trends® in Signal Processing, vol. 7, no. 3–4, 2014.
 [3] I. Goodfellow, Y. Bengio, A. Courville, and Y. Bengio, Deep learning. MIT press Cambridge, 2016, vol. 1.
 [4] M. Ribeiro, K. Grolinger, and M. A. Capretz, “Mlaas: Machine learning as a service,” in IEEE 14th International Conference on Machine Learning and Applications (ICMLA), 2015.
 [5] B. Furht and D. Kirovski, Multimedia security handbook. CRC press, 2004.
 [6] F. Hartung and M. Kutter, “Multimedia watermarking techniques,” Proceedings of the IEEE, vol. 87, no. 7, 1999.
 [7] G. Qu and M. Potkonjak, Intellectual property protection in VLSI designs: theory and practice. Springer Science & Business Media, 2007.
 [8] I. J. Cox, J. Kilian, F. T. Leighton, and T. Shamoon, “Secure spread spectrum watermarking for multimedia,” IEEE transactions on image processing, vol. 6, no. 12, 1997.
 [9] C.S. Lu, Multimedia Security: Steganography and Digital Watermarking Techniques for Protection of Intellectual Property: Steganography and Digital Watermarking Techniques for Protection of Intellectual Property. Igi Global, 2004.
 [10] Y. Uchida, Y. Nagai, S. Sakazawa, and S. Satoh, “Embedding watermarks into deep neural networks,” in Proceedings of the ACM on International Conference on Multimedia Retrieval, 2017.
 [11] Y. Nagai, Y. Uchida, S. Sakazawa, and S. Satoh, “Digital watermarking for deep neural networks,” International Journal of Multimedia Information Retrieval, vol. 7, no. 1, 2018.
 [12] E. L. Merrer, P. Perez, and G. Trédan, “Adversarial frontier stitching for remote neural network watermarking,” arXiv preprint arXiv:1711.01894, 2017.
 [13] Y. Adi, C. Baum, M. Cisse, B. Pinkas, and J. Keshet, “Turning your weakness into a strength: Watermarking deep neural networks by backdooring,” Usenix Security Symposium, 2018.
 [14] K. Grosse, P. Manoharan, N. Papernot, M. Backes, and P. McDaniel, “On the (statistical) detection of adversarial examples,” arXiv preprint arXiv:1702.06280, 2017.
 [15] B. D. Rouhani, M. Samragh, T. Javidi, and F. Koushanfar, “Safe machine learning and defeating adversarial attacks,” IEEE Security and Privacy (S&P) Magazine, 2018.
 [16] A. Choromanska, M. Henaff, M. Mathieu, G. B. Arous, and Y. LeCun, “The loss surfaces of multilayer networks,” in Artificial Intelligence and Statistics, 2015.
 [17] B. D. Rouhani, A. Mirhoseini, and F. Koushanfar, “Deep3: Leveraging three levels of parallelism for efficient deep learning,” in Proceedings of ACM 54th Annual Design Automation Conference (DAC), 2017.
 [18] S. Ioffe and C. Szegedy, “Batch normalization: Accelerating deep network training by reducing internal covariate shift,” arXiv preprint arXiv:1502.03167, 2015.
 [19] A. B. Patel, T. Nguyen, and R. G. Baraniuk, “A probabilistic theory of deep learning,” arXiv preprint arXiv:1504.00641, 2015.
 [20] D. Lin, S. Talathi, and S. Annapureddy, “Fixed point quantization of deep convolutional networks,” in International Conference on Machine Learning (ICML), 2016.
 [21] I. J. Goodfellow, J. Shlens, and C. Szegedy, “Explaining and harnessing adversarial examples,” arXiv preprint arXiv:1412.6572, 2014.
 [22] N. Papernot, P. McDaniel, I. Goodfellow, S. Jha, Z. B. Celik, and A. Swami, “Practical blackbox attacks against machine learning,” in Proceedings of ACM on Asia Conference on Computer and Communications Security, 2017.
 [23] S. Zagoruyko and N. Komodakis, “Wide residual networks,” arXiv preprint arXiv:1605.07146, 2016.

[24]
Y. LeCun, C. Cortes, and C. J. Burges, “The mnist database of handwritten digits,” 1998.
 [25] A. Krizhevsky and G. Hinton, “Learning multiple layers of features from tiny images,” 2009.
 [26] M. Abadi, P. Barham, J. Chen, Z. Chen, A. Davis, J. Dean, M. Devin, S. Ghemawat, G. Irving, M. Isard et al., “Tensorflow: A system for largescale machine learning.” in Usenix Symposium on Operating Systems Design and Implementation (OSDI), vol. 16, 2016.
 [27] S. Han, J. Pool, J. Tran, and W. Dally, “Learning both weights and connections for efficient neural network,” in Advances in Neural Information Processing Systems (NIPS), 2015.
 [28] S. Han, H. Mao, and W. J. Dally, “Deep compression: Compressing deep neural networks with pruning, trained quantization and huffman coding,” arXiv preprint arXiv:1510.00149, 2015.
 [29] B. D. Rouhani, A. Mirhoseini, and F. Koushanfar, “Delight: Adding energy dimension to deep neural networks,” in Proceedings of the International Symposium on Low Power Electronics and Design (ISLPED). ACM, 2016.
 [30] K. Simonyan and A. Zisserman, “Very deep convolutional networks for largescale image recognition,” arXiv preprint arXiv:1409.1556, 2014.
 [31] N. Tajbakhsh, J. Y. Shin, S. R. Gurudu, R. T. Hurst, C. B. Kendall, M. B. Gotway, and J. Liang, “Convolutional neural networks for medical image analysis: Full training or fine tuning?” IEEE transactions on medical imaging, vol. 35, no. 5, 2016.
 [32] N. F. Johnson, Z. Duric, and S. Jajodia, Information Hiding: Steganography and WatermarkingAttacks and Countermeasures: Steganography and Watermarking: Attacks and Countermeasures. Springer Science & Business Media, 2001, vol. 1.
 [33] J. T. Springenberg, A. Dosovitskiy, T. Brox, and M. Riedmiller, “Striving for simplicity: The all convolutional net,” arXiv preprint arXiv:1412.6806, 2014.

[34]
M. Liang and X. Hu, “Recurrent convolutional neural network for object
recognition,” in
Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR)
, 2015.  [35] Y. N. Dauphin, R. Pascanu, C. Gulcehre, K. Cho, S. Ganguli, and Y. Bengio, “Identifying and attacking the saddle point problem in highdimensional nonconvex optimization,” in Advances in Neural Information Processing Systems (NIPS), 2014.
Comments
There are no comments yet.