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

2025-09-23 · Sahil Tyagi, Andrei Cozma, Olivera Kotevska, Feiyi Wang arxiv

Federated Learning (FL) is critical for edge and High Performance Computing (HPC) where data is not centralized and privacy is crucial. We present OmniFed, a modular framework designed around decoupling and clear separation of concerns for configuration, orchestration, communication, and training logic. Its architecture supports configuration-driven prototyping and code-level override-what-you-need customization. We also support different topologies, mixed communication protocols within a single deployment, and popular training algorithms. It also offers optional privacy mechanisms including Differential Privacy (DP), Homomorphic Encryption (HE), and Secure Aggregation (SA), as well as compression strategies. These capabilities are exposed through well-defined extension points, allowing users to customize topology and orchestration, learning logic, and privacy/compression plugins, all while preserving the integrity of the core system. We evaluate multiple models and algorithms to measure various performance metrics. By unifying topology configuration, mixed-protocol communication, and pluggable modules in one stack, OmniFed streamlines FL deployment across heterogeneous environments. Github repository is available at https://github.com/at-aaims/OmniFed.

📄 PDF Abstract BibTeX arXiv:2509.19396

Code (0)

등록된 구현이 없습니다.

Tasks

Federated Learning

Relative Localization System Design for SnailBot: A Modular Self-reconfigurable Robot

2025-12-24 · Shuhan Zhang arxiv

This paper presents the design and implementation of a relative localization system for SnailBot, a modular self reconfigurable robot. The system integrates ArUco marker recognition, optical flow analysis, and IMU data p…

CSIT-Free Model Aggregation for Federated Edge Learning via Reconfigurable Intelligent Surface

2021-02-22 · Hang Liu, Xiaojun Yuan, Ying-Jun Angela Zhang

We study over-the-air model aggregation in federated edge learning (FEEL) systems, where channel state information at the transmitters (CSIT) is assumed to be unavailable. We leverage the reconfigurable intelligent surfa…

image-classification Image Classification

Modular Federated Learning

2022-09-07 · Kuo-Yun Liang, Abhishek Srinivasan, Juan Carlos Andresen

Federated learning is an approach to train machine learning models on the edge of the networks, as close as possible where the data is produced, motivated by the emerging problem of the inability to stream and centrally …

Federated Learning Time Series Time Series Analysis

Modular RAG: Transforming RAG Systems into LEGO-like Reconfigurable Frameworks

2024-07-26 · Yunfan Gao, Yun Xiong, Meng Wang, Haofen Wang

Retrieval-augmented Generation (RAG) has markedly enhanced the capabilities of Large Language Models (LLMs) in tackling knowledge-intensive tasks. The increasing demands of application scenarios have driven the evolution…

RAG Retrieval-augmented Generation Scheduling

Federated Machine Learning for Intelligent IoT via Reconfigurable Intelligent Surface

2020-04-13 · Kai Yang, Yuanming Shi, Yong Zhou, Zhanpeng Yang 외

Intelligent Internet-of-Things (IoT) will be transformative with the advancement of artificial intelligence and high-dimensional data analysis, shifting from "connected things" to "connected intelligence". This shall unl…

BIG-bench Machine Learning Self-Driving Cars