Constrained Bayesian Optimization for Hierarchical Federated Learning in IoT Networks for Plant Disease Classification
- URL: http://arxiv.org/abs/2609.06830v1
- Date: Sun, 06 Sep 2026 20:56:40 GMT
- Title: Constrained Bayesian Optimization for Hierarchical Federated Learning in IoT Networks for Plant Disease Classification
- Abstract summary: The deployment of Hierarchical Federated Learning (HFL) in resource-constrained Internet of Things (IoT) environments requires careful configuration to balance predictive performance with energy consumption and execution time.<n>This paper presents a constrained Bayesian Optimization framework for the efficient configuration of HFL deployments.<n>A weighted objective function captures user-defined trade-offs among energy consumption, execution time, and predictive performance, while explicit constraints ensure compliance with deployment-specific resource and accuracy requirements.
- Score: 3.157511963455073
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: The deployment of Hierarchical Federated Learning (HFL) in resource-constrained Internet of Things (IoT) environments requires careful configuration to balance predictive performance with energy consumption and execution time. This challenge is particularly relevant to smart agriculture, where distributed IoT devices can support automated plant disease classification while operating under limited computational and communication resources. This paper presents a constrained Bayesian Optimization framework for the efficient configuration of HFL deployments. The proposed approach jointly explores the deep learning backbone architecture, aggregation strategy, and number of communication rounds, while the federation size is determined according to the spatial coverage requirements of the agricultural deployment. A weighted objective function captures user-defined trade-offs among energy consumption, execution time, and predictive performance, while explicit constraints ensure compliance with deployment-specific resource and accuracy requirements. The framework is evaluated on an IoT-based plant disease classification task considering multiple deep learning architectures, federated aggregation strategies, and communication-round settings. Experimental results across 30 independent optimization runs show that the proposed approach explores only 11.11% of the search space, while consistently identifying solutions within 1% of the exhaustive-search optimum, with a mean optimality gap of only 0.056%.
Related papers
- Reinforcement Learning-Guided Evolutionary Policy Optimization for Preference-Adjustable Heterogeneous Agile Earth Observation Satellite Scheduling [50.176885352114674]
Heterogeneous agile Earth observation satellite (AEOS) scheduling requires task selection, satellite assignment, and observation sequencing.<n>This paper proposes an evolutionary policy optimization framework for heterogeneous AEOS scheduling with preference-adjustable weighted objectives.
arXiv Detail & Related papers (2026-08-25T12:16:28Z) - Performance and Energy Trade-Off Analysis of Hierarchical Federated Learning for Plant Disease Classification [3.265862388280526]
We present a design-space exploration of hierarchical federated learning architectures for plant disease classification.<n>We focus on the trade-offs between predictive performance and energy efficiency.<n>We introduce a power- and energy-aware optimization framework that enables the systematic evaluation and selection of model-aggregator configurations.
arXiv Detail & Related papers (2026-04-28T13:21:18Z) - DeFRiS: Silo-Cooperative IoT Applications Scheduling via Decentralized Federated Reinforcement Learning [62.347535250646196]
This paper proposes DeFRiS, a Decentralized Federated Reinforcement Learning framework for robust and scalable Silo-cooperative IoT application scheduling.<n>DeFRiS integrates three innovations: (i) an action-space-agnostic policy utilizing candidate resource scoring to enable seamless knowledge transfer across heterogeneous silos; (ii) a silo-optimized local learning mechanism combining Generalized Advantage Estimation with clipped policy updates to resolve sparse delayed reward challenges; and (iii) a Dual-Track Non-IID robust decentralized aggregation protocol leveraging gradient fingerprints for similarity-aware knowledge transfer and anomaly detection.
arXiv Detail & Related papers (2026-03-16T02:02:38Z) - FedACT: Concurrent Federated Intelligence across Heterogeneous Data Sources [12.329109810274586]
Real-world applications increasingly require multiple machine learning tasks simultaneously training their models across a shared pool of devices.<n>We introduce em FedACT, a novel resource-aware device scheduling approach designed to efficiently schedule heterogeneous devices across multiple concurrent FL jobs.<n>We show that em FedACT reduces the average JCT by up to 8.3(times) and improves model accuracy by up to 44.5%, compared to the state-of-the-art baselines.
arXiv Detail & Related papers (2026-03-11T23:52:17Z) - Device Association and Resource Allocation for Hierarchical Split Federated Learning in Space-Air-Ground Integrated Network [12.484890461669828]
6G facilitates deployment of Federated Learning (FL) in the Space-Air-Ground Integrated Network (SAGIN)<n>This paper proposes a Hierarchical Split Federated Learning framework and derives its upper bound of loss function.<n>We decompose the original problem into several subproblems, where an iterative optimization algorithm for device association and resource allocation is proposed.
arXiv Detail & Related papers (2026-01-20T10:24:10Z) - Small-Scale-Fading-Aware Resource Allocation in Wireless Federated Learning [27.931985523249352]
This paper proposes a small-scale-fading-aware resource allocation strategy using a multi-agent reinforcement learning (MARL) framework.<n>We establish a one-step convergence bound of the FL algorithm and formulate the resource allocation problem as a decentralized partially observable Markov decision process.<n>In our framework, each client serves as an agent that dynamically determines spectrum and power allocations within each coherence time slot.
arXiv Detail & Related papers (2025-05-06T13:41:59Z) - Cluster-Aware Multi-Round Update for Wireless Federated Learning in Heterogeneous Environments [25.405210975577834]
This paper proposes a clustering strategy that leverages prior knowledge similarity to group devices with similar data and communication characteristics.<n>A novel Cluster- Aware Multi-round Update (CAMU) strategy is proposed, which treats clusters as the basic units and adjusts the local update frequency based on the clustered contribution threshold.
arXiv Detail & Related papers (2025-05-06T02:48:48Z) - Analysis and Optimization of Wireless Federated Learning with Data
Heterogeneity [72.85248553787538]
This paper focuses on performance analysis and optimization for wireless FL, considering data heterogeneity, combined with wireless resource allocation.
We formulate the loss function minimization problem, under constraints on long-term energy consumption and latency, and jointly optimize client scheduling, resource allocation, and the number of local training epochs (CRE)
Experiments on real-world datasets demonstrate that the proposed algorithm outperforms other benchmarks in terms of the learning accuracy and energy consumption.
arXiv Detail & Related papers (2023-08-04T04:18:01Z) - Dynamic Scheduling for Federated Edge Learning with Streaming Data [56.91063444859008]
We consider a Federated Edge Learning (FEEL) system where training data are randomly generated over time at a set of distributed edge devices with long-term energy constraints.
Due to limited communication resources and latency requirements, only a subset of devices is scheduled for participating in the local training process in every iteration.
arXiv Detail & Related papers (2023-05-02T07:41:16Z) - Multi-Resource Allocation for On-Device Distributed Federated Learning
Systems [79.02994855744848]
This work poses a distributed multi-resource allocation scheme for minimizing the weighted sum of latency and energy consumption in the on-device distributed federated learning (FL) system.
Each mobile device in the system engages the model training process within the specified area and allocates its computation and communication resources for deriving and uploading parameters, respectively.
arXiv Detail & Related papers (2022-11-01T14:16:05Z) - Federated Learning for Energy-limited Wireless Networks: A Partial Model
Aggregation Approach [79.59560136273917]
limited communication resources, bandwidth and energy, and data heterogeneity across devices are main bottlenecks for federated learning (FL)
We first devise a novel FL framework with partial model aggregation (PMA)
The proposed PMA-FL improves 2.72% and 11.6% accuracy on two typical heterogeneous datasets.
arXiv Detail & Related papers (2022-04-20T19:09:52Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.