LACFormer: Toward accurate and efficient polyp segmentation
Quan Van Nguyen (R&D Lab, Sun Asterisk Inc Vietnam),* Mai Nguyen ( R&D Lab, Sun Asterisk Inc Vietnam), Thanh Tung Nguyen (Sun Asterisk Vietnam), Huy Trịnh Quang (Sun-asterisk), Toan Pham Van (R&D Lab, Sun Asterisk Inc Vietnam), Linh Bao Doan (Sun* Inc.)
The 34th British Machine Vision Conference
Abstract
Polyp segmentation is an essential task in medical image analysis for early detection of colorectal cancer. Deep learning models, particularly encoder-decoder architectures, have been successful in polyp segmentation. However, these models often struggle to capture long-range dependencies and exhibit limited performance on small polyps. In this paper, we propose LACFormer, a novel hierarchical Transformer-CNN model incorporating the Laplacian pyramid for polyp segmentation. The proposed model combines the strengths of Transformers and CNNs along with Laplacian images to overcome the limitations of previous models. Specifically, the hierarchical Transformer backbone captures long-range dependencies and hierarchically processes the features to generate multi-scale representations. These representations are then fused with a novel CNN decoder, which enhances feature representations and refines the segmentation masks. Besides, many novel modules for effective polyp segmentation are also proposed. We evaluated our model on five popular benchmark datasets for polyp segmentation, including Kvasir, CVC-Clinic DB, CVC-ColonDB, CVC-T, and ETIS-Larib. Experimental results show that LACFormer outperforms state-of-the-art models, achieving a Dice similarity coefficient (DSC) of 0.927 and a mean intersection-over-union (mIoU) of 0.878 on CVC-ClinicDB, a DSC of 0.831 and mIoU of 0.753 on CVC-ColonDB and a DSC of 0.824 and mIoU of 0.753 on ETIS-Larib.Video
Citation
@inproceedings{Nguyen_2023_BMVC,
author = {Quan Van Nguyen and Mai Nguyen and Thanh Tung Nguyen and Huy Trịnh Quang and Toan Pham Van and Linh Bao Doan},
title = {LACFormer: Toward accurate and efficient polyp segmentation},
booktitle = {34th British Machine Vision Conference 2023, {BMVC} 2023, Aberdeen, UK, November 20-24, 2023},
publisher = {BMVA},
year = {2023},
url = {https://papers.bmvc2023.org/0411.pdf}
}
Copyright © 2023
The British Machine Vision Conference is organised by