FeDNet: Feature Decoupled Network for polyp segmentation from endoscopy images

被引:10
作者
Su, Yanzhou [1 ]
Cheng, Jian [1 ]
Zhong, Chuqiao [1 ]
Zhang, Yijie [1 ]
Ye, Jin [2 ]
He, Junjun [2 ]
Liu, Jun [3 ]
机构
[1] Univ Elect Sci & Technol China, Sch Informat & Commun Engn, Chengdu 611731, Sichuan, Peoples R China
[2] Shanghai AI lab, Shanghai, Peoples R China
[3] Singapore Univ Technol & Design, Singapore 487372, Singapore
基金
中国国家自然科学基金;
关键词
Polyp segmentation; Feature decouple; Laplace pyramid; FPN;
D O I
10.1016/j.bspc.2023.104699
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
Early detection and diagnosis of colorectal polyps are critical to the diagnosis and treatment of colorectal cancer. When it comes to polyp segmentation, previous methods have limited benefit whether starting with the global contextual information to maintain the consistency of the information within the polyp or starting with the edge information to refine the prediction results. Therefore, a comprehensive way to obtain good polyp segmentation performance is to optimize both simultaneously. Depending on the above analysis, we explore in this paper how to improve the performance of polyp segmentation by optimizing the body and edge simultaneously. Inspired by the feature decoupled method in Laplacian pyramid, we decouple the input feature into the body and edge feature explicitly in an effective and reasonable way, and subsequently perform targeted optimization by introducing a novel Feature Decoupled Module (FDM). Furthermore, combined with FDM (with only 0.08 m network parameters), our approach can significantly outperform previous state-of-the-art methods, attaining a further improvement over the baseline. Especially, we achieve 92.4% mean Dice on the large-scale Kvasir dataset. Not only that, it also demonstrates strong generalization ability. It obtains the top performance on three datasets, where it achieves 82.3% and 81.0% mean Dice on CVC-ColonDB and ETIS, respectively, far exceeding the competitors. Code will be released.1
引用
收藏
页数:12
相关论文
共 52 条
[1]   SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation [J].
Badrinarayanan, Vijay ;
Kendall, Alex ;
Cipolla, Roberto .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2017, 39 (12) :2481-2495
[2]   WM-DOVA maps for accurate polyp highlighting in colonoscopy: Validation vs. saliency maps from physicians [J].
Bernal, Jorge ;
Javier Sanchez, F. ;
Fernandez-Esparrach, Gloria ;
Gil, Debora ;
Rodriguez, Cristina ;
Vilarino, Fernando .
COMPUTERIZED MEDICAL IMAGING AND GRAPHICS, 2015, 43 :99-111
[3]   THE LAPLACIAN PYRAMID AS A COMPACT IMAGE CODE [J].
BURT, PJ ;
ADELSON, EH .
IEEE TRANSACTIONS ON COMMUNICATIONS, 1983, 31 (04) :532-540
[4]  
Cao H., 2021, arXiv, DOI 10.48550/arXiv:2105.05537
[5]   End-to-End Object Detection with Transformers [J].
Carion, Nicolas ;
Massa, Francisco ;
Synnaeve, Gabriel ;
Usunier, Nicolas ;
Kirillov, Alexander ;
Zagoruyko, Sergey .
COMPUTER VISION - ECCV 2020, PT I, 2020, 12346 :213-229
[6]   HarDNet: A Low Memory Traffic Network [J].
Chao, Ping ;
Kao, Chao-Yang ;
Ruan, Yu-Shan ;
Huang, Chien-Hsiang ;
Lin, Youn-Long .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :3551-3560
[7]   DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFs [J].
Chen, Liang-Chieh ;
Papandreou, George ;
Kokkinos, Iasonas ;
Murphy, Kevin ;
Yuille, Alan L. .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2018, 40 (04) :834-848
[8]  
Deng-Ping Fan, 2020, Medical Image Computing and Computer Assisted Intervention - MICCAI 2020. 23rd International Conference. Proceedings. Lecture Notes in Computer Science (LNCS 12266), P263, DOI 10.1007/978-3-030-59725-2_26
[9]  
Dong B, 2024, Arxiv, DOI [arXiv:2108.06932, 10.48550/arXiv.2108.06932]
[10]  
Dosovitskiy A, 2021, Arxiv, DOI arXiv:2010.11929