过度拟合
计算机科学
卷积神经网络
人工智能
一般化
人工神经网络
模式识别(心理学)
情态动词
班级(哲学)
卷积(计算机科学)
数学
数学分析
化学
高分子化学
作者
Debendra Muduli,Ratnakar Dash,Banshidhar Majhi
标识
DOI:10.1016/j.bspc.2021.102825
摘要
This paper proposes a deep convolutional neural network (CNN) model for automated breast cancer classification from a different class of images, namely, mammograms and ultrasound. The model contains only five learnable layers: four convolutional layers and a fully connected layer. The model facilitates extracting prominent features automatically from the images with a smaller number of tunable parameters. Exhaustive simulation results on mammograms dataset, namely, MIAS, DDSM, and INbreast, as well as ultrasound datasets, namely, BUS-1 and BUS-2, depict that the suggested model outperforms the recent state-of-the-art schemes. Data augmentation technique has been employed to reduce overfitting and provide good generalization. The proposed CNN model achieves an accuracy of 96.55%, 90.68%, and 91.28% on MIAS, DDSM, and INbreast datasets, respectively. Similarly, the accuracies obtained are 100% and 89.73% on BUS-1 and BUS-2 datasets, respectively.
科研通智能强力驱动
Strongly Powered by AbleSci AI