[A lightweight convolutional neural network for myositis classification from muscle ultrasound images]

Hao Tan,Xun Lang,Tao Wang,Bingbing He,Zhiyao Li,Yu Lu,Yufeng Zhang
DOI: https://doi.org/10.7507/1001-5515.202301023
2024-10-25
Abstract:Existing classification methods for myositis ultrasound images have problems of poor classification performance or high computational cost. Motivated by this difficulty, a lightweight neural network based on a soft threshold attention mechanism is proposed to cater for a better IIMs classification. The proposed network was constructed by alternately using depthwise separable convolution (DSC) and conventional convolution (CConv). Moreover, a soft threshold attention mechanism was leveraged to enhance the extraction capabilities of key features. Compared with the current dual-branch feature fusion myositis classification network with the highest classification accuracy, the classification accuracy of the network proposed in this paper increased by 5.9%, reaching 96.1%, and its computational complexity was only 0.25% of the existing method. The obtained results support that the proposed method can provide physicians with more accurate classification results at a lower computational cost, thereby greatly assisting them in their clinical diagnosis.
What problem does this paper attempt to address?