Medical image classification plays an important role in disease diagnosis since it can provide important reference information for doctors. The supervised convolutional neural networks (CNNs) such as DenseNet provide the versatile and effective method for medical image classification tasks, but they require large amounts of data with labels and involve complex and time-consuming training process. The unsupervised CNNs such as principal component analysis network (PCANet) need no labels for training but cannot provide desirable classification accuracy. To realize the accurate medical image classification in the case of a small training dataset, we have proposed a light-weighted hybrid neural network which consists of a modified PCANet cascaded with a simplified DenseNet. The modified PCANet has two stages, in which the network produces the effective feature maps at each stage by convoluting inputs with various learned kernels. The following simplified DenseNet with a small number of weights will take all feature maps produced by the PCANet as inputs and employ the dense shortcut connections to realize accurate medical image classification. To appreciate the performance of the proposed method, some experiments have been done on mammography and osteosarcoma histology images. Experimental results show that the proposed hybrid neural network is easy to train and it outperforms such popular CNN models as PCANet, ResNet and DenseNet in terms of classification accuracy, sensitivity and specificity.
1 February 2023 in IEEE Journal of Biomedical and Health Informatics
In the field of disease diagnosis where only a small dataset of medical images may be accessible, the light-weight convolutional neural network (CNN) has become popular because it can help to avoid the over-fitting problem and improve computational efficiency. However, the feature extraction capability of the light-weight CNN is inferior to that of the heavy-weight counterpart. Although the attention mechanism provides a feasible solution to this problem, the existing attention modules, such as the squeeze and excitation module and the convolutional block attention module, have insufficient non-linearity, thereby influencing the ability of the light-weight CNN to discover the key features. To address this issue, we have proposed a spiking cortical model based global and local (SCM-GL) attention module. The SCM-GL module analyzes the input feature maps in parallel and decomposes each map into several components according to the relation between pixels and their neighbors. The components are weighted summed to obtain a local mask. Besides, a global mask is produced by discovering the correlation between the distant pixels in the feature map. The final attention mask is generated by combining the local and global masks, and it is multiplied by the original map so that the important components can be highlighted to facilitate accurate disease diagnosis. To appreciate the performance of the SCM-GL module, this module and some mainstream attention modules have been embedded into the popular light-weight CNN models for comparison. Experiments on the classification of brain MR, chest X-ray, and osteosarcoma image datasets demonstrate that the SCM-GL module can significantly improve the classification performance of the evaluated light-weight CNN models by enhancing the ability of discovering the suspected lesions and it is generally superior to state-of-the-art attention modules in terms of accuracy, recall, specificity and F1 score.