The Effect of Batch Size on the Performance MobileNet-V3 Model for Breast Cancer Classification of Ultrasound and Mammography Images

Many hyperparameters must be adjusted for a Convolutional Neural Network (CNN) to classify pictures accurately. The batch size or the total number of images used to train the forward and backward routes, is one of the most significant hyperparameters. The purpose of this work is to examine how batch size affects Convolutional Neural Networks (CNN) ability to classify images, particularly medical photos that show breast cancer. In this experiment, the MobileNet-V3 model is employed to train the network more quickly. The testing on ultrasound images highest level of accuracy was 98.05%, loss of 0.0496 on training data and a 84.00% accuracy, a loss 0.4772 on validation data with batch size 256. Testing on mammography data the best accuracy, with a value train accuracy of 94.97% and a loss of 0.1265, and validation data accuracy of 90.34% and a loss of 0.3177 with batch size 32

Authors

Institutions

Publication Details

Published
2024-08-29
DOI
https://doi.org/10.1109/icitacee62763.2024.10762760
Citations
2
Primary Topic
AI in cancer detection
Type
article
Field-Weighted Citation Impact
0.69
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

The Effect of Batch Size on the Performance MobileNet-V3 Model for Breast Cancer Classification of Ultrasound and Mammography Images

Anny Yuniarti, Chastine Fatichah, Muhammad Imron Rosadi
2 citations
AI in cancer detection
0.69
article

The Effect of Batch Size on the Performance MobileNet-V3 Model for Breast Cancer Classification of Ultrasound and Mammography Images

Anny Yuniarti, Chastine Fatichah, Muhammad Imron Rosadi
article en
2 citations

Abstract

Many hyperparameters must be adjusted for a Convolutional Neural Network (CNN) to classify pictures accurately. The batch size or the total number of images used to train the forward and backward routes, is one of the most significant hyperparameters. The purpose of this work is to examine how batch size affects Convolutional Neural Networks (CNN) ability to classify images, particularly medical photos that show breast cancer. In this experiment, the MobileNet-V3 model is employed to train the network more quickly. The testing on ultrasound images highest level of accuracy was 98.05%, loss of 0.0496 on training data and a 84.00% accuracy, a loss 0.4772 on validation data with batch size 256. Testing on mammography data the best accuracy, with a value train accuracy of 94.97% and a loss of 0.1265, and validation data accuracy of 90.34% and a loss of 0.3177 with batch size 32

Sepuluh Nopember Institute of Technology (ID)
Good health and well-being
Openalex Percentile: Top 23%
AI in cancer detection
0.69
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.

The Effect of Batch Size on the Performance MobileNet-V3 Model for Breast Cancer Classification of Ultrasound and Mammography Images — Anny Yuniarti, Chastine Fatichah, et al. · (2024) | TGRS Research Map | TGRS