SUSS Checkpoints

Download model weights and copy the corresponding weight vectors for each SUSS variant. Refer to the paper for detailed information on training setup of different varients.

SUSSBase

Baseline model trained on ImageNet with self-supervised augmentations.

w = [8.3633eāˆ’06, 4.1081eāˆ’08, 6.3725e-05, 6.0119e-06]

SUSSPieApp-RH

Finetuned on PieAPP MOS dataset, with dataset-specific distortions and human preference data for ranking loss and weight alignment.

w = [4.268e-06, 4.028e-05, 1.153e-04, 2.717e-08]

SUSSPIPAL-RH

Finetuned on PIPAL dataset, with dataset-specific distortions and human preference data for ranking loss and weight alignment.

w = [4.2111e-07, 4.1700e-07, 1.0702e-05, 8.4151e-08]