2K-Fold-Net and feature enhanced 4-Fold-Net for medical image segmentation

作者:

Highlights:

• A 2K-Fold-Net is proposed to generalize U-Net and its stacked or cascaded variants, with K sub-U-Nets and hence the name.

• The influence of the fold-pair number K on its performance of segmenting medical images is studied.

• A special case with K=2 is realized and empowered with the feature enhancing functionalities of the attention-aware feature enhancement method, and is thus named as “Enhanced-Feature-4-Fold-Net”, or EF3-Net in short.

• The proposed networks are tested by four medical image segmentation datasets, and show superior performance as compared with some other popular variants of U-Net.

摘要

•A 2K-Fold-Net is proposed to generalize U-Net and its stacked or cascaded variants, with K sub-U-Nets and hence the name.•The influence of the fold-pair number K on its performance of segmenting medical images is studied.•A special case with K=2 is realized and empowered with the feature enhancing functionalities of the attention-aware feature enhancement method, and is thus named as “Enhanced-Feature-4-Fold-Net”, or EF3-Net in short.•The proposed networks are tested by four medical image segmentation datasets, and show superior performance as compared with some other popular variants of U-Net.

论文关键词:2K-Fold-Net,EF3-Net,U-Net,AFE,Image segmentation

论文评审过程:Received 10 June 2021, Revised 22 February 2022, Accepted 5 March 2022, Available online 13 March 2022, Version of Record 17 March 2022.

论文官网地址:https://doi.org/10.1016/j.patcog.2022.108625