Rethinking the transfer learning for FCN based polyp segmentation in colonoscopy

Wen, Yan, Zhang, Lei, Meng, Xiangli and Ye, Xujiong (2023) Rethinking the transfer learning for FCN based polyp segmentation in colonoscopy. IEEE Access, 11 . 16183 -16193. ISSN 2169-3536

Full content URL: https://doi.org/10.1109/ACCESS.2023.3245519

Documents
Rethinking the transfer learning for FCN based polyp segmentation in colonoscopy
Open Access manuscript
[img]
[Download]
[img]
Preview
PDF
Rethinking_the_Transfer_Learning_for_FCN_Based_Polyp_Segmentation.pdf - Whole Document
Available under License Creative Commons Attribution 4.0 International.

1MB
Item Type:Article
Item Status:Live Archive

Abstract

Besides the complex nature of colonoscopy frames with intrinsic frame formation artefacts such as light reflections and the diversity of polyp types/shapes, the publicly available polyp segmentation training datasets are limited, small and imbalanced. In this case, the automated polyp segmentation using a deep neural network remains an open challenge due to the overfitting of training on small datasets. We proposed a simple yet effective polyp segmentation pipeline that couples the segmentation (FCN) and classification (CNN) tasks. We find the effectiveness of interactive weight transfer between dense and coarse vision tasks that mitigates the overfitting in learning. This motivates us to design a new training scheme within our segmentation pipeline. Our method is evaluated on CVC-EndoSceneStill and Kvasir-SEG datasets. It achieves 4.34% and 5.70% Polyp-IoU improvements compared to the state-of-the-art methods on the EndoSceneStill and Kvasir-SEG datasets, respectively and achieves real-time performance in inference. The
model and code are available at https://github.com/MELSunny/Keras-FCN.

Keywords:Colonoscopy, real-time polyp segmentation, transfer learning, convolutional neural network
Subjects:G Mathematical and Computer Sciences > G400 Computer Science
Divisions:College of Science > School of Computer Science
Related URLs:
ID Code:53764
Deposited On:17 Mar 2023 11:48

Repository Staff Only: item control page