Abstract:Accurate classification of breast cancer subtypes from gene expression data is critical for diagnosis and treatment selection. However, such datasets are characterized by high dimensionality and limited sample size, posing challenges for machine learning models.
In this study, we evaluate the impact of model complexity and feature selection on subtype classification performance using TCGA-BRCA gene expression data. Logistic regression, random forest, and support vector machine (SVM) models were trained using varying numbers of highly variable genes (50 to 20,518). Performance was evaluated using stratified 5-fold cross-validation and assessed with accuracy and macro F1 score. While all models achieved high accuracy, macro F1 analysis revealed substantial differences in subtype-level performance. Logistic regression demonstrated the most stable and balanced performance across subtypes, including improved detection of rare classes. Random forest underperformed on minority subtypes despite strong overall accuracy, while SVM showed sensitivity to feature dimensionality. These findings highlight the importance of model simplicity, evaluation metrics, and feature selection in high-dimensional biological classification tasks.
| Comments: | 8 pages, 4 figures, 3 tables. Independent research study using TCGA-BRCA RNA-seq data |
| Subjects: | Machine Learning (cs.LG); Genomics (q-bio.GN) |
| Cite as: | arXiv:2605.06562 [cs.LG] |
| (or arXiv:2605.06562v1 [cs.LG] for this version) | |
| https://doi.org/10.48550/arXiv.2605.06562 arXiv-issued DOI via DataCite (pending registration) |
From: Meena Al Hasani [view email]
[v1]
Thu, 7 May 2026 16:55:46 UTC (764 KB)
此内容由惯性聚合(RSS阅读器)自动聚合整理，仅供阅读参考。 原文来自 — 版权归原作者所有。