A rotation meanout network with invariance for dermoscopy image classification and retrieval. (December 2022)

• Record Type: Journal Article
• Authors: Zhang, Yilan; Xie, Fengying; Song, Xuedong; Zhou, Hangning; Yang, Yiguang; Zhang, Haopeng; Liu, Jie
• Is Part Of: Computers in biology and medicine. Volume 151:Part A(2022)
• Publication Date: 2022-12
• Subjects: Dermoscopy -- Image classification -- Image retrieval -- Rotation invariance -- Convolutional neural networks; Medicine -- Data processing -- Periodicals; Biology -- Data processing -- Periodicals; 610.285
• DOI: 10.1016/j.compbiomed.2022.106272
• Languages: English
• ISSNs: 0010-4825

Abstract

Abstract: The computer-aided diagnosis (CAD) system can provide a reference basis for the clinical diagnosis of skin diseases. Convolutional neural networks (CNNs) can not only extract visual elements such as colors and shapes but also semantic features. As such they have made great improvements in many tasks of dermoscopy images. The imaging of dermoscopy has no principal orientation, indicating that there are a large number of skin lesion rotations in the datasets. However, CNNs lack rotation invariance, which is bound to affect the robustness of CNNs against rotations. To tackle this issue, we propose a rotation meanout (RM) network to extract rotation-invariant features from dermoscopy images. In RM, each set of rotated feature maps corresponds to a set of outputs of the weight-sharing convolutions and they are fused using meanout strategy to obtain the final feature maps. Through theoretical derivation, the proposed RM network is rotation-equivariant and can extract rotation-invariant features when followed by the global average pooling (GAP) operation. The extracted rotation-invariant features can better represent the original data in classification and retrieval tasks for dermoscopy images. The RM is a general operation, which does not change the network structure or increase any parameters, and can be flexibly embedded in any part of CNNs. Extensive experiments are conducted on a dermoscopy image dataset. The results show that our method outperforms other anti-rotation methods and achieves great improvements in skin disease classification and retrieval tasks, indicating the potential of rotation invariance in the field of dermoscopy images. Highlights: A rotation meanout (RM) network is proposed to extract rotation-invariant features. RM can be flexibly embedded in convolutional neural networks (CNNs). The proposed method does not change the network structure or increase the parameters. RM improves the performance of CNNs in skin disease classification and retrieval.