Attentive Octave Convolutional Capsule Network for Medical Image Classification

Applied Sciences

卷:12

Hao Zhao¹  Yanping Zhang²  Zan Li³  Hong Zhang³  Zhengzhen Li³ 

[1] Department of Computer Science and Technology, School of Information, Renmin University of China, Beijing 100872, China;

[2] Department of Computer Science, School of Engineering and Applied Science, Gonzaga University, Spokane, WA 99258, USA;

[3] School of Computer Science and Technology, Minzu University of China, Beijing 100081, China;

关键词: medical image classification;    capsule network;    octave convolution;    attention mechanism;   

DOI  :  10.3390/app12052634

来源: DOAJ

【 摘 要 】

Medical image classification plays an essential role in disease diagnosis and clinical treatment. More and more research efforts have been dedicated to the design of effective methods for medical image classification. As an effective framework, the capsule network (CapsNet) can realize translation equivariance. Lots of current research applies capsule networks in medical image analysis. In this paper, we propose an attentive octave convolutional capsule network (AOC-Caps) for medical image classification. In AOC-Caps, an AOC module is used to replace the traditional convolution operation. The purpose of the AOC module is to process and fuse the high- and low-frequency information in the input image simultaneously, and weigh the important parts automatically. Following the AOC module, a matrix capsule is used and the expectation maximization (EM) algorithm is applied to update the routing weights. The proposed AOC-Caps and comparative methods are tested on seven datasets, including PathMNIST, DermaMNIST, OCTMNIST, PneumoniaMNIST, OrganMNIST_Axial, OrganMNIST_Coronal, and OrganMNIST_Sagittal, which are from MedMNIST. In the experiments, baselines include the traditional CNN models, automated machine learning (AutoML) methods, and related capsule network methods. The experimental results demonstrate that the proposed AOC-Caps achieves better performance on most of the seven medical image datasets.

【 授权许可】

Unknown