高级检索
张学顶, 张术昌, 王红霞, 王亚东. 多尺度小波池化协方差网络:对噪声鲁棒的病理学图像分类算法[J]. 计算机辅助设计与图形学学报, 2023, 35(4): 538-552. DOI: 10.3724/SP.J.1089.2023.19379
引用本文: 张学顶, 张术昌, 王红霞, 王亚东. 多尺度小波池化协方差网络:对噪声鲁棒的病理学图像分类算法[J]. 计算机辅助设计与图形学学报, 2023, 35(4): 538-552. DOI: 10.3724/SP.J.1089.2023.19379
Zhang Xueding, Zhang Shuchang, Wang Hongxia, and Wang Yadong. Multi-Scale Wavelet Pooling Covariance Network: A Robust Image Classification Algorithm for Noisy Pathological Images[J]. Journal of Computer-Aided Design & Computer Graphics, 2023, 35(4): 538-552. DOI: 10.3724/SP.J.1089.2023.19379
Citation: Zhang Xueding, Zhang Shuchang, Wang Hongxia, and Wang Yadong. Multi-Scale Wavelet Pooling Covariance Network: A Robust Image Classification Algorithm for Noisy Pathological Images[J]. Journal of Computer-Aided Design & Computer Graphics, 2023, 35(4): 538-552. DOI: 10.3724/SP.J.1089.2023.19379

多尺度小波池化协方差网络:对噪声鲁棒的病理学图像分类算法

Multi-Scale Wavelet Pooling Covariance Network: A Robust Image Classification Algorithm for Noisy Pathological Images

  • 摘要: 将基于深度学习的图像分类方法用于辅助病理学诊断优势突出,但获取病理学切片过程中产生的噪声会影响网络的泛化性能,进而降低分类算法的准确率.针对该问题,提出了一种鲁棒的病理学图像分类算法——多尺度小波池化协方差(multi-scale wavelet pooling covariance,MWPC)网络.MWPC网络主要由小波池化层、复合卷积层、多尺度特征融合和协方差特征提取层4个核心模块构成,其中小波池化层在抑制噪声影响的同时,保护了有效特征不受损失.多尺度特征融合将浅层特征与深层特征结合,使深层特征能够保留更多图像细节.协方差特征提取层可以获取图像的高阶统计特征,提高网络的泛化性能.在病理图像数据集上的测试结果表明,MWPC网络针对组织病理学图像分块级别的五分类任务,在无噪声条件下准确率可以达到90.90%,比ResNet提高1.68%,比Inception-v3分类网络提高0.43%;在模拟椒盐噪声、高斯噪声和柯西噪声等条件下,其噪声鲁棒性能提升明显,且能够降低平均噪声误差.多种网络模块的消融实验结果表明,MWPC网络能够提高网络分类性能和噪声鲁棒性.

     

    Abstract: Using image classification method based on deep learning are outstanding for pathological diagnosis. However, the noise generated in the process of obtaining pathological slices can affect the generalization performance of the network, thereby reducing the accuracy of the classification algorithm. In response to this problem, the paper proposes a new robust pathological image classification method — multi-scale wavelet pooling covariance (MWPC) network. MWPC network is composed of four core modules: wavelet pooling layer, composite convolution layer, multi-scale feature fusion layer and covariance feature extraction layer. Wavelet pooling layer can suppress the influence of noise while protecting effective features from loss.The accuracy rate can reach 90.90% under noise-free conditions, which is 1.68% higher than ResNet and 0.43% higher than Inception-v3. Under the conditions of simulated salt and pepper noise, Gaussian noise and Cauchy noise, the noise robustness of MWPC is significantly improved, which also reduces the mean noise error. The ablation experiment about network modules shows that the MWPC network can improve performance and noise robustness. Multi-scale feature fusion layer combines the shallow features with the image features, so that the image features can retain more image details. Covariance feature extraction layer can obtain high-order statistical features of the image, and improve the generalization performance of the network. On the clinical data set, the proposed MWPC network in the paper is aimed at the five-class task of histopathological image patches.The accuracy rate can reach 90.90% under noise-free conditions, which is 1.68% higher than ResNet and 0.43% higher than Inception-v3. Under the conditions of simulated salt and pepper noise, Gaussian noise and Cauchy noise, the noise robustness of MWPC is significantly improved, which also reduces the mean noise error. The ablation experiment about network modules shows that the MWPC network can improve performance and noise robustness.

     

/

返回文章
返回