Федеративное обучение на основе блокчейна: стимулирование обмена данными и наказание нечестного поведения

Основная информация

• ID статьи: 2307.10492
• Название: Blockchain-Based Federated Learning: Incentivizing Data Sharing and Penalizing Dishonest Behavior
• Авторы: Амир Джаберзаде, Аджай Кумар Шреста, Файджан Ахмад Хан, Мохаммед Афаан Шейх, Бхаргав Дейв, Джейсон Генг
• Учреждения: Bayes Solutions (США) и Vancouver Island University (Канада)
• Классификация: cs.LG (Машинное обучение)
• Дата публикации: июль 2023 г.
• Ссылка на статью: https://arxiv.org/abs/2307.10492

Аннотация

В данной работе предлагается комплексная система для решения проблем безопасности и доверия при обмене данными путём интеграции федеративного обучения с блокчейном, смарт-контрактами и IPFS (InterPlanetary File System). Система обеспечивает механизмы стимулирования, контроль доступа и механизмы наказания для содействия безопасному и взаимному обмену данными. Экспериментальные результаты демонстрируют, что модель достигает точности более 95% при обучении CNN на наборе данных MNIST, одновременно обеспечивая безопасность и справедливость процесса обмена данными. Платформа поддерживает одновременное обучение нескольких рабочих узлов и сохраняет конфиденциальность и безопасность данных посредством децентрализованной архитектуры и технологии блокчейна.

Предпосылки и мотивация исследования

1. Основные проблемы, которые необходимо решить

Данное исследование направлено на решение следующих ключевых проблем:

• Проблема информационных силосов: сложность обмена и интеграции данных между различными организациями
• Конфиденциальность и безопасность: риск утечки конфиденциальной информации при централизованном хранении и обмене данными
• Отсутствие доверия: недостаток надёжных механизмов доверия между участниками
• Недостаточные стимулы: отсутствие эффективных механизмов стимулирования для содействия обмену высококачественными данными
• Вредоносное поведение: необходимость предотвращения и наказания участников, предоставляющих низкокачественные или вредоносные данные

2. Значимость проблемы

С растущей важностью обмена данными в сотрудничестве и инновациях становится критически важным обеспечить управление и обмен данными безопасным и надёжным способом. Традиционные подходы к управлению данными сталкиваются с множественными проблемами, включая согласованность данных, совместимость, конфиденциальность, безопасность, контроль доступа, собственность и вознаграждение за обмен.

3. Ограничения существующих методов

• Традиционное федеративное обучение: зависит от центрального сервера, что создаёт риск единой точки отказа и возможность компрометации центрального сервера, угрожающей конфиденциальности всей системы
• Централизованное хранилище: увеличивает риск утечки данных и вызывает проблемы с собственностью и контролем над данными
• Существующие варианты FedAvg: хотя предложены различные улучшения (методы с импульсом, адаптивные скорости обучения и т.д.), они всё ещё недостаточны в защите конфиденциальности, механизмах стимулирования и предотвращении вредоносного поведения

4. Мотивация исследования

Данная работа направлена на построение децентрализованной системы федеративного обучения путём интеграции блокчейна, смарт-контрактов, IPFS и криптографических методов, одновременно решая множественные проблемы защиты конфиденциальности, механизмов стимулирования, контроля доступа и наказания за вредоносное поведение.

Основные вклады

1. Предложена комплексная децентрализованная система федеративного обучения: интегрирует доверие данных, IPFS, блокчейн и смарт-контракты в федеративное обучение для реализации безопасного и взаимного обмена данными
2. Разработаны механизмы стимулирования и наказания на основе залога: посредством смарт-контрактов требуют от участников предоставления залога, применяют экономические штрафы к участникам, предоставляющим низкокачественные или вредоносные данные, и распределяют штрафы честным участникам
3. Реализована двойная схема шифрования: сочетает симметричное шифрование (AES) и асимметричное шифрование (RSA) для защиты конфиденциальности моделей и данных с дополнительными вычислительными затратами всего 2%
4. Построено децентрализованное хранилище моделей на основе IPFS: избегает рисков централизованного хранилища и поддерживает одноранговый обмен моделями
5. Подтверждена эффективность системы: достигнута точность более 95% на наборе данных MNIST, что доказывает осуществимость и эффективность децентрализованной архитектуры

Подробное описание методов

Определение задачи

Задача данного исследования заключается в построении децентрализованной платформы федеративного обучения, позволяющей нескольким участникам (рабочим узлам) совместно обучать глобальную модель машинного обучения без обмена исходными данными. Система должна удовлетворять следующим требованиям:

• Входные данные: локальные наборы данных рабочих узлов, начальная модель, количество раундов обучения, общая сумма вознаграждения
• Выходные данные: обученная глобальная модель
• Ограничения: защита конфиденциальности данных, предотвращение вредоносного поведения, справедливое распределение вознаграждений, децентрализованная архитектура

Архитектура модели

1. Проектирование общей архитектуры

Система включает две категории участников:

• Инициатор (Requester): инициирует задачу федеративного обучения, развёртывает смарт-контракт, устанавливает параметры обучения (количество раундов N, общая сумма вознаграждения D), загружает начальную модель в IPFS
• Рабочие узлы (Workers): участвуют в задаче обучения, обучают модель на локальных данных, оценивают модели других узлов, получают вознаграждение в зависимости от производительности

Основные компоненты:

• Блокчейн и смарт-контракты: координируют задачи FL, управляют информацией об участниках, распределяют вознаграждения и штрафы
• Хранилище IPFS: децентрализованное хранилище обучаемых моделей
• Модуль шифрования: защищает конфиденциальность моделей и данных

2. Функциональность и реализация каждого модуля

a) Доверие данных, контроль доступа и механизмы стимулирования

• Участники должны зарегистрироваться и предоставить залог (collateral deposit)
• Залог служит средством экономического наказания, предотвращая предоставление участниками низкокачественных или вводящих в заблуждение данных
• При нечестном поведении участника залог конфискуется и распределяется между честными участниками
• Смарт-контракт обновляет и распределяет общую компенсацию в зависимости от вклада участников
• Обеспечивает, что каждый участник может зарегистрироваться только один раз, а компенсация распределяется только при положительной общей сумме

b) Хранилище IPFS

• Использует InterPlanetary File System как одноранговую распределённую файловую систему
• Модели хранятся на устройствах пользователей без необходимости централизованного хранилища
• Снижает риск утечки данных и повышает контроль над данными и собственностью

c) Конфиденциальность и защита приватности

• Применяет гибридную схему шифрования:
  • Использует симметричный ключ (AES) для шифрования фактических данных/моделей
  • Использует асимметричный ключ (RSA) для шифрования симметричного ключа
  • Обеспечивает, что только получатели с соответствующим приватным ключом могут расшифровать данные
• Использует библиотеку cryptography на Python для реализации функций шифрования
• Реализует методы получения, расшифровки и загрузки зашифрованного состояния модели
• Оптимизирует использование памяти: ведёт список хешей загруженных моделей, очищает список при достижении указанного количества

d) Функциональность смарт-контрактов Смарт-контракт включает следующие ключевые функции:

• initializeTask: инициатор инициализирует задачу FL, устанавливает URI модели и количество раундов, требует внесения залога
• startTask: инициатор запускает задачу, статус переходит в "запущена"
• joinTask: рабочий узел присоединяется к задаче, регистрируется и получает URI модели
• submitScore: рабочий узел отправляет оценку модели после оценки в каждом раунде
• removeWorker: рабочий узел выходит из задачи
• nextRound: инициатор переходит к следующему раунду
• getSubmissions: инициатор получает все отправки текущего раунда
• submitRoundTopK: получает лучших K рабочих узлов по производительности
• distributeRewards: распределяет вознаграждение лучшим рабочим узлам (лучшие K получают половину вознаграждения, остальное распределяется в меньших долях)

3. Рабочий процесс

1. Фаза инициализации:
   • Инициатор развёртывает смарт-контракт, устанавливает количество раундов обучения N и общее вознаграждение D
   • Инициатор загружает начальную модель в IPFS
   • Рабочие узлы присоединяются к задаче через смарт-контракт
2. Фаза обучения (всего N раундов):
   • В начале каждого раунда рабочие узлы получают из IPFS все обучаемые модели других рабочих узлов
   • Рабочие узлы оценивают эти модели на локальных данных и вычисляют оценки
   • Оценки отправляются в смарт-контракт
   • Смарт-контракт агрегирует оценки и определяет лучших K рабочих узлов по производительности
   • Распределяет вознаграждение в зависимости от производительности
   • Рабочие узлы обучают модель на локальных данных
   • Обученная модель загружается в IPFS
   • Повторяется N раундов
3. Завершающая фаза:
   • После завершения обучения инициатор получает из IPFS финальную глобальную модель
   • Вызывает функцию смарт-контракта для закрытия задачи

4. Методы агрегирования/усреднения

• Рабочие узлы получают из хранилища IPFS свою модель и модели других рабочих узлов
• Используют функцию усреднения для сложения всех моделей и деления на количество рабочих узлов, предоставивших модели
• Получают усреднённую модель для повышения точности
• Этот метод избегает большого объёма коммуникации между центральным сервером и клиентами в централизованном FedAvg, снижая перегрузку канала и риск атак на конфиденциальность

Технические инновации

1. Отличия от базовых методов

• Децентрализованная архитектура: не зависит от центрального сервера, избегает единой точки отказа и атак на конфиденциальность
• Экономические механизмы стимулирования: стимулирует честное поведение и наказывает вредоносное поведение через систему залогов и вознаграждений
• Двойное шифрование: сочетает AES и RSA, контролируя затраты на уровне 2%
• Блокчейн + IPFS: использует неизменяемость блокчейна и децентрализованное хранилище IPFS

2. Анализ обоснованности проектирования

• Механизм залога: эффективно ограничивает поведение участников экономическими средствами, более действенен, чем чисто технические методы
• Многомерная оценка производительности: учитывает точность, согласованность, точность и полноту, комплексно оценивая вклад рабочих узлов
• Гибридное шифрование: симметричное шифрование эффективно (подходит для больших данных), асимметричное шифрование безопасно (подходит для обмена ключами), сочетание обеспечивает баланс эффективности и безопасности
• Хранилище IPFS: естественно соответствует децентрализованной архитектуре, механизм адресации по содержимому гарантирует целостность данных

Экспериментальная установка

Набор данных

• Название набора данных: MNIST (рукописные цифры)
• Масштаб данных:
  • Обучающий набор: 60 000 изображений
  • Тестовый набор: 10 000 изображений
• Задача: классификация рукописных цифр 0-9
• Распределение данных: обучающий набор равномерно распределяется между рабочими узлами в начале обучения
• Оценка: каждый рабочий узел использует тестовый набор для оценки и выставления оценок

Метрики оценки

• Точность (Accuracy): процент правильной классификации
• Точность (Precision): 0,973
• Полнота (Recall): 0,97
• Время сходимости: время, необходимое модели для достижения целевой точности

Методы сравнения

• Шифрование vs без шифрования: сравнение влияния двойного шифрования на время сходимости
• Различное количество рабочих узлов: 3 рабочих узла vs 5 рабочих узлов

Детали реализации

• Модель: простая сверточная нейронная сеть (CNN), N слоёв
• Фреймворк: PyTorch
• Блокчейн: блокчейн Ethereum
• Среда моделирования: Ganache (локальная тестовая среда блокчейна Ethereum)
• Оборудование: процессор Xeon, 8 ядер
• Способ обучения: реализация децентрализованной системы клиент-сервер на локальной машине с последовательным выполнением (также возможно параллельное)
• Максимальное количество раундов: 90 эпох

Результаты экспериментов

Основные результаты

1. Анализ производительности

• Точность: достигает более 95% точности в течение 90 эпох
• Точность (Precision): 0,973
• Полнота (Recall): 0,97
• Общее время обучения (3 рабочих узла): 6525,46 секунд
• Время сходимости каждого рабочего узла: примерно 36 минут
• Вывод: время сходимости сопоставимо с децентрализованной системой федеративного обучения

2. Анализ затрат на шифрование

• Дополнительные затраты двойного шифрования:
  • Всего для 3 рабочих узлов: 2 минуты 34 секунды
  • На каждый рабочий узел: 51 секунда
  • Доля затрат на коммуникацию: всего 2% от времени, необходимого для сходимости
• Вывод: затраты на процессы двойного шифрования и расшифровки, а также протокол безопасной передачи пар ключей минимальны и приемлемы при обеспечении той же точности

3. Сравнение по количеству рабочих узлов

• 3 рабочих узла:
  • Более стабильная картина точности
  • Причина: каждый рабочий узел имеет больше обучающих данных
• 5 рабочих узлов:
  • Достигает приемлемой точности при аналогичном количестве эпох
  • Может ускорить процесс обучения и расширить масштаб обучения
  • Снижает требования к вычислительной мощности каждого рабочего узла, позволяя использовать устройства с ограниченными ресурсами
• Вывод:
  • Увеличение количества рабочих узлов не оказывает негативного влияния на сходимость модели
  • Количество рабочих узлов должно выбираться в соответствии с пропорцией обучающего набора данных
  • В реальных сценариях увеличение обучающего набора данных может повысить стабильность модели с несколькими рабочими узлами

Абляционные эксперименты

Статья в основном проводит абляционные эксперименты по затратам на шифрование:

• Сравнивает время сходимости с двойным шифрованием и без шифрования
• Доказывает, что механизм шифрования добавляет только 2% затрат, подтверждая эффективность проектирования

Анализ случаев

Статья демонстрирует эволюцию точности в процессе обучения:

• Все три рабочих узла имеют низкую начальную точность
• В течение первого раунда (3 эпохи) точность значительно повышается
• Впоследствии рабочие узлы обучаются последовательно, точность постепенно повышается
• В итоге все рабочие узлы достигают точности более 95%

Экспериментальные выводы

1. Осуществимость децентрализованной архитектуры: эксперименты доказывают, что децентрализованное федеративное обучение может достичь производительности, сопоставимой с централизованными методами
2. Контролируемые затраты на шифрование: двойная схема шифрования добавляет только 2% временных затрат, доказывая хороший баланс между безопасностью и эффективностью
3. Масштабируемость: увеличение количества рабочих узлов не ухудшает производительность модели, а может ускорить обучение и снизить требования к вычислениям на отдельном узле
4. Важность распределения данных: количество рабочих узлов должно соответствовать масштабу обучающего набора данных для сохранения стабильности обучения

Связанные работы

1. Область федеративного обучения

• FedAvg и его варианты:
  • FedAvg 2: базовый алгоритм федеративного усреднения
  • Методы с импульсом 6: для локального обучения клиентов
  • Адаптивный FedAvg 7: с адаптивной скоростью обучения
  • Ленивые и квантованные градиенты 8: снижение коммуникации
  • Схемы типа Ньютона 9: FedDANE
• Децентрализованный градиентный спуск:
  • DGD и его варианты 10-13
  • DSGD 14: децентрализованный стохастический градиентный спуск
  • Асинхронный DSGD 15
  • Квантованный DSGD 16

2. Блокчейн + федеративное обучение

• Умное здравоохранение 18: архитектура защиты конфиденциальности с использованием блокчейна и федеративного обучения
• Интернет вещей в автомобилях 19: решение федеративного обучения на основе блокчейна с механизмом стимулирования на основе репутации

3. Преимущества данной работы по сравнению с связанными работами

• Комплексная система: интегрирует механизмы стимулирования, наказания, контроля доступа и защиты конфиденциальности
• Эффективное шифрование: двойная схема шифрования с затратами всего 2%
• Практическая ориентация: проверена на реальных наборах данных
• Экономические стимулы: инновационно вводит механизм залога, ограничивая поведение участников с экономической точки зрения

Заключение и обсуждение

Основные выводы

1. Предложенная децентрализованная архитектура федеративного обучения успешно интегрирует блокчейн, смарт-контракты и IPFS, реализуя безопасное и эффективное обучение глобальной модели
2. Экспериментальные результаты показывают, что система достигает точности более 95% в течение 90 эпох, а время сходимости сопоставимо с централизованной системой федеративного обучения
3. Двойная схема шифрования добавляет минимальные затраты в 2%, доказывая хороший баланс между безопасностью и эффективностью
4. Метод эффективно решает множественные проблемы управления и обмена данными, устанавливая доверие между заинтересованными сторонами, содействуя взаимному обмену данными и предотвращая поведение, которое может угрожать безопасности и точности данных

Ограничения

1. Масштаб экспериментов: тесты проводились только на локальной машине с последовательным выполнением, не проверены в крупномасштабной распределённой среде
2. Единственный набор данных: использован только набор данных MNIST, отсутствуют проверки на более сложных наборах данных и задачах
3. Затраты на блокчейн: не проведён подробный анализ затрат на транзакции блокчейна и проблем масштабируемости
4. Обнаружение вредоносного поведения: механизм залога зависит от точной оценки производительности, но не рассмотрены более сложные типы вредоносного поведения (например, отравление моделей)
5. Выбор рабочих узлов: не обсуждается динамический выбор и управление рабочими узлами, а также обработка динамического присоединения и выхода узлов
6. Вызовы практического развёртывания: не рассмотрены сетевые задержки, гетерогенность узлов и другие проблемы при реальном развёртывании

Направления будущих исследований

Статья явно указывает следующие направления будущих исследований:

• Исследование масштабируемости: изучение масштабируемости в реальных сценариях
• Проверка осуществимости: проверка осуществимости модели в практических приложениях

Другие потенциальные направления:

• Тестирование системы на более сложных наборах данных и задачах
• Исследование более продвинутых механизмов обнаружения и защиты от вредоносного поведения
• Оптимизация затрат на транзакции блокчейна и пропускной способности
• Разработка механизма динамического управления рабочими узлами
• Исследование производительности при гетерогенных устройствах и сетевых условиях

Глубокая оценка

Преимущества

1. Инновационность метода

• Интеграция нескольких технологий: инновационно интегрирует блокчейн, смарт-контракты, IPFS и криптографические методы в федеративное обучение, формируя полную экосистему
• Экономические механизмы стимулирования: система залогов и вознаграждений ограничивает поведение участников с экономической точки зрения, являясь эффективным дополнением к техническим методам
• Гибридная схема шифрования: сочетание AES+RSA обеспечивает баланс между эффективностью и безопасностью

2. Достаточность экспериментов

• Предоставляет многомерную оценку: точность, точность, полноту и другие метрики
• Сравнивает производительность с шифрованием и без шифрования
• Тестирует влияние различного количества рабочих узлов
• Предоставляет конкретные данные по времени и производительности

3. Убедительность результатов

• Точность более 95% доказывает эффективность метода
• Затраты на шифрование 2% доказывают практичность решения
• Время сходимости, сопоставимое с существующими методами, доказывает конкурентоспособность

4. Ясность изложения

• Архитектура проектирования ясна, процесс описан подробно
• Предоставлены диаграммы архитектуры системы и графики результатов
• Полное описание функций смарт-контрактов

Недостатки

1. Ограничения метода

• Недостаточное обнаружение вредоносного поведения: в основном полагается на оценку производительности, недостаточная защита от продвинутых атак (отравление моделей, атаки на градиенты и т.д.)
• Установка залога: не обсуждается, как определить разумный размер залога
• Отказоустойчивость Византии: не ясно указано, сколько вредоносных узлов может допустить система

2. Недостатки экспериментальной установки

• Слишком простой набор данных: MNIST классический, но простой набор данных, сложно отражает сложные сценарии
• Отсутствие тестирования в реальной среде: тесты только на локальной машине с последовательным выполнением, не проверены в реальной распределённой среде
• Отсутствие сравнительных экспериментов: не проведено прямое сравнение с другими решениями блокчейн+федеративное обучение
• Анализ затрат на блокчейн отсутствует: не предоставлены ключевые показатели, такие как затраты на газ и задержка транзакций

3. Недостатки анализа

• Отсутствие анализа масштабируемости: не обсуждается производительность при значительном увеличении количества рабочих узлов
• Влияние сетевых условий: не рассмотрена производительность при различных сетевых условиях
• Обработка гетерогенности: не обсуждается влияние гетерогенности устройств и данных
• Недостаточный теоретический анализ: отсутствуют доказательства сходимости и теоретические гарантии

Влияние

1. Вклад в область

• Комплексное решение: предоставляет полную систему, интегрирующую несколько технологий, служит ориентиром для последующих исследований
• Практическая ориентация: уделяет внимание механизмам стимулирования и наказания за вредоносное поведение, более соответствует практическим потребностям приложений
• Пионерская работа: проводит полезное исследование в области блокчейн+федеративное обучение

2. Практическая ценность

• Защита конфиденциальности: применима к чувствительным к конфиденциальности областям, таким как здравоохранение и финансы
• Децентрализация: подходит для сценариев, где не доверяют центральному серверу
• Механизмы стимулирования: может содействовать обмену данными и сотрудничеству
• Но практическое развёртывание всё ещё сталкивается с проблемами: затраты на блокчейн, масштабируемость и другие проблемы требуют дальнейшего решения

3. Воспроизводимость

• Преимущества:
  • Подробное описание архитектуры системы и рабочего процесса
  • Предоставлено описание функций смарт-контрактов
  • Указан используемый технологический стек (PyTorch, Ethereum, Ganache и т.д.)
• Недостатки:
  • Код не открыт
  • Отсутствует подробная установка гиперпараметров
  • Не предоставлен полный код смарт-контрактов

Применимые сценарии

1. Высокоприменимые сценарии

• Сотрудничество в медицинских данных: несколько больниц совместно обучают модели, защищая конфиденциальность пациентов
• Финансовый контроль рисков: несколько банков совместно используют признаки данных, не раскрывая исходные данные
• Федеративная система рекомендаций: несколько платформ сотрудничают для улучшения алгоритмов рекомендаций
• Граничные вычисления: устройства IoT совместно обучают модели

2. Условия применимости

• Участники не доверяют друг другу и не желают использовать центральный сервер
• Требования к конфиденциальности данных высоки, централизованное хранилище невозможно
• Требуются механизмы стимулирования для содействия обмену данными
• Можно принять определённые затраты на транзакции блокчейна

3. Неприменимые сценарии

• Приложения с экстремальными требованиями к реальному времени (транзакции блокчейна имеют задержку)
• Сценарии с экстремально большим количеством участников (ограничения масштабируемости)
• Устройства с экстремально ограниченными вычислительными ресурсами (шифрование и операции блокчейна требуют затрат)
• Сценарии с уже существующим надёжным центральным сервером (необходимость децентрализации снижается)

Библиография

Статья цитирует 21 важный источник, ключевые источники включают:

1. Delacroix & Lawrence (2019): базовый подход к доверию данных
2. McMahan et al. (2017): оригинальная статья алгоритма FedAvg
3. Sun et al. (2022): последние достижения в децентрализованном федеративном усреднении
4. Singh et al. (2022): применение блокчейна и федеративного обучения в медицинском Интернете вещей
5. Wang et al. (2022): защита конфиденциальности федеративного обучения на основе блокчейна в Интернете вещей автомобилей
6. Shrestha et al. (2020, 2021): платформа блокчейна для обмена пользовательскими данными и проектирование механизмов стимулирования

Резюме

Данная работа предлагает инновационную систему федеративного обучения на основе блокчейна, решая проблемы доверия, стимулирования и конфиденциальности в децентрализованном машинном обучении путём интеграции нескольких технологий (блокчейн, смарт-контракты, IPFS, гибридное шифрование). Эксперименты подтверждают эффективность метода, но практическое развёртывание, масштабируемость и защита от сложных атак требуют дальнейших исследований. Данная работа предоставляет ценные идеи для защиты конфиденциальности при совместном машинном обучении и имеет потенциал применения в чувствительных к конфиденциальности областях, таких как здравоохранение и финансы.