code for aistats 2023
The dataset cifar10_embedding_train.mat and cifar10_embedding_test.mat cannot be uploaded due to their large size. Please contact the author if you need these datasets.
Simply run 'mnist_script.m' will reproduce the results with the MNIST dataset.