Medical image segmentation plays a pivotal role in modern healthcare applications, significantly assisting healthcare professionals in making accurate diagnosis. However, limited annotated data hinders the development of robust medical image segmentation models due to strict patient privacy regulations. Semi-supervised learning offers a solution by leveraging both labeled and unlabeled data, enabling models to learn from larger datasets while capitalizing on limited labeled data. Despite its benefits, semi-supervised learning continues to face concerns related to data privacy. Federated learning (FL) addresses these concerns by enabling multiple parties to collaboratively train a model without sharing their local data. As the need for data-efficient and privacy-preserving machine learning techniques grows, semi-supervised federated learning emerges as a potential paradigm to tackle this issue. Our work underscores the importance of semi-supervised federated learning approaches that can effectively leverage both labeled and unlabeled data distributed across multiple sites. However, non-Independent and Identically Distributed (non-IID) data and class data imbalance remain significant challenges in semi-supervised federated learning, hindering overall performance. To address these issues, we propose the Federated Learning-Pseudo labeled Segmentation with Class balance (FL-PSeC) framework, which focuses on tackling class imbalance across clients’ local data by assigning higher weightage to underrepresented classes when generating annotations from unlabeled data. This strategy not only improves the IID nature of the data but also leads to enhanced global model performance. Our experiments on HAM10000, BUS, BUSIS, and UDIAT datasets demonstrate the efficacy of FL-PSeC in adapting to a class imbalance in FL, outperforming existing methods.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

FL-PSeC: Federated Learning-Pseudo Labeled Medical Image Segmentation with Personalized Class Balancing Semi-supervised Approach

  • Ishu Priya,
  • C. Krishna Mohan

摘要

Medical image segmentation plays a pivotal role in modern healthcare applications, significantly assisting healthcare professionals in making accurate diagnosis. However, limited annotated data hinders the development of robust medical image segmentation models due to strict patient privacy regulations. Semi-supervised learning offers a solution by leveraging both labeled and unlabeled data, enabling models to learn from larger datasets while capitalizing on limited labeled data. Despite its benefits, semi-supervised learning continues to face concerns related to data privacy. Federated learning (FL) addresses these concerns by enabling multiple parties to collaboratively train a model without sharing their local data. As the need for data-efficient and privacy-preserving machine learning techniques grows, semi-supervised federated learning emerges as a potential paradigm to tackle this issue. Our work underscores the importance of semi-supervised federated learning approaches that can effectively leverage both labeled and unlabeled data distributed across multiple sites. However, non-Independent and Identically Distributed (non-IID) data and class data imbalance remain significant challenges in semi-supervised federated learning, hindering overall performance. To address these issues, we propose the Federated Learning-Pseudo labeled Segmentation with Class balance (FL-PSeC) framework, which focuses on tackling class imbalance across clients’ local data by assigning higher weightage to underrepresented classes when generating annotations from unlabeled data. This strategy not only improves the IID nature of the data but also leads to enhanced global model performance. Our experiments on HAM10000, BUS, BUSIS, and UDIAT datasets demonstrate the efficacy of FL-PSeC in adapting to a class imbalance in FL, outperforming existing methods.