Skip to content

A Stable and Reliable Capsules Network for Image classification

Notifications You must be signed in to change notification settings

Woodman718/CapsNets

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

41 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Experimental equipment

The proposed method is implemented using PyTorch. All experiments were conducted on a tower workstation equipped with an Intel Core i5-11400KF and an NVIDIA GeForce RTX 3070.

Results

  1. Evaluation metrics on the HAM10000 (Augment).
Evaluation metricsComparison with other methods
Type Precision Recall F1 Accuracy
akiec 0.992 0.996 0.994
bcc 0.9896 0.9961 0.9929
bkl 0.9934 0.9882 0.9908
df 0.9981 1.0 0.9991
mel 0.9858 0.9897 0.9877
nv 1.0 0.9881 0.994
vasc 1.0 1.0 1.0
overall: 0.9941 0.994 0.9941 0.9937
Method Accuracy [%] Params(M) FLOPs(G)
Inception V3 92.10 22.80 5.73
ResNet 50 92.31 25.60 4.10
DenseNet-201 92.87 20.01 4.28
IRv2-SA 93.47 47.5 25.46
IM-CNN 95.10 - -
Proposed(Ours) 99.37 1.41 2.74

2 Evaluation metrics on the HAM10000.

1). Evaluation metrics and LCK

Evaluation metrics LKC(large-kernel convolution)
Type Precision Recall F1 Accuracy
akiec 1.0 0.9394 0.9687
bcc 0.8983 1.0 0.9464
bkl 0.9444 0.9027 0.9231
df 0.8 0.7273 0.7619
mel 0.8872 1.0 0.9402
nv 0.9954 0.9714 0.9832
vasc 0.8235 1.0 0.9032
overall: 0.907 0.9344 0.9181 0.9652

LKC

LKC with different kernel sizes.  
The N of the label ”kernel-N” indicates the size of the convolution kernel.  
For instance, kernel-21 means using an LKC with a 21×21 convolution kernel.

2). Attention.

Atten

3 Generalization Performance

Dataset:  https://www.kaggle.com/datasets/tawsifurrahman/covid19-radiography-database
The COVID-19 Radiography Database consisted of 21165 images.
Among them, covid(3616),normal(10192),opacity(6012),viral(1345).
Evaluation MetricsDistribution of the COVID-19 Radiography Dataset
Type Precision Recall F1 Accuracy
covid 0.9972 1.0 0.999
normal 0.999 0.996 0.998
opacity 0.995 0.997 0.996
viral 0.9926 1.0 0.996
overall: 0.9972

dis_data

Source Data: http://dx.doi.org/10.5281/zenodo.1214456
Jakob Nikolas Kather, Johannes Krisam, et al., "Predicting survival from colorectal cancer histology slides using deep learning: A retrospective multicenter study," PLOS Medicine, vol. 16, no. 1, pp. 1–22, 01 2019.
This is a slightly different version of the "NCT-CRC-HE-100K" image set: This set contains 100,000 images in 9 tissue classes at 0.5 MPP and was created from the same raw data as "NCT-CRC-HE-100K". 
However, no color normalization was applied to these images. Consequently, staining intensity and color slightly varies between the images. Please note that although this image set was created from the same data as "NCT-CRC-HE-100K", the image regions are not completely identical because the selection of non-overlapping tiles from raw images was a stochastic process.
Evaluation MetricsNCT-CRC-HE-100K-NONORM
Type Precision Recall F1 Accuracy
ADI 1.0 1.0 1.0
BACK 1.0 1.0 1.0
DEB 1.0 1.0 1.0
LYM 1.0 0.998 0.999
MUC 0.9978 0.998 0.998
MUS 0.9985 0.999 0.999
NORM 0.9989 1.0 0.999
STR 0.999 0.997 0.998
TUM 0.9979 0.999 0.999
overall: 0.9991

dis_data

Dataset

Data

The distribution of the seven disease types before and after data augmentation. In the clusters of bars with the same color, the left bar represents the sample distribution after data augmentation, while the right bar represents the initial distribution of the dataset.

Example of Skin lesions in HAM10000 dataset.
Among them, BKL, DF, NV, and VASC are benign tumors, whereas AKIEC, BCC, and MEL are malignant tumors.

Available:
https://challenge.isic-archive.com/data/#2018
https://dataverse.harvard.edu/dataset.xhtml?persistentId=doi:10.7910/DVN/DBW86T
https://aistudio.baidu.com/aistudio/datasetdetail/218024 (ours)

Citation:
P. Tschandl, C. Rosendahl, and H. Kittler, “The ham10000 dataset,a large collection of multi-source dermatoscopic images of common pigmented skin lesions,” Scientific data, vol. 5, no. 1, pp. 1–9, 2018.

License

The dataset is released under a Creative Commons Attribution 4.0 License. For more information, see https://creativecommons.org/licenses/by/4.0/ .

Related Work

a.

@article{WangIMCC,
  author={Wang, Sutong and Yin, Yunqiang and Wang, Dujuan and Wang, Yanzhang and Jin, Yaochu},
  journal={IEEE Transactions on Cybernetics}, 
  title={Interpretability-Based Multimodal Convolutional Neural Networks for Skin Lesion Diagnosis}, 
  year={2022},
  volume={52},
  number={12},
  pages={12623-12637},
  doi={10.1109/TCYB.2021.3069920}
}

b

@article{xia2017exploring,
  title={Exploring Web images to enhance skin disease analysis under a computer vision framework},
  author={Xia, Yingjie and Zhang, Luming and Meng, Lei and Yan, Yan and Nie, Liqiang and Li, Xuelong},
  journal={IEEE Transactions on Cybernetics},
  volume={48},
  number={11},
  pages={3080--3091},
  year={2017},
  publisher={IEEE}
}

Citation

If you use our method for your research or aplication, please consider citation:

@ARTICLE{LanCapsNets,
  author={Lan, Zhangli and Cai, Songbai and Zhu, Jiqiang and Xu, Yuantong},
  journal={XXX on XXX}, 
  title={A Novel Skin Cancer Assisted Diagnosis Method based on Capsule Networks with CBAM}, 
  year={},
  volume={},
  number={},
  pages={},
  doi={10.36227/techrxiv.23291003},
}

@ARTICLE{9791221,
  author={Lan, Zhangli and Cai, Songbai and He, Xu and Wen, Xinpeng},
  journal={IEEE Access}, 
  title={FixCaps: An Improved Capsules Network for Diagnosis of Skin Cancer}, 
  year={2022},
  volume={10},
  number={},
  pages={76261-76267},
  doi={10.1109/ACCESS.2022.3181225}
}

About

A Stable and Reliable Capsules Network for Image classification

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published