SHIELD: A Secure Heuristic Integrated Environment for Load Distribution in Rural-AI

Abstract

The increasing adoption of edge computing in rural areas is leading to a substantial rise in data generation, necessitating the need for development of advanced load balancing algorithms. This is particularly important in applications that utilise existing, though limited, computational and data communication infrastructures. Furthermore, rural communities have growing concerns regarding the privacy, security, and ownership of the data produced within their agricultural fields. Load distribution in rural edge devices can enhance agricultural practices by improving resource usage, decision-making, and addressing network connectivity challenges. Managing resource utilisation in this way also improves economic investments made in managing and deploying edge devices in rural environments. In this work, we propose SHIELD, a security-aware load balancing framework, primarily designed for edge-based systems in rural areas. For handling environments with limited connectivity, SHIELD efficiently manages tasks and computational resources by categorising them into restricted, public and private, shared respectively. It also allocates tasks considering key performance factors such as completion time, resource utilisation, failure rate, and security. The framework is evaluated on a weed detection scenario in precision agriculture, using three federated learning (FL) variants (local model training, global model aggregation, and model prediction) with the ResNet-50 model trained on the DeepWeeds image classification dataset. The proposed framework also integrates encryption and task replication techniques for data confidentiality, integrity, and availability. Experimental results show that SHIELD demonstrates an average of 23% (using Parsl), 29% (using OpenWhisk) improvement in failure rate and 18 s (Parsl), 13 s (OpenWhisk) average improvement in makespan compared to other task allocation approaches, such as secure variants of random, round robin, and least loaded.