Global-Local Framework for Medical Image Segmentation with Intra-class Imbalance Problem

Yifan Zhou, Bing Yang, Xiaolu Lin, Risa Higashita, Jiang Liu

Research output: Chapter in Book/Conference proceedingConference contributionpeer-review

Abstract

Deep learning methods have been demonstrated effective in medical image segmentation tasks. The results are affected by data imbalance problems. The inter-class imbalance is often considered, while the intra-class imbalance is not. The intra-class imbalance usually occurs in medical images due to external influences such as noise interference and changes in camera angle, resulting in insufficient discriminative representations within classes. Deep learning methods are easy to segment regions without complex textures and varied appearances. They are susceptible to the intra-class imbalance problem in medical images. In this paper, we propose a two-stage global-local framework to solve the intra-class imbalance problem and increase segmentation accuracy. The framework consists of (1) an auxiliary task network(ATN), (2) a local patch network(LPN), and (3) a fusion module. The ATN has a shared encoder and two separate decoders that perform global segmentation and key points localization. The key points guide to generating the fuzzy patches for the LPN. The LPN focuses on challenging patches to get a more accurate result. The fusion module generates the final output according to the global and local segmentation results. Furthermore, we have performed experiments on a private iris dataset with 290 images and a public CAMUS dataset with 1800 images. Our method achieves an IoU of 0.9280 on the iris dataset and an IoU of 0.8511 on the CAMUS dataset. The results on both datasets show that our method achieves superior performance over U-Net, CE-Net, and U-Net++.

Original languageEnglish
Title of host publicationCACML 2023 - Conference Proceedings 2023 2nd Asia Conference on Algorithms, Computing and Machine Learning
PublisherAssociation for Computing Machinery
Pages366-370
Number of pages5
ISBN (Electronic)9781450399449
DOIs
Publication statusPublished - 17 Mar 2023
Externally publishedYes
Event2nd Asia Conference on Algorithms, Computing and Machine Learning, CACML 2023 - Shanghai, China
Duration: 17 Mar 202319 Mar 2023

Publication series

NameACM International Conference Proceeding Series

Conference

Conference2nd Asia Conference on Algorithms, Computing and Machine Learning, CACML 2023
Country/TerritoryChina
CityShanghai
Period17/03/2319/03/23

Keywords

  • Deep Learning
  • Intra-class imbalance
  • Medical image segmentation

ASJC Scopus subject areas

  • Human-Computer Interaction
  • Computer Networks and Communications
  • Computer Vision and Pattern Recognition
  • Software

Fingerprint

Dive into the research topics of 'Global-Local Framework for Medical Image Segmentation with Intra-class Imbalance Problem'. Together they form a unique fingerprint.

Cite this