Paper
21 June 2024 Semi-supervised medical image classification based on DenseNet and capsule network combination model
Hao Shen, Jiaqing Mo, Jinxun Ren
Author Affiliations +
Proceedings Volume 13167, International Conference on Remote Sensing, Mapping, and Image Processing (RSMIP 2024); 131672H (2024) https://doi.org/10.1117/12.3029635
Event: International Conference on Remote Sensing, Mapping and Image Processing (RSMIP 2024), 2024, Xiamen, China
Abstract
Medical image classification plays a crucial role in clinical treatment and medical education. However, traditional classification methods have reached performance limits and require substantial time and effort for feature extraction and selection. In this study, we introduce a novel deep learning model that combines the feature extraction part of DenseNet with the dynamic routing mechanism of capsule networks. DenseNet, known for its dense connections, aids in gradient propagation and feature reuse. Capsule networks introduce capsule layers with sensitivity to spatial relationships. By merging these two structures, our aim is to enhance the model's ability to model object hierarchy and pose variations. Our approach initially incorporates the dynamic routing mechanism of capsule networks on top of DenseNet, achieving an effective fusion of the two structures. Experimental results on the Chest X-Ray14 dataset demonstrate a significant improvement in accuracy and robustness compared to traditional DenseNet and capsule networks. To address the issue of limited annotation in medical image data, we introduce a semi-supervised learning approach and improve training balance through anti-curriculum pseudo-labeling. Furthermore, to obtain more accurate pseudo-labels, we adopt a mixed model for pseudo-label generation. This method not only effectively balances the pseudo-label predictions generated by two models but also mitigates the challenges associated with threshold-based confidence intervals in the context of multi-class, multi-label scenarios. In summary, the combination of deep convolutional neural networks and capsule networks, along with semi-supervised learning and anti-curriculum pseudo-labeling, holds great promise for medical image classification. It has significant application value in clinical practice and presents a wide-ranging prospect for the future.
(2024) Published by SPIE. Downloading of the abstract is permitted for personal use only.
Hao Shen, Jiaqing Mo, and Jinxun Ren "Semi-supervised medical image classification based on DenseNet and capsule network combination model", Proc. SPIE 13167, International Conference on Remote Sensing, Mapping, and Image Processing (RSMIP 2024), 131672H (21 June 2024); https://doi.org/10.1117/12.3029635
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Data modeling

Education and training

Medical imaging

Image classification

Performance modeling

Statistical modeling

Feature extraction

Back to Top