Study for Performance of MobileNetV1 and MobileNetV2 based on Breast Cancer
DOI:
https://doi.org/10.54097/hset.v39i.6340Keywords:
MobileNetV1 and MobileNetV2; Breast Cancer; Artificial Intelligence.Abstract
Artificial intelligence is constantly evolving and can provide effective help in all aspects of people's lives. The experiment is mainly to study the use of artificial intelligence in the field of medicine. The purpose of this experiment was to compare which of MobileNetV1 and MobileNetV2 models was better at detecting histopathological images of the breast downloaded at Kaggle. When the doctor looks at the pathological image, there may be errors that lead to errors in judgment, and the observation speed is slow. Rational use of artificial intelligence can effectively reduce the error of doctor diagnosis in breast cancer judgment and speed up doctor diagnosis. The dataset was downloaded from Kaggle and then normalized. The basic principle of the experiment is to let the neural network model learn the downloaded data set. Then find the pattern and be able to judge on your own whether breast tissue is cancer. In the dataset, benign tumor pictures and malignant tumor pictures have been classified, of which 198738 are benign tumor pictures and 78, 786 are malignant tumor pictures. After calling MobileNetV1 and MobileNetV2, the dataset is trained separately, the training accuracy and validation accuracy rate are obtained, and the image is drawn. It can be observed that MobileNetV1 has better validation accuracy and overfit during MobileNetV2 training. From the experimental results, it can be seen that in the case of processing this dataset, MobileNetV1 is much better than MobileNetV2.
Downloads
References
Wild C P et al. 2020 World Cancer Report: Cancer research for cancer prevention Lyon: International Agency for Research on Cancer.
Tan X 2022 An exploratory study of deep learning supply chain In Proceedings of the 44th International Conference on Software Engineering p 86-98.
Qiu Y 2020 Improved denoising autoencoder for maritime image denoising and semantic segmentation of USV China Communications 17(3) p 46-57.
Spanhol F A et al. 2016 Breast cancer histopathological image classification using Convolutional Neural Networks International Joint Conference on Neural Networks (IJCNN) 2016: 2560-2567 International Joint Conference on Neural Networks (IJCNN) 2560-2567.
Zou W K et al. 2020 Histopathological image classification of breast cancer based on convolutional neural network Computer Engineering and Design, 41(6): 1749- 1754.
Tan Y J et al. 2017 Breast cancer detection using convolutional neural networks for mammogram imaging system 2017 International Conference on Robotics Automation and Sciences IEEE 1 – 5.
Zhang X et al. 2018 Classification of whole breast and tomosynthesis images using deep convolutional neural networks. IEEE Transactions on Nanobio science, 17(3): 237 - 242.
Kaggle 2017 Breast Histopathogloy Images https://www. kaggle.com/ datasets/ paultimothymooney/ breast-histopathology-images.
Barret Z et al. 2018 Learning Transferable Architectures for Scalable Image Recognition. ArXiv.org, https:// arxiv.org/abs/1707.07012.
Wang M et al. 2017 Design of Efficient Convolutional Layers Using Single Intra-Channel Convolution, Topological Subdivisioning and Spatial Bottleneck Structure ArXiv.org.
Downloads
Published
Issue
Section
License
This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.
How to Cite
