DOI: 10.1371/journal.pone.0343457 ISSN: 1932-6203

An efficient brain tumor classification system with accurate recognition and complementary downsampling blocks

Jincan Zhang, Chuanqi Cai, XingHua Tan, Wenna Chen, Xin Zhao, Ganqin Du, Hongwei Jiang

Due to the substantial impact of brain tumors on human health, precise identification of tumor types is vital for patient prognosis and guiding treatment decisions. Magnetic Resonance Imaging (MRI) technology, owing to its non-invasive nature and high resolution, is indispensable in accurately classifying brain tumor types. However, manual interpretation of MRI images is not only time-consuming and labor-intensive but also susceptible to subjective bias. Hence, many automatic brain tumor diagnosis systems utilizing deep learning techniques have been developed. This paper introduces an advanced modularized Convolutional Neural Network (CNN) that incorporates ConvDAtt block, Complementary Information (CI) downsampling block, and Spatial Pyramid Pooling (SPP) block. Firstly, the ConvDAtt block combines convolution, depth-separable convolution and Coordinated Attention (CA) mechanism. The module alleviates the gradient vanishing problem in deep neural network training, enhancing the training stability and performance of the model. In the ConvDAtt block, the CA mechanism can not only capture information across channels, but also sense direction and position sensitive information, improving the accuracy of the model’s identification of objects of interest. Secondly, the CI block is a downsampling block with convolution and pooling functions. In the CI block, there is a complementary relationship between the pooling layer and the convolution layer, which is mainly reflected in two aspects, feature extraction and dimensionality reduction, enlargement of receptive field. Moreover, the complementary relationship can solve the problem of information loss in the process of downsampling, while also significantly reducing computational cost. Thirdly, the SPP block extracts features from different scales of the input data and integrates these features, thus enhancing the performance of the network, and enhancing the robustness of the model. The combination of ConvDAtt block, CI block and SPP block can effectively extract the information features and discriminant features of brain tumor MRI images, which has the advantages of parameter reduction, stable training and clear structure.