Enhancing Performance for Highly Imbalanced Medical Data via Data Regularization in a Federated Learning Setting
Fuente:
arXiv
Saved in:
| Main Authors: | , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2024
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866914816953679872 |
|---|---|
| author | Tsoumplekas, Georgios Siniosoglou, Ilias Argyriou, Vasileios Moscholios, Ioannis D. Sarigiannidis, Panagiotis |
| author_facet | Tsoumplekas, Georgios Siniosoglou, Ilias Argyriou, Vasileios Moscholios, Ioannis D. Sarigiannidis, Panagiotis |
| contents | The increased availability of medical data has significantly impacted healthcare by enabling the application of machine / deep learning approaches in various instances. However, medical datasets are usually small and scattered across multiple providers, suffer from high class-imbalance, and are subject to stringent data privacy constraints. In this paper, the application of a data regularization algorithm, suitable for learning under high class-imbalance, in a federated learning setting is proposed. Specifically, the goal of the proposed method is to enhance model performance for cardiovascular disease prediction by tackling the class-imbalance that typically characterizes datasets used for this purpose, as well as by leveraging patient data available in different nodes of a federated ecosystem without compromising their privacy and enabling more resource sensitive allocation. The method is evaluated across four datasets for cardiovascular disease prediction, which are scattered across different clients, achieving improved performance. Meanwhile, its robustness under various hyperparameter settings, as well as its ability to adapt to different resource allocation scenarios, is verified. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2405_20430 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Enhancing Performance for Highly Imbalanced Medical Data via Data Regularization in a Federated Learning Setting Tsoumplekas, Georgios Siniosoglou, Ilias Argyriou, Vasileios Moscholios, Ioannis D. Sarigiannidis, Panagiotis Machine Learning Artificial Intelligence The increased availability of medical data has significantly impacted healthcare by enabling the application of machine / deep learning approaches in various instances. However, medical datasets are usually small and scattered across multiple providers, suffer from high class-imbalance, and are subject to stringent data privacy constraints. In this paper, the application of a data regularization algorithm, suitable for learning under high class-imbalance, in a federated learning setting is proposed. Specifically, the goal of the proposed method is to enhance model performance for cardiovascular disease prediction by tackling the class-imbalance that typically characterizes datasets used for this purpose, as well as by leveraging patient data available in different nodes of a federated ecosystem without compromising their privacy and enabling more resource sensitive allocation. The method is evaluated across four datasets for cardiovascular disease prediction, which are scattered across different clients, achieving improved performance. Meanwhile, its robustness under various hyperparameter settings, as well as its ability to adapt to different resource allocation scenarios, is verified. |
| title | Enhancing Performance for Highly Imbalanced Medical Data via Data Regularization in a Federated Learning Setting |
| topic | Machine Learning Artificial Intelligence |
| url | https://arxiv.org/abs/2405.20430 |