HiFuse: Hierarchical Multi-Scale Feature Fusion Network for Medical Image Classification

Xiangzuo Huo,Gang Sun,Shengwei Tian,Yan Wang,Long Yu,Jun Long,Wendong Zhang,Aolun Li
DOI: https://doi.org/10.48550/arXiv.2209.10218
2022-09-21
Abstract:Medical image classification has developed rapidly under the impetus of the convolutional neural network (CNN). Due to the fixed size of the receptive field of the convolution kernel, it is difficult to capture the global features of medical images. Although the self-attention-based Transformer can model long-range dependencies, it has high computational complexity and lacks local inductive bias. Much research has demonstrated that global and local features are crucial for image classification. However, medical images have a lot of noisy, scattered features, intra-class variation, and inter-class similarities. This paper proposes a three-branch hierarchical multi-scale feature fusion network structure termed as HiFuse for medical image classification as a new method. It can fuse the advantages of Transformer and CNN from multi-scale hierarchies without destroying the respective modeling so as to improve the classification accuracy of various medical images. A parallel hierarchy of local and global feature blocks is designed to efficiently extract local features and global representations at various semantic scales, with the flexibility to model at different scales and linear computational complexity relevant to image size. Moreover, an adaptive hierarchical feature fusion block (HFF block) is designed to utilize the features obtained at different hierarchical levels comprehensively. The HFF block contains spatial attention, channel attention, residual inverted MLP, and shortcut to adaptively fuse semantic information between various scale features of each branch. The accuracy of our proposed model on the ISIC2018 dataset is 7.6% higher than baseline, 21.5% on the Covid-19 dataset, and 10.4% on the Kvasir dataset. Compared with other advanced models, the HiFuse model performs the best. Our code is open-source and available from <a class="link-external link-https" href="https://github.com/huoxiangzuo/HiFuse" rel="external noopener nofollow">this https URL</a>.
Image and Video Processing,Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
### Problems the Paper Attempts to Solve This paper aims to address several key issues in medical image classification: 1. **Difficulty in Capturing Global Features**: Due to the fixed receptive field of convolutional kernels, traditional Convolutional Neural Networks (CNNs) find it challenging to capture the global features of medical images. 2. **Lack of Local Feature Details**: Although Transformers based on self-attention mechanisms can model long-range dependencies, their high computational complexity and lack of local inductive bias result in insufficient local feature details. 3. **Intra-class Variation and Inter-class Similarity**: Medical images have a large amount of noise, scattered features, intra-class variation, and inter-class similarity, making it difficult for models to distinguish between different categories of images. 4. **Fusion of Multi-scale Features**: To improve classification accuracy, it is necessary to effectively fuse local and global features at different levels. To this end, the paper proposes a new three-branch hierarchical multi-scale feature fusion network structure, called HiFuse, for medical image classification. HiFuse combines the advantages of Transformers and CNNs, efficiently extracting local features and global representations through a multi-scale hierarchical structure, and designs an adaptive hierarchical feature fusion block (HFF block) to fully utilize feature information at different levels. This structure not only improves classification accuracy but also reduces computational complexity.