PLRV-O: Advancing Differentially Private Deep Learning via Privacy Loss Random Variable Optimization

## Контекст В статье рассматривается одна из самых распространенных проблем в области машинного обучения — обеспечение конфиденциальности данных при обучении моделей глубокого обучения. Данная проблема становится актуальна при обработке конфиденциальных данных, таких как личные данные или медицинские данные. Основным методом обеспечения конфиденциальности является Differentially Private Stochastic Gradient Descent (DP-SGD), который использует механизм шума для защиты информации. Однако существуют ряд ограничений в существующих подходах. Например, стандартные механизмы шума, такие как гауссовский и лапласианский, ограничены в своей гибкости, так как их параметры не позволяют оптимизировать по отдельности эффект защиты и качество модели. Это приводит к проблемам при оптимизации требуемого уровня конфиденциальности и максимального качества модели, особенно в условиях разных задач и ограничений, таких как время обучения и размер батча. ## Метод Для решения этой проблемы предлагается новый подход, основанный на оптимизации Privacy Loss Random Variable (PLRV). Ранее, параметры шума в DP-SGD были ограничены двумя факторами: значением шума и уровнем конфиденциальности (эпсилон). В новом подходе предлагается расширить поиск пространства параметров шума, чтобы включить в него дополнительные параметры, такие как размер модели и количество итераций обучения. Это позволяет оптимизировать качество модели и уровень конфиденциальности независимо друг от друга. Архитектура фреймворка PLRV-O включает в себя многоуровневую оптимизацию, позволяющую учесть задачи с различными ограничениями, такими как размер батча и стратегии выбора данных. ## Результаты На практике проведены эксперименты на двух датасетах: CIFAR-10 и SST-2. На CIFAR-10 был использован модель ViT, а на SST-2 — RoBERTa-large. Результаты показали, что новый подход PLRV-O значительно улучшает качество модели в сравнении с традиционными методами, такими как Gaussian Noise. Например, fine-tuned ViT на CIFAR-10 достиг 94.03% точности при epsilon примерно 0.5, в то время как Gaussian Noise дал только 83.93%. Аналогичные результаты были получены и для SST-2, где RoBERTa-large показал 92.20% точности при epsilon 0.2, в то время как Gaussian Noise дал лишь 50.25%. Эти результаты указывают на значительную улучшение качества моделей при применении PLRV-O. ## Значимость Предложенный подход имеет широкие применения в области конфиденциального машинного обучения. Он позволяет улучшить качество моделей в условиях жестких ограничений конфиденциальности. Благодаря гибкости PLRV-O, модели могут быть оптимизированы для различных задач, включая обучение с учетом размера модели