A Fast and Interpretable Logistic Regression Framework for Breast Tumor Classification Using the Wisconsin Diagnostic Dataset
Cheng, W.; Yu, Z.
Show abstract
Early and reliable discrimination between malignant and benign breast tumors is essential for clinical decision-making and for reducing unnecessary invasive procedures. This study presents a lightweight and reproducible machine-learning pipeline that integrates standard feature normalization with logistic regression to classify breast tumors using the Breast Cancer Wisconsin (Diagnostic) dataset (WDBC), which contains 569 samples described by 30 quantitative features derived from digitized fine-needle aspirate (FNA) images [1-3]. We implemented an end-to-end workflow in Python (scikit-learn), including stratified train-test splitting, model training, and evaluation with clinically meaningful metrics such as accuracy, sensitivity, specificity, and ROC-AUC [4-6]. On the held-out test set (n=114), the proposed approach achieved 98.25% accuracy and a ROC-AUC of 0.9954, with a confusion matrix indicating only two misclassifications (1 malignant predicted as benign, 1 benign predicted as malignant). Specifically, when treating malignant cases as the clinically critical positive class, the method yielded 97.62% sensitivity and 98.61% specificity. These results demonstrate that a simple, interpretable model can achieve near state-of-the-art performance on structured biomedical features while remaining computationally efficient and suitable for rapid prototyping.In addition, we benchmarked logistic regression against several classical baselines (SVM with RBF kernel, random forest, and kNN) under the same train-test split and evaluation protocol. The results indicate that increasing model complexity yields limited performance gains on WDBC, suggesting that an interpretable linear classifier can already approach the performance ceiling on this feature-engineered dataset. Future work will focus on external validation, calibration for risk estimation, and multimodal extensions to incorporate imaging or omics signals.
Matching journals
The top 7 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- A Machine Learning Ensemble Based on Radiomics to Predict BI-RADS Category and Reduce the Biopsy Rate of Ultrasound-Detected Suspicious Breast Masses 94%
- Demarcation line determination for diagnosis of gastric cancer disease range using unsupervised machine learning in magnifying narrow-band imaging 92%
- An Explainable Web-Based Diagnostic System for Alzheimer's Disease Using XRAI and Deep Learning on Brain MRI 91%
Similar papers in this journal
- Classification performance bias between training and test sets in a limited mammography dataset 95%
- Enhancing Breast Ultrasound Segmentation through Fine-tuning and Optimization Techniques: Sharp Attention UNet 94%
- Automated Detection of COVID-19 through Convolutional Neural Network using Chest x-ray images 94%
Similar papers in this journal
- On evaluation metrics for medical applications of artificial intelligence 95%
- Automated and Manual Quantification of Tumour Cellularity in Digital Slides for Tumour Burden Assessment 95%
- Mitigating Machine Learning Bias Between High Income and Low-Middle Income Countries for Enhanced Model Fairness and Generalizability 94%
Similar papers in this journal
- piNET: An Automated Proliferation Index Calculator Framework for Ki67 Breast Cancer Images 97%
- Breast invasive ductal carcinoma classification on whole slide images with weakly-supervised and transfer learning 95%
- From Variability to Standardization: The Impact of Breast Density on Background Parenchymal Enhancement in Contrast-Enhanced Mammography and the Need for a Structured Reporting System 91%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.