plot_dme_l1.py

#!/usr/bin/env python3
#
# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.

import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns

epsilons = np.arange(0.5, 5.1, 0.5)
mean_cldp, std_cldp = np.zeros(epsilons.shape), np.zeros(epsilons.shape)
mean_skellam, std_skellam = np.zeros(epsilons.shape), np.zeros(epsilons.shape)
mean_mvu, std_mvu = np.zeros(epsilons.shape), np.zeros(epsilons.shape)
mean_baseline, std_baseline = np.zeros(epsilons.shape), np.zeros(epsilons.shape)

for i in range(len(epsilons)):
    checkpoint = np.load("dme_results/dme_multi_l1_d_128_samples_10000_eps_%.2f_skellam_16_100.00_mvu_3_9.npz" % epsilons[i])
    mean_cldp[i] = checkpoint["squared_error_cldp"].mean()
    std_cldp[i] = checkpoint["squared_error_cldp"].std()
    mean_skellam[i] = checkpoint["squared_error_skellam"].mean()
    std_skellam[i] = checkpoint["squared_error_skellam"].std()
    mean_mvu[i] = checkpoint["squared_error_mvu"].mean()
    std_mvu[i] = checkpoint["squared_error_mvu"].std()
    mean_baseline[i] = checkpoint["squared_error_baseline"].mean()
    std_baseline[i] = checkpoint["squared_error_baseline"].std()
    
plt.figure(figsize=(8,6))
with sns.color_palette("deep"):
    plt.errorbar(epsilons, mean_cldp, std_cldp, label="CLDP", linewidth=3)
    plt.errorbar(epsilons, mean_skellam, std_skellam, label="Skellam ($\\delta > 0$)", linewidth=3)
    plt.errorbar(epsilons, mean_mvu, std_mvu, label="MVU", linewidth=3)
plt.errorbar(epsilons, mean_baseline, std_baseline, label="Laplace", color='k', linestyle='--', linewidth=3)

plt.yscale('log')
plt.xticks(fontsize=20)
plt.xlabel("$\\epsilon$", fontsize=24)
plt.yticks(fontsize=20)
plt.ylabel("MSE", fontsize=24)
plt.legend(fontsize=20, loc="upper right")
plt.grid('on')
plt.title("$L_1, d=128$", fontsize=24)
plt.savefig("figures/dme_l1.pdf", bbox_inches="tight")