Automatic Polyp Segmentation Using Modified Recurrent Residual Unet Network

Abstract

Colorectal cancer is a dangerous disease with a high mortality rate. To increase the likelihood of successful treatment, early detection of polyps is a useful solution. The Unet-architecture network model is showing success in medical image segmentation including analysis of polyps from colonoscopy images. Traditional Unet and Unet-based models are often huge, requiring training and deployment with a high-performance system. Designing models with compact size and high performance would be an important goal. In this study, we proposed to modify the Residual Recurrent Unet architecture to improve the size of the model while ensuring the model performance. The proposed model has flexibility in changing the number of filters in convolution units. By taking advantage of the strengths of residual and recurrent structures in terms of reuse of convolutional functions, the new model, therefore, was not only smaller in size but also has superior performance compared to the traditional Unet model and the others. The evaluations were performed on three public Colonoscopy image datasets: CVC-ClinicDB, ETIS-LaribPolypDB, and CVC-ColonDB. The Dice score on CVC-ClinicDB reached 94.59%, ETIS-LaribPolypDB reached 92.73% and 93.31% on CVC-ColonDB dataset. The experimental results obtained from the proposed network on datasets were better than those in recent related studies. The introduced model has a smaller size than the traditional model nevertheless has outstanding performance, therefore, it would be extremely productive for developing applications on low-performance devices.