Abstract
AbstractThis paper introduces a Lightweight U-Net (LWU-Net) method for efficient gastro-intestinal tract segmentation in resource-constrained environments. The proposed model seeks to strike a balance between computational efficiency, memory efficiency, and segmentation accuracy. The model achieves competitive performance while reducing computational power needed with improvements including depth-wise separable convolutions and optimised network depth. The evaluation is conducted using data from a Kaggle competition-UW Madison gastrointestinal tract image segmentation, demonstrating the model’s effectiveness and generalizability. The findings demonstrate that the LWU-Net model has encouraging promise for precise medical diagnoses in resource-constrained settings, enabling effective image segmentation with slightly less than a fifth of as many trainable parameters as the U-Net model.
Publisher
Cold Spring Harbor Laboratory
Reference29 articles.
1. Agarap, A.F. , 2018. Deep learning using rectified linear units (relu). arXiv preprint arXiv:1803.08375.
2. Deep learning for detection and segmentation of artefact and disease instances in gastrointestinal endoscopy;Medical image analysis,2021
3. Asperti, A. , Mastronardo, C. , 2017. The effectiveness of data augmentation for detection of gastrointestinal diseases from endoscopical images. arXiv preprint arXiv:1712.03689.
4. Chen, L.C. , Zhu, Y. , Papandreou, G. , Schroff, F. , Adam, H. , 2018. Encoder-decoder with atrous separable convolution for semantic image segmentation, in: Proceedings of the European conference on computer vision (ECCV), pp. 801–818.
5. U-net: deep learning for cell counting, detection, and morphometry;Nature methods,2019