Comparing Machine Learning Techniques for Breast Cancer Prediction: A Comprehensive Study

This study conducted a comprehensive comparative analysis of machine learning (ML) techniques for predicting breast cancer risk, using the Breast Cancer Wisconsin (Di-agnostic) dataset. We explored the efficacy of 10 ML models and 3 feature selection methods along with hyperparameter optimization. Our primary focus of performance is recall, accuracy, and F1 measures. We obtained our metrics from cross validation for realistic performance predictions. ML models included support vector, decision tree, logistic regression, k-nearest neighbors, random forest, Gaussian Naïve Bayes, and 4 flavors of boosting ensemble algorithms. Feature selection techniques included univariate, recursive selections and principal component analysis. Our analysis provides valuable and quickly attainable insights into the practical application of ML techniques in clinical settings for breast cancer screening.

Authors

Institutions

Publication Details

Published
2024-11-20
DOI
https://doi.org/10.1109/fmlds63805.2024.00064
Primary Topic
AI in cancer detection
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Comparing Machine Learning Techniques for Breast Cancer Prediction: A Comprehensive Study

Zsolt Ugray, Tonmoy Roy
AI in cancer detection
article

Comparing Machine Learning Techniques for Breast Cancer Prediction: A Comprehensive Study

Zsolt Ugray, Tonmoy Roy
article en

Abstract

This study conducted a comprehensive comparative analysis of machine learning (ML) techniques for predicting breast cancer risk, using the Breast Cancer Wisconsin (Di-agnostic) dataset. We explored the efficacy of 10 ML models and 3 feature selection methods along with hyperparameter optimization. Our primary focus of performance is recall, accuracy, and F1 measures. We obtained our metrics from cross validation for realistic performance predictions. ML models included support vector, decision tree, logistic regression, k-nearest neighbors, random forest, Gaussian Naïve Bayes, and 4 flavors of boosting ensemble algorithms. Feature selection techniques included univariate, recursive selections and principal component analysis. Our analysis provides valuable and quickly attainable insights into the practical application of ML techniques in clinical settings for breast cancer screening.

Utah State University (US)
Openalex Percentile: Top 75%
AI in cancer detection
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.