1. 基于自分块轻量化 Transformer 的 医学图像分割网络.
- Author
-
张文杰, 宋艳涛, 王克琪, and 张越
- Subjects
- *
COMPUTER-assisted image analysis (Medicine) , *COMPUTER-aided diagnosis , *DIAGNOSTIC imaging , *PARALLEL programming , *DATA science - Abstract
The traditional medical image segmentation network has a large number of parameters and slow computing speed, and cannot applies effectively to the real-time detection technology. To address this issue, this paper proposed a lightweight medical image segmentation network called SPTFormer. Firstly, this network constructed a self-blocking Transformer module, which reshaped the feature map through an adaptive blocking strategy and utilized parallel computing to improve the attention operation speed while paying attention to local detail features. Secondly, this network constructed an SR-CNN module, which used the shift-restored operation to improve the ability to capture local spatial information. By experimenting on ISIC 2018, BUSI, CVC-ClinicDB and 2018 data science bowl, compared with the TransUNet model based on Transformer, the accuracy of the proposed network improves by 4.28%, 3.74%, 6.50%, and 1. 16%, respectively, the GPU computation time reduces by 58%. The proposed network has better performance in medical image segmentation applications, which can well balance the network accuracy and complexity, and provides a new solution for real-time computer-aided diagnosis. [ABSTRACT FROM AUTHOR]
- Published
- 2024
- Full Text
- View/download PDF