Feature Selection Strategies for Efficient Breast Cancer Classification

Read full text
Authors: Abe Jacob
Cited by: 0

Abstract

This project explores how selecting the right subset of features can improve the efficiency, accuracy, and interpretability of machine learning models for breast cancer diagnosis. Using the Breast Cancer Wisconsin (Diagnostic) dataset, the study analyzes how different groups of input features—such as mean values, standard errors, and worst-case measurements—affect classification performance when using a Random Forest classifier. The core of the project is a custom Python framework that allows flexible experimentation with various feature combinations. It enables users to systematically train and evaluate models on different subsets of the dataset’s 30 features, track accuracy, and visualize results. Through two carefully designed experiments, the study demonstrates that worst-case measurements and a small set of top-ranked morphological features (like radius, texture, perimeter, and area) can achieve near-maximum classification accuracy—often outperforming models trained on all 30 features. These findings highlight the potential of dimensionality reduction for building lightweight, interpretable diagnostic models in medical settings. All code and results are provided in a public GitHub repository to support reproducibility and future exploration.

Fields of Study & Subjects

Artificial Intelligence in Healthcare AI in cancer detection Gene expression and cancer classification

Publication details

Journal / source
Zenodo (CERN European Organization for Nuclear Research)
Publication date
10 January 2026
Document type
Preprint