A Federated Learning Framework for Stenosis Detection
- URL: http://arxiv.org/abs/2310.19445v1
- Date: Mon, 30 Oct 2023 11:13:40 GMT
- Title: A Federated Learning Framework for Stenosis Detection
- Authors: Mariachiara Di Cosmo, Giovanna Migliorelli, Matteo Francioni, Andi
Mucaj, Alessandro Maolo, Alessandro Aprile, Emanuele Frontoni, Maria Chiara
Fiorentino, and Sara Moccia
- Abstract summary: This study explores the use of Federated Learning (FL) for stenosis detection in coronary angiography images (CA)
Two heterogeneous datasets from two institutions were considered: dataset 1 includes 1219 images from 200 patients, which we acquired at the Ospedale Riuniti of Ancona (Italy)
dataset 2 includes 7492 sequential images from 90 patients from a previous study available in the literature.
- Score: 70.27581181445329
- License: http://creativecommons.org/licenses/by-nc-sa/4.0/
- Abstract: This study explores the use of Federated Learning (FL) for stenosis detection
in coronary angiography images (CA). Two heterogeneous datasets from two
institutions were considered: Dataset 1 includes 1219 images from 200 patients,
which we acquired at the Ospedale Riuniti of Ancona (Italy); Dataset 2 includes
7492 sequential images from 90 patients from a previous study available in the
literature. Stenosis detection was performed by using a Faster R-CNN model. In
our FL framework, only the weights of the model backbone were shared among the
two client institutions, using Federated Averaging (FedAvg) for weight
aggregation. We assessed the performance of stenosis detection using Precision
(P rec), Recall (Rec), and F1 score (F1). Our results showed that the FL
framework does not substantially affects clients 2 performance, which already
achieved good performance with local training; for client 1, instead, FL
framework increases the performance with respect to local model of +3.76%,
+17.21% and +10.80%, respectively, reaching P rec = 73.56, Rec = 67.01 and F1 =
70.13. With such results, we showed that FL may enable multicentric studies
relevant to automatic stenosis detection in CA by addressing data heterogeneity
from various institutions, while preserving patient privacy.
Related papers
- Leveraging Pre-trained Models for Robust Federated Learning for Kidney Stone Type Recognition [1.7243216387069678]
Using pre-trained models, this research suggests a strong FL framework to improve kidney stone diagnosis.
We achieved a peak accuracy of 84.1% with seven epochs and 10 rounds during LPO stage, and 77.2% during FRV stage, showing enhanced diagnostic accuracy and robustness against image corruption.
arXiv Detail & Related papers (2024-09-30T04:23:47Z) - Enhancing Diagnostic Reliability of Foundation Model with Uncertainty Estimation in OCT Images [41.002573031087856]
We developed a foundation model with uncertainty estimation (FMUE) to detect 11 retinal conditions on optical coherence tomography ( OCT)
FMUE achieved a higher F1 score of 96.76% than two state-of-the-art algorithms, RETFound and UIOS, and got further improvement with thresholding strategy to 98.44%.
Our model is superior to two ophthalmologists with a higher F1 score (95.17% vs. 61.93% &71.72%)
arXiv Detail & Related papers (2024-06-18T03:04:52Z) - Vision Transformer for Efficient Chest X-ray and Gastrointestinal Image
Classification [2.3293678240472517]
This study uses different CNNs and transformer-based methods with a wide range of data augmentation techniques.
We evaluated their performance on three medical image datasets from different modalities.
arXiv Detail & Related papers (2023-04-23T04:07:03Z) - MS Lesion Segmentation: Revisiting Weighting Mechanisms for Federated
Learning [92.91544082745196]
Federated learning (FL) has been widely employed for medical image analysis.
FL's performance is limited for multiple sclerosis (MS) lesion segmentation tasks.
We propose the first FL MS lesion segmentation framework via two effective re-weighting mechanisms.
arXiv Detail & Related papers (2022-05-03T14:06:03Z) - Vision Transformers for femur fracture classification [59.99241204074268]
The Vision Transformer (ViT) was able to correctly predict 83% of the test images.
Good results were obtained in sub-fractures with the largest and richest dataset ever.
arXiv Detail & Related papers (2021-08-07T10:12:42Z) - Deep learning-based COVID-19 pneumonia classification using chest CT
images: model generalizability [54.86482395312936]
Deep learning (DL) classification models were trained to identify COVID-19-positive patients on 3D computed tomography (CT) datasets from different countries.
We trained nine identical DL-based classification models by using combinations of the datasets with a 72% train, 8% validation, and 20% test data split.
The models trained on multiple datasets and evaluated on a test set from one of the datasets used for training performed better.
arXiv Detail & Related papers (2021-02-18T21:14:52Z) - FLANNEL: Focal Loss Based Neural Network Ensemble for COVID-19 Detection [61.04937460198252]
We construct the X-ray imaging data from 2874 patients with four classes: normal, bacterial pneumonia, non-COVID-19 viral pneumonia, and COVID-19.
To identify COVID-19, we propose a Focal Loss Based Neural Ensemble Network (FLANNEL)
FLANNEL consistently outperforms baseline models on COVID-19 identification task in all metrics.
arXiv Detail & Related papers (2020-10-30T03:17:31Z) - CovidDeep: SARS-CoV-2/COVID-19 Test Based on Wearable Medical Sensors
and Efficient Neural Networks [51.589769497681175]
The novel coronavirus (SARS-CoV-2) has led to a pandemic.
The current testing regime based on Reverse Transcription-Polymerase Chain Reaction for SARS-CoV-2 has been unable to keep up with testing demands.
We propose a framework called CovidDeep that combines efficient DNNs with commercially available WMSs for pervasive testing of the virus.
arXiv Detail & Related papers (2020-07-20T21:47:28Z) - Deep Learning Based Detection and Localization of Intracranial Aneurysms
in Computed Tomography Angiography [5.973882600944421]
A two-step model was implemented: a 3D region proposal network for initial aneurysm detection and 3D DenseNetsfor false-positive reduction.
Our model showed statistically higher accuracy, sensitivity, and specificity when compared to the available model at 0.25 FPPV and the best F-1 score.
arXiv Detail & Related papers (2020-05-22T10:49:23Z) - CNN-CASS: CNN for Classification of Coronary Artery Stenosis Score in
MPR Images [0.0]
We develop an automated model to identify stenosis severity in MPR images.
The model predicts one of three classes: 'no stenosis' for normal, 'non-significant' - 1-50% of stenosis detected,'significant' - more than 50% of stenosis.
For stenosis score classification, the method shows improved performance comparing to previous works, achieving 80% accuracy on the patient level.
arXiv Detail & Related papers (2020-01-23T15:20:22Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.