Medical image segmentation is a branch of computer vision that is essential to diagnose diseases and improve patient care. In this study is focused on the dual-channel UNet (DC-UNet) model to detect polyps. Polyps are lesions that vary in size and are key to preventing colorectal cancer through early identification. To train and evaluate the DC-UNet, we use the challenging CVC-ClinicDB dataset, which includes colonoscopy video frames with annotated polyp regions. Given the complexity of this dataset, we propose using genetic algorithms to optimize the DC-UNet model’s hyperparameters. Specifically, we tune the selection of gradient-based optimizers, dropout rates, learning rates, \(\beta _1\) for first-order moment, and \(\beta _2\) for second-order moment. We tune these model hyperparameters to achieve optimal results, improving its performance by 4. 36% compared to the standard hyperparameter values. This approach offers a promising direction to improve medical image analysis and outcomes in early cancer detection.

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

Genetic Algorithms for Hyperparameter Tuning of a DC-UNet Model for Medical Image Segmentation

  • Krishna Román,
  • Rolando Armas,
  • Carlos Montenegro

摘要

Medical image segmentation is a branch of computer vision that is essential to diagnose diseases and improve patient care. In this study is focused on the dual-channel UNet (DC-UNet) model to detect polyps. Polyps are lesions that vary in size and are key to preventing colorectal cancer through early identification. To train and evaluate the DC-UNet, we use the challenging CVC-ClinicDB dataset, which includes colonoscopy video frames with annotated polyp regions. Given the complexity of this dataset, we propose using genetic algorithms to optimize the DC-UNet model’s hyperparameters. Specifically, we tune the selection of gradient-based optimizers, dropout rates, learning rates, \(\beta _1\) for first-order moment, and \(\beta _2\) for second-order moment. We tune these model hyperparameters to achieve optimal results, improving its performance by 4. 36% compared to the standard hyperparameter values. This approach offers a promising direction to improve medical image analysis and outcomes in early cancer detection.