SSL-WM

November 21, 2023 ยท View on GitHub

The implementation of _SSL-WM: A Black-Box Watermarking Approach for Encoders Pre-trained by Self-Supervised Learning.

Getting started

Train a clean model

python train_simclr_clean.py

Inject the watermark

python train_simclr.py

Finetune the watermarked model to another task:

python train_downstream.py

Test the accuracy

python test_simclr.py

Test the MAD

python test_mad.py

Contents

  1. 1Getting started
  2. 1.1Train a clean model
  3. 1.2Inject the watermark
  4. 1.3Finetune the watermarked model to another task:
  5. 1.4Test the accuracy
  6. 1.5Test the MAD