Artificial neural network‐driven federated learning for heart stroke prediction in healthcare 4.0 underlying 5G

Abstract

SummaryIn recent years, smart healthcare, artificial intelligence (AI)‐aided diagnostics, and automated surgical robots are just a few of the innovations that have emerged and gained popularity with the advent of Healthcare 4.0. Such technologies are powered by machine learning (ML) and deep learning (DL), which are preferable for disease diagnosis, identifying patterns, prescribing treatments, and forecasting diseases like stroke prediction, cancer prediction and so forth. Nevertheless, much data is needed for AI, ML, and DL‐based systems to train effectively and provide the desired outcomes. Further, it raises concerns about data privacy, security, communication overhead, regulatory compliance and so forth. Federated learning (FL) is a technology that protects data security and privacy by limiting data sharing and utilizing model information of distributed systems to enhance performance. However, existing approaches are traditionally verified on pre‐established datasets that fail to capture real‐life applicability. Therefore, this study proposes an AI‐enabled stroke prediction architecture consisting of FL based on the artificial neural network (ANN) model using data from actual stroke cases. This architecture can be implemented on healthcare‐based wearable devices (WD) for real‐time use as it is effective, precise, and computationally affordable. In order to continuously enhance the performance of the global model, the proposed FL‐based architecture aggregates the optimizer weights of many clients using a fifth‐generation (5G) communication channel. Then, the performance of the proposed FL‐based architecture is studied based on multiple parameters such as accuracy, precision, recall, bit error rate, and spectral noise. It outperforms the traditional approaches regarding accuracy, which is 5% to 10% higher.