splitDyn: Federated Split Neural Network for Distributed Edge AI Applications

Split learning (SL) is a popular distributed machine learning (ML) method used to enable ML. It divides a neural network based model into subnetworks. Then, it separately trains the subnetworks on distributed parties (e.g., client and server). In distributed ML, data are generated and collected on the client-side. In contrast, the collected data are processed using an application deployed on the server side. However, when applied in practice using Internet of things systems and clients, numerous obstacles occur because of limited configuration and resources. Dividing neural networks in the SL is the biggest problem and an open question in numerous studies. This study introduces splitDyn, which is a new dynamic SL solution to solve the aforementioned problems. This method provides a solution for eliminating their inherent drawbacks. The main idea is to apply a Round-Robin schedule to select the client for the training process. Then, the next idea is to use the Hungarian optimization algorithm to assign a layer to a client and enhance the accuracy. The proposed method reasonably achieved better accuracy and reduced processing time than the other learning models. Furthermore, it applies the incident datasets to predict the incident event and in edge computing for edge artificial intelligence (AI) applications.