A Federated Learning Platform as a Service for Advancing Stroke Management in European Clinical Centers

Diogo Reis Santos,Albert Sund Aillet,Antonio Boiano,Usevalad Milasheuski,Lorenzo Giusti,Marco Di Gennaro,Sanaz Kianoush,Luca Barbieri,Monica Nicoli,Michele Carminati,Alessandro E. C. Redondi,Stefano Savazzi,Luigi Serio

2024-10-02

Abstract:The rapid evolution of artificial intelligence (AI) technologies holds transformative potential for the healthcare sector. In critical situations requiring immediate decision-making, healthcare professionals can leverage machine learning (ML) algorithms to prioritize and optimize treatment options, thereby reducing costs and improving patient outcomes. However, the sensitive nature of healthcare data presents significant challenges in terms of privacy and data ownership, hindering data availability and the development of robust algorithms. Federated Learning (FL) addresses these challenges by enabling collaborative training of ML models without the exchange of local data. This paper introduces a novel FL platform designed to support the configuration, monitoring, and management of FL processes. This platform operates on Platform-as-a-Service (PaaS) principles and utilizes the Message Queuing Telemetry Transport (MQTT) publish-subscribe protocol. Considering the production readiness and data sensitivity inherent in clinical environments, we emphasize the security of the proposed FL architecture, addressing potential threats and proposing mitigation strategies to enhance the platform's trustworthiness. The platform has been successfully tested in various operational environments using a publicly available dataset, highlighting its benefits and confirming its efficacy.

Computers and Society,Distributed, Parallel, and Cluster Computing,Machine Learning

What problem does this paper attempt to address?

The paper attempts to address the issues of data privacy and security in advancing stroke management at European clinical centers. Specifically, the authors propose a Platform-as-a-Service (PaaS) system based on Federated Learning (FL) to tackle key issues in the following ways:

1. **Data Privacy and Security**: In the medical field, the sensitivity of patient data makes data sharing very difficult. Traditional centralized machine learning methods require data to be aggregated on a central server for training, which not only increases the risk of data breaches but may also violate data protection regulations (such as the EU's General Data Protection Regulation, GDPR). Federated Learning addresses this issue by allowing each medical institution to train models locally without exchanging raw data.

2. **Model Performance and Reliability**: Through Federated Learning, multiple medical institutions can collaboratively train a global model, thereby improving the model's generalization ability and prediction accuracy. This is particularly important for stroke management, as stroke is a complex disease that requires multifaceted data support to make accurate diagnoses and treatment recommendations.

3. **Specific Challenges in Clinical Applications**: Stroke management involves multiple key Clinical Endpoints (CEPs), including the clinical response to acute reperfusion therapy, the probability of early discharge, and the risk of long-term adverse outcomes. The proposed platform aims to provide reliable predictive models for these clinical endpoints through Federated Learning technology, thereby helping doctors make better decisions and improve patient treatment outcomes.

In summary, the core objective of this paper is to develop a secure and efficient Federated Learning platform to address medical data privacy and security issues while enhancing the clinical effectiveness of stroke management.