.png){kind=icon}
返回
Using Segmentation to Boost Classification Performance and Explainability in CapsNets
.png){kind=icon}
OA
AI
DOI：10.3390/make6030068.png)
.png){kind=icon}
原文求助.png){kind=icon}
分享.png){kind=icon}
收藏
摘要
En 中文
In this paper, we present Combined-CapsNet (C-CapsNet), a novel approach aimed at enhancing the performance and explainability of Capsule Neural Networks (CapsNets) in image classification tasks. Our method involves the integration of segmentation masks as reconstruction targets within the CapsNet architecture. This integration helps in better feature extraction by focusing on significant image parts while reducing the number of parameters required for accurate classification. C-CapsNet combines principles from Efficient-CapsNet and the original CapsNet, introducing several novel improvements such as the use of segmentation masks to reconstruct images and a number of tweaks to the routing algorithm, which enhance both classification accuracy and interoperability. We evaluated C-CapsNet using the Oxford-IIIT Pet and SIIM-ACR Pneumothorax datasets, achieving mean F1 scores of 93% and 67%, respectively. These results demonstrate a significant performance improvement over traditional CapsNet and CNN models. The method's effectiveness is further highlighted by its ability to produce clear and interpretable segmentation masks, which can be used to validate the network's focus during classification tasks. Our findings suggest that C-CapsNet not only improves the accuracy of CapsNets but also enhances their explainability, making them more suitable for real-world applications, particularly in medical imaging.
Keyword:
capsule network
explainability
image classification
reconstruction
segmentation
期刊
M
IF:
6
论文数:
841
被引数:
1.8K
机构
引用论文
Detection of Non-Stationary GW Signals in High Noise From Cohen's Class of Time-Frequency Representations Using Deep Learning.png)
IEEE ACCESSOAAI
IEEE ACCESSIF3.6
OAAI分享收藏
分享收藏分享收藏
查看更多内容