Research on Improving TransUnet Network for Feature Classification

Jiangjie Yin,Ci Wang,LiPing Liu,Haolong Yang,Xuehong Sun,Yanni Wang
DOI: https://doi.org/10.1117/12.3004083
2023-01-01
Abstract:To address the low accuracy of TransUent network in small-area feature recognition, two feature classification networks based on improved TransUent network are proposed: DS-TransUnet network and DS-A-TransUnet network.DS-TransUnet network introduces a residual module in the bottom layer of the Convolutional Neural Network (CNN) module of the hybrid encoder of TransUent network to deepen the network and uses convolutional operation instead of maximum pooling in the hybrid encoder to reduce the loss of feature information. The DS-TransUnet network introduces a residual module at the bottom of the Convolutional Neural Networks (CNN) module of the TransUent network to deepen the network, and uses convolutional operations instead of maximum pooling in the hybrid encoder to reduce the loss of feature information, and uses depth-separable convolution instead of regular convolution to reduce the residual blocks and the large number of parameters caused by convolutional operations; the DS-A-TransUnet network On the basis of DS-TransUnet network, a void space pyramidal pooling module is introduced, and the classification accuracy is further improved by using void convolution with different expansion rates in parallel to obtain different size of perceptual fields and fully extract multi-scale features. The experimental results show that the recognition accuracy of DS-TransUnet network is 3.06% and 2.85% higher than that of TransUent network for arable land and water bodies, respectively, and the overall recognition accuracy reaches 89.14%; the recognition accuracy of DS-A-TransUnet network is 3.17% and 4.79% higher than that of TransUent network for arable land and water bodies, respectively, and the overall recognition accuracy reaches 90.53%. The overall recognition accuracy reached 90.53%.
What problem does this paper attempt to address?