A Dual-path Network Chest Film Disease Classification Method Combined with a Triple Attention Mechanism

Qiang Li,Xu Wang,Xin Guan
DOI: https://doi.org/10.11999/JEIT220172
2023-01-01
Abstract:In recent years, medical image processing with CNN has made remarkable research progress in the task of chest film disease classification. However, compared with single structure CNN, dual-path network can combine the characteristics of different CNN to improve the ability of disease classification. Secondly, for different diseases, their location, size, shape, and texture are different, the attention mechanism helps the model to extract different pathological features and improve the classification accuracy. Therefore, focusing on the chest film disease classification problem, a dual path convolution neural network TADPN(Triple Attention Dual Path Network) combined with a triple attention mechanism is proposed. TADPN takes the dual-path network combined with ResNet and DenseNet as the backbone network and uses three different forms of attention mechanisms to improve the backbone network. The network complexity and classification accuracy are improved while maintaining the stability of the parameters. In this paper, the validity of TADPN is compared with the six advanced algorithms on the ChestXray14 dataset. The experiments show the progressiveness of the dual-path CNN and the triple attention mechanism, as well as the effectiveness of TADPN. The average AUC value of 14 diseases reaches 0.8185, which is 1.1% higher than that of previous generations.
What problem does this paper attempt to address?