OmniFed: A Modular Framework for Configurable Federated Learning from Edge to HPC

## Контекст Federated Learning (FL) является ключевым подходом для обработки данных в средах, где они распределены по многочисленным устройствам, не предполагая централизованного хранения. Это относится к области edge computing и High Performance Computing (HPC). Одним из основных вызовов FL является обеспечение конфигурируемости и поддержки различных архитектур, в том числе в средах с неоднородными коммуникационными протоколами и требованиями к приватности. Существующие решения часто либо ограничиваются узкой областью применения, либо требуют значительных настроек для адаптации к конкретным средам. Эти признаки требуют разработки универсальной, конфигурируемой и модульной платформы для FL. ## Метод OmniFed представляет собой модульную архитектуру с четкой разделением ответственностей для конфигурирования, оркестрации, обмена данными и логики обучения. Она поддерживает гибкую настройку топологий, смешанные протоколы связи в одной среде и популярные тренировочные алгоритмы. Рамки для расширения позволяют пользователям вносить изменения в топологию, логику обучения, а также использовать механизмы приватности, такие как Differential Privacy (DP), Homomorphic Encryption (HE) и Secure Aggregation (SA), а также сжатия данных. Эта модульность достигается благодаря предоставлению прозрачных точек расширения, которые не нарушают центральную логику фреймворка. ## Результаты OmniFed протестирован с различными моделями и алгоритмами, измеряя производительность, точность и эффективность. Множество экспериментов проводилось в средах с разными уровнями ресурсов, включая edge-устройства и HPC-системы. Результаты показали, что OmniFed эффективно адаптируется к разным условиям работы, обеспечивая конфигурируемость и быструю настройку под конкретные задачи. Например, тесты с множественными протоколами связи и сжатием данных показали значительную уменьшение времени обучения и улучшение точности моделей. ## Значимость OmniFed предоставляет широкий спектр приложений, включая обучение моделей на небольших устройствах, а также на больших вычислительных платформах. Его модульность и поддержка разных топологий и протоколов связи делают его универсальным решением для широкого круга задач FL. Благодаря плагинам для приватности и сжатия, OmniFed обеспечивает высокую гибкость и соответствие критическим требованиям к безопасности и производительности. Это делает платформу незаменимым инструментом для исследователей и разработчиков FL в различных сферах. ## Выводы OmniFed успешно решает проблему конфигурируемости и модульности в FL, предлагая гибкую архитектуру для различных сред и задач. Дальнейшими направлениями исследований будут расширение модулей для персональных устрой