HAB‐Net: Hierarchical asymmetric convolution and boundary enhancement network for brain tumor segmentation

Yuanjing Hu,Aibin Huang,Rui Xu
DOI: https://doi.org/10.1049/ipr2.13065
IF: 2.3
2024-02-28
IET Image Processing
Abstract:This paper proposes the model HAB‐Net, which consists of two new modules that can tackle the problem of how to capture intricate tumour structures and boundaries: A hierarchical and pseudo‐convolutional module (hierarchical asymmetric convolutions [HAC] module), and a boundary attention module (BAM). The HAC module adopts depth wise separable convolution, hierarchical convolution, and coordinate attention, which can reduce computational complexity while improving segmentation accuracy and the BAM aims to identify fuzzy boundary shapes by enhancing tumour boundary information. Experiments on the BraTS datasets indicate our HAB‐Net performs outstandingly. Brain tumour segmentation (BTS) is crucial for diagnosis and treatment planning by delineating tumour boundaries and subregions in multi‐modality bio‐imaging data. Several BTS models have been proposed to address specific technical challenges encountered in this field. However, accurately capturing intricate tumour structures and boundaries remains a difficult task. To overcome this challenge, HAB‐Net, a model that combines the strengths of convolutional neural networks and transformer architectures, is presented. HAB‐Net incorporates a custom‐designed hierarchical and pseudo‐convolutional module called hierarchical asymmetric convolutions (HAC). In the encoder, a coordinate attention is included to extract feature maps. Additionally, swin transformer, which has a self‐attention mechanism, is integrated to effectively capture long‐range relationships. Moreover, the decoder is enhanced with a boundary attention module (BAM) to improve boundary information and overall segmentation performance. Extensive evaluations conducted on the BraTS2018 and BraTS2021 datasets demonstrate significant improvements in segmentation accuracy for tumour regions.
computer science, artificial intelligence,engineering, electrical & electronic,imaging science & photographic technology
What problem does this paper attempt to address?