Model Barrier: A Compact Un-Transferable Isolation Domain for Model Intellectual Property Protection
Code release for "Model Barrier: A Compact Un-Transferable Isolation Domain for Model Intellectual Property Protection" (CVPR 2023)
Model Barrier: A Compact Un-Transferable Isolation Domain for Model Intellectual Property Protection (CVPR 2023)
We propose a Compact Un-Transferable Isolation Domain (CUTI-domain), which acts as a barrier to block illegal transfers from authorized to unauthorized domains, to protect the intellectual property (IP) of AI models.
The code is implemented with CUDA 11.4, Python 3.8.5 and Pytorch 1.8.0.
MNIST dataset can be found here.
USPS dataset can be found here.
SVHN dataset can be found here.
MNIST-M dataset can be found here.
CIFAR-10 dataset can be found here.
CIFAR-10 dataset can be found here.
Office-Home dataset can be found here.
VisDA 2017 dataset can be found here.
Target-Specified CUTI-Domain
python train_ts_dight.py
Ownership Verification
python train_owner_dight.py
Target-free CUTI-Domain
python train_tf_dight.py
Applicability Authorization
python train_author_dight.py
If you find this code useful for your research, please cite our paper:
@article{wang2023model,
title={Model Barrier: A Compact Un-Transferable Isolation Domain for Model Intellectual Property Protection},
author={Wang, Lianyu and Wang, Meng and Zhang, Daoqiang and Fu, Huazhu},
journal={arXiv preprint arXiv:2303.11078},
year={2023}
}
Some codes are adapted from NTL and SWIN-Transformer. We thank them for their excellent projects.
If you have any problem about our code, feel free to contact
or describe your problem in Issues.