Analysis of Multi-centric AI-based frame-works in prostate segmentation
Prostate cancer (PCa) is the second most frequently diagnosed cancer in men worldwide.Several studies have investigated AI-based computer-aided diagnosis (CAD) for PCa diagnosis. This project aims at developing novel approaches for integrating AI-based CAD systems for patient selection and monitoring in multi-centric collaborative studies. Working with multi-center data in the medical field raises several issues related to the handling of sensitive data. The EU GDPR clearly states that data of this kind need special attention and cannot be shared among various entities except under specific assumptions. This is why Federated Learning (FL) has gained so much interest in the medical field in recent years. The FL approach is decentralized in terms of training data and on-device computations. During FL, raw data are stored on end-user devices, which work together to train a common model. On a central server, locally computed updates and analysis results are received and aggregated for an enhanced global model benefiting from distributed learning. Afterward, the updated model is shared with clients, so that knowledge can be shared. Considering the large number of papers published in FL for health care, the first question this project tries to answer is how useful a federated approach actually is. In fact, other embedding approaches have proven useful in similar contexts over the years. From these considerations, the purpose of this paper is to compare the results, obtained on the same conditions, of FL and consensus-based models in order to define cases where each of them is more convenient.Talking of results, our focus is not only on comparing the selected metric (that in this context,for our segmentation problem is the Dice Loss) but also on considering the costs of a modelin terms of CO2 emissions, training time, and, in future, also economic cost for a company. This project is developed in collaboration between Inria, 3IA, and KCL, under the supervision of professors Sebastien Ourselin, Marco Lorenzi, and Michela Antonelli.