Next Article in Journal
Modified Significance Analysis of Microarrays in Heterogeneous Diseases
Next Article in Special Issue
Deep Learning for Integrated Analysis of Insulin Resistance with Multi-Omics Data
Previous Article in Journal
The Impact of Air Pollution Exposure on the MicroRNA Machinery and Lung Cancer Development
Previous Article in Special Issue
Validation of the Italian Version of the Educational Needs Assessment Tool in Rheumatoid Arthritis Patients and Factors Associated with Educational Needs
 
 
Font Type:
Arial Georgia Verdana
Font Size:
Aa Aa Aa
Line Spacing:
Column Width:
Background:
Article

Breast Cancer Type Classification Using Machine Learning

Department of Genetics, School of Medicine, Louisiana State University Health Sciences Center, 533 Bolivar, New Orleans, LA 70112, USA
*
Author to whom correspondence should be addressed.
J. Pers. Med. 2021, 11(2), 61; https://doi.org/10.3390/jpm11020061
Submission received: 23 December 2020 / Revised: 12 January 2021 / Accepted: 15 January 2021 / Published: 20 January 2021
(This article belongs to the Special Issue Use of Clinical Decision Support Software within Health Care Systems)

Abstract

:
Background: Breast cancer is a heterogeneous disease defined by molecular types and subtypes. Advances in genomic research have enabled use of precision medicine in clinical management of breast cancer. A critical unmet medical need is distinguishing triple negative breast cancer, the most aggressive and lethal form of breast cancer, from non-triple negative breast cancer. Here we propose use of a machine learning (ML) approach for classification of triple negative breast cancer and non-triple negative breast cancer patients using gene expression data. Methods: We performed analysis of RNA-Sequence data from 110 triple negative and 992 non-triple negative breast cancer tumor samples from The Cancer Genome Atlas to select the features (genes) used in the development and validation of the classification models. We evaluated four different classification models including Support Vector Machines, K-nearest neighbor, Naïve Bayes and Decision tree using features selected at different threshold levels to train the models for classifying the two types of breast cancer. For performance evaluation and validation, the proposed methods were applied to independent gene expression datasets. Results: Among the four ML algorithms evaluated, the Support Vector Machine algorithm was able to classify breast cancer more accurately into triple negative and non-triple negative breast cancer and had less misclassification errors than the other three algorithms evaluated. Conclusions: The prediction results show that ML algorithms are efficient and can be used for classification of breast cancer into triple negative and non-triple negative breast cancer types.

1. Introduction

Despite remarkable progress in screening and patient management, breast cancer (BC) remains the second most diagnosed and the second leading cause of cancer deaths among women in the United States [1,2]. According to the American Cancer Association, there were 268,600 women newly diagnosed with BC in 2019, of which 41,760 died from the disease [1,2]. BC is a highly heterogeneous disease encompassing multiple types and many subtypes [3,4]. The majority of BCs respond to endocrine and targeted therapies, and generally have good prognosis and survival rates [3,4]. However, a significant proportion of BC are triple negative breast cancers (TNBC) [4,5]. TNBC is a specific subtype of BC characterized by lack of expression of the three most targeted biomarkers in BC treatment: estrogen receptor (ER), progesterone receptor (PR), and human epidermal growth factor receptor (HER-2) [2,6]. It accounts for 15% to 20% of all BCs diagnosed annually [4]. TNBC tumors are characterized by a more aggressive clinical behavior, poor prognosis, higher recurrence rates and poor survival rates [7,8,9,10,11,12,13,14]. Currently, there are no Food and Drug Administration (FDA) approved targeted therapies for this dreadful disease. Cytotoxic chemotherapy remains the main effective therapeutic modality, although some patients develop resistance and many others who survive surfer many side effects [15]. The long-term side effects of chemotherapy are well-known and include infertility, osteopenia and osteoporosis, heart damage and in rare cases leukemia, as well as financial losses, all of which can severely impact the quality of life for the survivors [15]. Thus, there is an urgent need for the development of accurate algorithms for identifying and distinguishing truly TNBC tumors which could be prioritized for specialized treatment from non-TNBC tumors that can be safely treated using endocrine or targeted therapeutics.
Traditionally, classification of breast cancer patients into those with TNBC and non-TNB has been largely determined by immunohistochemical staining [16,17]. Discordance in assessment of tumor biomarkers by histopathological assays has been reported [16]. Recently, Viale et al. compared immunohistochemical (IHC) versus molecular subtyping using molecular BluePrint and MammaPrint in a population of patients enrolled in MINDACT [17]. These authors also compared outcome based on molecular subtyping (MS) versus surrogate pathological subtyping (PS) as defined by the 2013 St. Gallen guidelines [18]. They discovered and concluded that molecular classification can help to identify a larger group of patients with low risk of recurrence compared with the more contemporarily used classification methodology including high-quality assessed Ki67 [16,17]. Moreover, while traditional classification methods have been relatively effective, they lack the accuracy and specificity to identify those breast cancers that are truly TNBC from non-TNBC. Therefore, novel approaches are needed to address this critical unmet need.
BC screening in the United States has been routinely performed with mammography, digital breast tomosynthesis, ultrasound and magnetic resonance [19,20,21]. These breast imaging modalities for BC screening have resulted in a new and growing field of radiomics [19,20]. Radiomics analysis using contrast-enhanced spectral mammography images in BC diagnosis has revealed that textural features could provide complementary information about the characterization of breast lesions [20]. Radiomics has also been used in BC classification and prediction [21]. However, molecular classification of BC into TNBC and non-TNBC has received little attention. Given that TNBC tends to affect younger premenopausal women who are not recommended for screening using mammography, there is a need for the development of new classification algorithms.
Recently, the application of machine learning (ML) to molecular classification of tumors has come into sharper focus [22,23,24]. ML methods have been applied to breast cancer survival prediction [22], for diagnostic ultrasound of TNBC [23] and breast cancer outcome prediction with tumor tissue images [24]. However, to date, ML has not been applied to classification of patients with TNBC and non-TNBC using RNA-sequence (gene expression) data. The objective of this study was to investigate the potential for application of ML to classification of BC into TNBC and non-TNBC using RNA-Sequence data derived from the two patient populations. Our working hypothesis was that genomic alterations in patients diagnosed with TNBC tumors and non-TNBC tumors could lead to measurable changes enabling classification of the two patient groups. We addressed this hypothesis by evaluating the performance of four ML algorithms using publicly available data on TNBC and non-TNBC from The Cancer Genome Atlas (TCGA) [25].

2. Materials and Methods

The overall design and execution strategy used in this study is presented in Figure 1. Below we provide a detailed description of the sources of gene expression variation data along with clinical data used in this investigation, as well as the data processing and analysis strategies used.

2.1. Source of Gene Expression Data

We used publicly available RNA-Seq data on TNBC and non-TNBC from The Cancer Genome Atlas (TCGA) [25]. Gene expression data and clinical information were downloaded from the Genomics Data Commons (GDC) using the data transfer tool [26]. The data set included 1222 samples and 60,485 probes. Using the sample barcodes, we linked the gene expression data with molecular data and ascertained the samples as either TNBC or non-TNBC. Samples without clinical phenotyping or labels were excluded from the data sets and were not included in downstream analysis. We performed quality control (QC) and noise reduction on the original gene expression data matrix to remove rows with insufficient information or missing data. Due to the large difference in gene expression values, in order to facilitate later modeling and rapid training convergence, we normalized the expression profile data. The QCed data set was normalized using the LIMMA [27] and edgeR Bioconductor package implemented in R [27]. The probe IDs were matched with gene symbols using the Ensemble database. In our analyses, we used counts per million reads (CPM) and log-CPM. CPM and log-CPM values were calculated using a counts matrix alone and have been successfully used in RNA-Seq data processing [28]. After data processing and QC, the final data set used in downstream analysis consisted of 934 tumor samples distributed as 116 TNBC and 818 non-TNBC samples, and 57,179 probes. The probes were matched with gene symbols using the Ensemble database [29].

2.2. Differential Gene Expression Analysis and Feature Selection

The classification approach proposed in this article is a binary classification model. However, because of the large number of genes (herein called features) involved, which was much larger than the number of samples, the correlation between features was relatively complex, and the dependence between correlations was affected. This presented challenges in the application of ML. For example, with high dimensionality of the data, it takes a long time to analyze the data, train the model and identify the best classifiers. Therefore, as a first step, we addressed the data dimensionality problem to overcome the influence of unfavorable factors and improve the accuracy of feature selection. To address this need, we used various statistical methods.
Using a quality controlled normalized data set, we performed supervised analysis comparing gene expression levels between TNBC and non-TNBC samples to discover a set of significantly differentially expressed genes between TNBC and non-TNBC. For this differential expression analysis, we used the LIMMA package implemented in R [27]. We used the false discovery rate (FDR) procedure to correct for multiple hypothesis testing [30]. In addition, we calculated the log2 Fold Change (Log2 FC), defined as the median of gene expressed minus the gene expression value for each gene. Genes were ranked on FDR adjusted p-values and Log2 FC. Significantly (p < 0.05) differentially expressed genes were identified and selected. For feature selection, we used significantly differentially expressed genes between the two types of breast cancer as the features. These features were selected at different threshold levels.

2.3. Modeling Prediction and Performance Evaluation

As noted above, the research content of this paper was based on a binary classification model with application to pattern recognition classification problem [31]. Under this approach 90% of the data set was randomly selected as the training set and the remaining 10% as the test set. There are many methods for performing classification tasks [32], including Logistic Regression, Nearest Neighbor, Naïve Bayes, Support Vector Machine, Decision Tree Algorithm and Random Forests Classification [32]. In this investigation, we evaluated four methods for performance, including, Support Vector Machines (SVM), K-nearest neighbor (kNN), Naïve Bayes (NGB) and Decision tree (DT).
The basic model for Support Vector Machine is to find the best separation hyperplane in the feature space to maximize the interval between positive and negative samples on the training set. SVM is a supervised learning algorithm used to solve two classification problems. The K-nearest neighbor classification algorithm is a theoretically mature method and one of the simplest machine learning algorithms. The idea of this method is in the feature space, if most of the k nearest (i.e., the nearest neighbors in the feature space) samples near a given sample belong to a certain category, that sample also belongs to this category. Naïve Bayes is a generative model of supervised learning. It is simple to implement, has no iteration, and has high learning efficiency. It will perform well in a large sample size. However, because the assumption is too strong (assuming that the feature conditions are independent), it is not applicable in scenarios where the feature conditions of the input vector are related. Decision Tree is based on the known probability of occurrence of various situations by constructing a decision tree to obtain the probability that the expected value of the net present value is greater than or equal to zero, evaluate project risk, and determine its feasibility. DT is a graphical method of intuitive use for probability analysis.
The methods were evaluated for performance to identify the best performing algorithm, which was further evaluated. For each method, we repeated the modeling process 10 times and used a confusion matrix (CM) [33] to display the classification results. Due to the small data sets used, we performed a 10-fold cross-validation evaluation of the classification performance of the methods we tested to validate their performance. We also computed accuracy, sensitivity and specificity and used them as performance measures for comparing the four classification algorithms employed.
For evaluation and comparison of the classification and misclassification performance of the four ML algorithms, we used 4 different scenarios in which any sample could end up or fall into: (a) true positive (TP) which means the sample was predicted as TNBC and was the correct prediction; (b) true negative (TN) which means the sample was predicted as non-TNBC and this was the correct prediction; (c) false positive (FP) which means the sample was predicted as TNBC, but was non-TNBC, and (d) false negative (FN) which means the sample was predicted as non-TNBC, but was TNBC. Using this information, we evaluated the classification results of the model by calculating the overall accuracy, sensitivity, specificity, precision, and F1 Score indicators. These performance measures or indicators were defined and computed as follows:
Accuracy = (TP + TN)/(TP + FP + FN + TN).
Recall = TP/(TP + FN)
Specificity = TN/(TN + FP)
Precision = TP/(TP + FP)
F1 Score = 2 * (Recall * Precision)/(Recall + Precision)
To further validate the methods, the classification results were also compared with classic feature selection methods such as SVM-RFE [34], ARCO [35], Relief [36] and mRMR [37]. The SVM-REF relies on constructing feature ranking coefficients based on the weight vector generated by SVM during training. Under this approach, a feature with the smallest ranking coefficient in each iteration is removed, until finally obtaining a descending ranking of all feature attributes. Area under the Receiver Operating Characteristic Curve (AUC) has been commonly used by the machine learning community in feature selection. The Relief algorithm is a feature weighting algorithm, which assigns different weights to features according to the correlation of each feature and category, and features whose weight are less than a certain threshold are removed. The mRMR algorithm was used to ensure the maximum correlation while removing redundant features, which is equivalent to obtaining a set of “purest” feature subsets. This is particularly useful when the features are very different. For implementation of classification models using ML algorithms and performance measurements, we used the Waikato Environment for Knowledge Analysis (WEKA) [38], an open source implemented in the Java-based framework.

3. Results

3.1. Result of Differential Expression and Feature Selection

The objective of this investigation was to identify a set of significantly (p < 0.05) differentially expressed genes that could distinguish TNBC from non-TNBC, and could be used as features for developing algorithms for classification of the two types of BC. We hypothesized that genomic alterations in women diagnosed with TNBC and those diagnosed with non-TNBC could lead to measurable changes distinguishing the two types of BC. To address this hypothesis, we performed whole transcriptome analysis comparing gene expression levels between TNBC and non-TNBC. The genes were ranked based on estimates of p-values and logFC. Only significantly (p < 0.05) differentially expressed genes with a high logFC identified after correcting for multiple hypothesis testing were selected and used as features in model development and validation. Note that all the estimates of the p-values were adjusted for multiple hypothesis testing using the false discovery rate procedure [30]. The analysis produced a signature of 5502 significantly (p < 0.05, |logFC| > 1) differentially expressed genes distinguishing patients with TNBC from non-TNBC. A summary of the results showing the top 30 most highly significantly differentially expressed genes along with estimates of p-value and logFC are presented in Table 1. A complete list of all the 5502 significantly (p < 0.05, |logFC| > 1) differentially expressed genes is presented in Supplementary Table S1.

3.2. Result of Classification

The objective of this investigation was to develop a classification algorithm based on ML that could accurately identify genes distinguishing truly TNBC from non-TNBC. The rationale was that molecular based classification using ML algorithms could provide a framework to accurately identify women at high risk of developing TNBC that could be prioritized for specialized treatment. To address this need, we evaluated the performance of four classification algorithms using the 5502 significantly differentially expressed genes identified from differential gene expression analysis using different threshold levels (p-values). The evaluated classifiers included the kNNs, NGB, DT and SVM. Each of these classifiers was modeled 10 times. Each algorithm was evaluated for accuracy, sensitivity/recall and specificty, computed as averages of the number of times each was modeled. The results showing accuracy, recall and specificity for the four classification algorithms computed as averages are shown in Table 2.
Among the four classification algorithms evaluated, SVM had the best performance with an accuracy of 90%, a recall of 87% and a specificty of 90%, followed by KNN, with an accuracy of 87%, a recall of 76 and specificty of 88%. Although NGB and DT were relatively accurate, they performed badly on recall. The variability in the evaluation parameters can be partially explained by the large numbers of features used and the unbalanced study design.
As noted above, the large number of features (5502 genes) can affect the performance of the classification algorithms. Therefore, to determine the optimal performance of each algorithm, we performed addition tests on the algorithms using smaller numbers of genes selected using different threshold levels. Under this approach the 5502 genes were ranked on FDR adjusted p-values. We selected the top 200, 150, 100 and 50 genes for use in the performance evaluation of each model using the same parameters as above, accuracy, recall and specificity. For each set of genes, we tested the performance of all four algorithms. The results of this investigation are presented in Figure 2 with plots showing the performance of each model under a specified number of genes plotted as a function of sample size. In the figure the x-axis accuracy shows the sample size and y-axis shows the accuracy.
The results show that the performance of each algorithm as function of sample size was relatively consistent. The performance of all classification algorithms increased with increasing sample size (Figure 2). No single classification technique proved to be significantly superior to all others in the experiments we performed (Figure 2). This can partially be explained by the small samples used in the investigation and the unbalanced design of the study project. In general, the plot showed that the SVM algorithm was better than the other three algorithms at higher sample sizes, i.e., greater than 50 (Figure 2). The DT algorithms performed worse than the others.

3.3. Performance Evalaution of SVM

Following evaluation of all the four algorithms and the discovery that SVM had the best performance, we decided to test this algorithm using different numbers to determine its robustness. We evaluated this algorithm using varying numbers of significant genes as determine by p-value and FDR. That is from 1 to 5502 genes. The tests were performed using the same parameters as those above using these smaller feature sets.
Figure 3 shows results of performance for each number of genes and for overall significant genes. The top and bottom of the box are the 75th and 25th percentiles, respectively. The top and bottom bar are the maximum and minimum value. The circles are the outliers. Figure 3 shows that performance variance was larger when the number of genes was less.
The results showing details of model performance using the training and test sets are shown in Table 3 which displays the most significant results from these experiments. As shown in Figure 3 and Table 3, the best classification performance was achieved using the top 256 genes as features. In general, the smaller sets of genes achieved slightly better scores compared to using all features/genes, though the improvement was not highly significant.

3.4. Comparative Evaluation and Validation of SVM Results

To further validate the developed algorithms, we compared the classification results from this investigation with classic feature selection methods such as SVM-RFE (SVM-Recursive Feature Elimination) [34], ARCO ((Area Under the Curve (AUC) and Rank Correlation coefficient Optimization) [35], Relief [36] and mRMR (minimal redundancy-maximal-relevance) [37] using our data. The mRMR method recorded the highest classification when the number of features/genes was 32, which recorded an accuracy of 83%. The ARCO method achieved the highest classification accuracy (82%) with 64 feature genes. The SVM-RFE method produced the highest classification accuracy (73%) with 128 feature genes, whereas the Relief method recorded the highest classification accuracy (70) with 16 feature genes. As evidenced, the classification accuracy of the above methods was lower than the classification of BC into TNBC and non-TNBC models developed and implemented in this investigation.

4. Discussion

We evaluated the performance of four ML-based classification algorithms: kNNs, NGB, DT and SVM for classification of breast cancer into TNBC and non-TNBC using gene expression data. The investigation revealed that ML algorithms could classify BC into TNBC and non-TNBC. SVM algorithm was the most accurate among the four algorithms. This is consistent with previous reoprts [39]. Nindrea et al. compared SVM to artificial neural network (ANN), decision tree (DT), Naïve Bayes (NB) and K-Nearest Neighbor (KNN) in a meta-analysis of classification algorithms in BC and found that SVM was superior to the other three algorithms [39]. BC classification using imaging data has also been reported [40].
The main difference and novel aspect of our investigation is that it is the first study to report application of ML to classification of BC into TNBC and non-TNBC using RNA-seq data. The clinical significance of this investigation is that ML algorithms could be used not only to improve diagnostic accuracy, but also for identifying women at high risk of developing TNBC which could be prioritized for treatment.
As noted earlier in this report, breast cancer is a highly heterogeneous disease. Thus, one of the major challenges is building accurate and computationally efficient algorithms for classifying patients to guide therapeutic decision making at the point of care. Our investigation shows that among ML-based classification algorithms, SVM out performed the other algorithms and provides the best framewrok for BC classification. This is consistent with previous reports [41,42,43,44]. The clinical significance is that, in addition to classification of BC into TNBC and non-TNBC as demonstrated in this investigation, SVM could also be used for efficient risk, diagnosis and outcome predictions where it has been reported to be superior to other algorithms [41,42,43,44]. Althouh we did not investigate use of ML and in particular SVM algorithm for risk, diagnosis and outcome prediction in this investigation, several studies have reported such application in BC and have also shown its superiority to other algorithms [41,42,43,44], which is consistent with our investigation.
Traditional classification of TNBC and non-TNBC involves use of immunohstochemical staining conducted by hispothologists. In addition, imaging has been used extensively in BC classification [19,40] and radiomics is increasingly being used as a diagnostic tool [20,21]. While there is no doubt that BC clasification using histopathology, imaging and radiomics has been relatively effective, ML algorithms proposed in this investigation provides a novel framework for accurate classification of BC tumors into TNBC and non-TNBC and could complement imaging modalities. More importantly, ML algorithms could help reduce the possible human errors that can occurr because of fatigued or inexperienced experts when medical data is to be examined in shorter time and in more detail. Moreover, given the aggressivenees and lethality of TNBC, accurate identifification of patients with this lethal disease in the early stages may lead to early interventions and improved outcomes.
Our investigation revealed that ML algorithms offer the potential for classifying BC into TNBC and non-TNBC. However, limitations of the study must be acknowledged. First the data size was relatively small and the design was unbalanced with TNBC samples being significantly fewer than non-TNBC. This has the practical consequence of reducing the statistical power of models and also introducing sampling errors in feature selections from differentiall expression analysis. Second, although our ML evaluated and compared the performance of four algorithms, there are many other algorithms that we did not evaluate. However, not withstanding this weakness, evaluation of other algorithms has shown that SVM is superior in BC classification [41,42,43,44]. Lastly, but not least, both TNBC and non-TNBC consist of multiple subtypes of BC and the proposed ML algorithms did not address that problem, as such an undertaking was beyond the scope of this investigation given the small samples sizes and lack of information for ascertaining subtypes.

5. Conclusions

The investigation revealed that ML algorithms can accurately classify BC into the two primary types, TNBC and non-TNBC. The investigation confirmed that the SVM algorithm is able to calculately classify BC into TNBC and non-TNBC more accurately, and with more sensitivity, specificity and lower misclassification errors than other ML algorithms. Further research is recommended to investigate the power of ML algorithms in classifications of subtypes of TNBC and non-TNBC, to identify the best classification features and to integrate radiomics with genomics data. These are subjects of our future investigations.

6. Patents

No patents resulted from the work reported in this manuscript.

Supplementary Materials

The following are available online at https://www.mdpi.com/2075-4426/11/2/61/s1. Supplementary Table S1 complete list of significantly differentially expressed genes distinguishing TNBC from non-TNBC.

Author Contributions

Conceptualization, C.H. and J.W.; methodology, C.H. and J.W.; software, J.W.; validation, C.H. and J.W.; formal analysis, C.H. and J.W.; investigation, C.H. and J.W.; resources, J.W.; data curation, J.W.; writing—original draft preparation, J.W.; writing—review and editing, C.H.; visualization, J.W.; supervision, C.H.; project administration, C.H.; funding acquisition, C.H. All authors have read and agreed to the published version of the manuscript.

Funding

This research was supported by internal funds from the LSUHSC-School of Medicine Startup funds and external funds from the UAB Center for Clinical Grant number # UL1TR001417 and The Louisiana Center for Translational Sciences LSUHSC # U54 GM12254691. All the views expressed in this manuscript are those of the authors and do not represent the funding sources or agency.

Data Availability Statement

The data that support the findings of this study are provided in supplementary tables as documented below, and original data sets are also made available in the TCGA (https://www.cancer.gov/about-nci/organization/ccg/research/structural-genomics/tcga) and are downloadable via the Genomics Data Commons https://gdc.cancer.gov/.

Acknowledgments

The authors wish to thank the participants who donated the samples to the TCGA project used to generate the data used in this project, technical support from TCGA and GDC staff as well as administrative staff from the Department of Genetics.

Conflicts of Interest

The authors declare no conflict of interest. The funders had no role in the design of the study; in the collection, analyses, or interpretation of data; in the writing of the manuscript; or in the decision to publish the results.

References

  1. Siegel, R.L.; Miller, K.D.; Jemal, A. Cancer Statistics, 2019. CA Cancer J. Clin. 2019, 69, 7–34. [Google Scholar] [CrossRef] [Green Version]
  2. American Cancer Society. Cancer Facts and Figures Report 2019; American Cancer Society: Atlanta, GA, USA, 2019. [Google Scholar]
  3. Dietze, E.C.; Sistrunk, C.; Miranda-Carboni, G.; O’Regan, R.; Seewaldt, V.L. Triple-negative breast cancer in African-American women: Disparities versus biology. Nat. Rev. Cancer 2015, 15, 248–254. [Google Scholar] [CrossRef] [Green Version]
  4. Perou, C.M. Molecular Stratification of Triple-Negative Breast Cancers. Oncologist 2010, 15, 39–48. [Google Scholar] [CrossRef] [Green Version]
  5. Xu, H.; Eirew, P.; Mullaly, S.C.; Aparicio, S. The omics of triple-negative breast cancers. Clin. Chem. 2014, 60, 122–133. [Google Scholar] [CrossRef] [Green Version]
  6. Homero, G., Jr.; Maximiliano, R.G.; Jane, R.; Duarte, C. Survival Study of Triple-Negative and Non-Triple-Negative Breast Cancer in a Brazilian Cohort. Clin. Med. Insights Oncol. 2018, 12, 1179554918790563. [Google Scholar]
  7. Joyce, D.P.; Murphy, D.; Lowery, A.J.; Curran, C.; Barry, K.; Malone, C.; McLaughlin, R.; Kerin, M.J. Prospective comparison of outcome after treatment for triple-negative and non-triple-negative breast cancer. Surgeon 2017, 15, 272–277. [Google Scholar] [CrossRef] [PubMed]
  8. Li, X.; Yang, J.; Peng, L.; Sahin, A.A.; Huo, L.; Ward, K.C.; O’Regan, R.; Torres, M.A.; Meisel, J.L. Triple-negative breast cancer has worse overall survival and cause-specific survival than non-triple-negative breast cancer. Breast Cancer Res. Treat. 2017, 161, 279–287. [Google Scholar] [CrossRef]
  9. Pan, X.B.; Qu, S.; Jiang, Y.M.; Zhu, X.D. Triple Negative Breast Cancer versus Non-Triple Negative Breast Cancer Treated with Breast Conservation Surgery Followed by Radiotherapy: A Systematic Review and Meta-Analysis. Breast Care 2015, 10, 413–416. [Google Scholar] [CrossRef] [PubMed] [Green Version]
  10. Ye, J.; Xia, X.; Dong, W.; Hao, H.; Meng, L.; Yang, Y.; Wang, R.; Lyu, Y.; Liu, Y. Cellular uptake mechanism and comparative evaluation of antineoplastic e_ects of paclitaxel-cholesterol lipid emulsion on triple-negative and non-triple-negative breast cancer cell lines. Int. J. Nanomed. 2016, 11, 4125–4140. [Google Scholar] [CrossRef] [Green Version]
  11. Qiu, J.; Xue, X.; Hu, C.; Xu, H.; Kou, D.; Li, R.; Li, M. Comparison of Clinicopathological Features and Prognosis in Triple-Negative and Non-Triple Negative Breast Cancer. J. Cancer 2016, 7, 167–173. [Google Scholar] [CrossRef] [Green Version]
  12. Podo, F.; Santoro, F.; di Leo, G.; Manoukian, S.; de Giacomi, C.; Corcione, S.; Cortesi, L.; Carbonaro, L.A.; Trimboli, R.M.; Cilotti, A.; et al. Triple-Negative versus Non-Triple-Negative Breast Cancers in High-Risk Women: Phenotype Features and Survival from the HIBCRIT-1 MRI-Including Screening Study. Clin. Cancer Res. 2016, 22, 895–904. [Google Scholar] [CrossRef] [PubMed] [Green Version]
  13. Nabi, M.G.; Ahangar, A.; Wahid, M.A.; Kuchay, S. Clinicopathological comparison of triple negative breast cancers with non-triple negative breast cancers in a hospital in North India. Niger. J. Clin. Pract. 2015, 18, 381–386. [Google Scholar]
  14. Koshy, N.; Quispe, D.; Shi, R.; Mansour, R.; Burton, G.V. Cisplatin-gemcitabine therapy in metastatic breast cancer: Improved outcome in triple negative breast cancer patients compared to non-triple negative patients. Breast 2010, 19, 246–248. [Google Scholar] [CrossRef] [PubMed]
  15. Milica, N.; Ana, D. Mechanisms of Chemotherapy Resistance in Triple-Negative Breast Cancer-How We Can Rise to the Challenge. Cells 2019, 8, 957. [Google Scholar]
  16. Giuseppe, V.; Leen, S.; de Snoo, F.A. Discordant assessment of tumor biomarkers by histopathological and molecular assays in the EORTC randomized controlled 10041/BIG 03-04 MINDACT trial breast cancer: Intratumoral heterogeneity and DCIS or normal tissue components are unlikely to be the cause of discordance. Breast Cancer Res. Treat. 2016, 155, 463–469. [Google Scholar]
  17. Viale, G.; de Snoo, F.A.; Slaets, L.; Bogaerts, J. Immunohistochemical versus molecular (BluePrint and MammaPrint) subtyping of breast carcinoma. Outcome results from the EORTC 10041/BIG 3-04 MINDACT trial. Breast Cancer Res. Treat. 2018, 167, 123–131. [Google Scholar] [CrossRef]
  18. Michael, U.; Bernd, G.; Nadia, H. Gallen international breast cancer conference 2013: Primary therapy of early breast cancer evidence, controversies, consensus—Opinion of a german team of experts (zurich 2013). Breast Care 2013, 8, 221–229. [Google Scholar]
  19. Annarita, F.; Teresa, M.B.; Liliana, L. Ensemble Discrete Wavelet Transform and Gray-Level Co-Occurrence Matrix for Microcalcification Cluster Classification in Digital Mammography. Appl. Sci. 2019, 9, 5388. [Google Scholar]
  20. Liliana, L.; Annarita, F.; Teresa, M.; Basile, A. Radiomics Analysis on Contrast-Enhanced Spectral Mammography Images for Breast Cancer Diagnosis: A Pilot Study. Entropy 2019, 21, 1110. [Google Scholar]
  21. Allegra, C.; Andrea, D.; Iole, I. Radiomics in breast cancer classification and prediction. In Seminars Cancer Biology; Academic Press: Cambridge, MA, USA, 2020. [Google Scholar]
  22. Mitra, M.; Mohadeseh, M.; Mahdieh, M.; Amin, B. Machine learning models in breast cancer survival prediction. Technol. Health Care 2016, 24, 31–42. [Google Scholar]
  23. Tong, W.; Laith, R.S.; Jiawei, T.; Theodore, W.C.; Chandra, M.S. Machine learning for diagnostic ultrasound of triple-negative breast cancer. Breast Cancer Res. Treat. 2019, 173, 365–373. [Google Scholar]
  24. Riku, T.; Dmitrii, B.; Mikael, L. Breast cancer outcome prediction with tumour tissue images and machine learning. Breast Cancer Res. Treat 2019, 177, 41–52. [Google Scholar]
  25. Weinstein, J.N.; The Cancer Genome Atlas Research Network; Collisson, E.A. The Cancer Genome Atlas Pan-Cancer analysis project. Nat. Genet. 2013, 45, 1113–1120. [Google Scholar] [CrossRef] [PubMed]
  26. National Cancer Institute. The Genomics Data Commons. Available online: https://gdc.cancer.gov/ (accessed on 19 December 2020).
  27. Ritchie, M.E.; Phipson, B.; Wu, D. limma powers differential expression analyses for RNA-sequencing and microarray studies. Nucleic Acids Res. 2015, 43, e47. [Google Scholar] [CrossRef]
  28. Kas, K.; Schoenmakers, E.F.; Van de Ven, W.J. Physical map location of the human carboxypeptidase M gene (CPM) distal to D12S375 and proximal to D12S8 at chromosome 12q15. Genomics 1995, 30, 403–405. [Google Scholar]
  29. Mihaly, V.; Peter, T. The Protein Ensemble Database. Adv. Exp. Med. Biol. 2015, 870, 335–349. [Google Scholar]
  30. Benjamini, Y.; Yosef, H. Controlling the false discovery rate: A practical and powerful approach to multiple testing. J. R. Stat Soc. 1995, 57, 289–300. [Google Scholar] [CrossRef]
  31. Shawe-Taylor, J.; Nello, C. Kernel Methods for Pattern Analysis; Cambridge University Press: Cambridge, UK, 2004; ISBN 0-521-81397-2. [Google Scholar]
  32. Bernhard, S.; Smola, A.J. Learning with Kernels; MIT Press: Cambridge, MA, USA, 2002; ISBN 0-262-19475-9. [Google Scholar]
  33. Powers, D.M.W. Evaluation: From Precision, Recall and F-Measure to ROC, Informedness, Markedness & Correlation. J. Mach. Learn. Technol. 2011, 2, 37–63. [Google Scholar]
  34. Huang, M.L.; Hung, Y.H.; Lee, W.M.; Li, R.K.; Jiang, B.R. SVM-RFE based feature selection and Taguchi parameters optimization for multiclass SVM classifier. Sci. World J. 2014, 795624. [Google Scholar] [CrossRef]
  35. Piñero, P.; Arco, L.; García, M.M.; Caballero, Y.; Yzquierdo, R.; Morales, A. Two New Metrics for Feature Selection in Pattern Recognition. In Progress in Pattern Recognition, Speech and Image Analysis. CIARP 2003. Lecture Notes in Computer Science; Sanfeliu, A., Ruiz-Shulcloper, J., Eds.; Springer: Berlin/Heidelberg, Germany, 2003. [Google Scholar]
  36. Kira, K.; Rendell, L. The Feature Selection Problem: Traditional Methods and a New Algorithm. In Proceedings of the AAAI-92 Proceedings, San Jose, CA, USA, 12–16 July 1992. [Google Scholar]
  37. Auffarth, B.; Lopez, M.; Cerquides, J. Comparison of redundancy and relevance measures for feature selection in tissue classification of CT images. In Proceedings of the Industrial Conference on Data Mining, Berlin, Germany, 12–14 July 2010; pp. 248–262. [Google Scholar]
  38. Tony, C.S.; Eibe, F. Introducing Machine Learning Concepts with WEKA. Methods Mol. Biol. 2016, 1418, 353–378. [Google Scholar]
  39. Ricvan, D.N.; Teguh, A.; Lutfan, L.; Iwan, D. Diagnostic Accuracy of Different Machine Learning Algorithms for Breast Cancer Risk Calculation: A Meta-Analysis. Asian Pac. J. Cancer Prev. 2018, 19, 1747–1752. [Google Scholar]
  40. La Forgia, D. Radiomic Analysis in Contrast-Enhanced Spectral Mammography for Predicting Breast Cancer Histological Outcome. Diagnostics 2020, 10, 708. [Google Scholar] [CrossRef] [PubMed]
  41. Asri, H.; Mousannif, H.; Al Moatassime, H.; Noel, T. Using machine learning algorithms for breast cancer risk prediction and diagnosis. Procedia Comput. Sci. 2016, 83, 1064–1069. [Google Scholar] [CrossRef] [Green Version]
  42. Polat, K.; Gunes, S. Breast cancer diagnosis using least square support vector machine. Digit. Signal Process 2007, 17, 694–701. [Google Scholar] [CrossRef]
  43. Akay, M.F. Support vector machines combined with feature selection for breast cancer diagnosis. Expert Syst. Appl. 2006, 36, 3240–3247. [Google Scholar] [CrossRef]
  44. Heidari, M.; Khuzani, A.Z.; Hollingsworth, A.B. Prediction of breast cancer risk using a machine learning approach embedded with a locality preserving projection algorithm. Phys. Med. Biol. 2018, 63, 035020. [Google Scholar] [CrossRef]
Figure 1. Project design, data processing and analysis workflow for classification of triple negative breast cancers (TNBC) and non-TNBC using machine learning method. GDC denotes the genomics data commons; DEG denotes differentially expressed genes.
Figure 1. Project design, data processing and analysis workflow for classification of triple negative breast cancers (TNBC) and non-TNBC using machine learning method. GDC denotes the genomics data commons; DEG denotes differentially expressed genes.
Jpm 11 00061 g001
Figure 2. Classification average accuracy of machine learning (ML) methods of different training sample and top k-gene markers, k = 50 (A), k = 100 (B), k = 150 (C), and k = 200 (D), where k is the number of the top most highly significant genes used for various algorithms in each subfigure, on the training and the test sets of breast cancer (BC). For each panel, the x-axis is the sample size used for training, and the y-axis represents the classification accuracy. The blue, red, yellow and green lines represent the K-nearest neighbors (kNNs), Naïve Bayes (NGB), Decision tree (DT) and Support Vector Machines (SVM) method, respectively.
Figure 2. Classification average accuracy of machine learning (ML) methods of different training sample and top k-gene markers, k = 50 (A), k = 100 (B), k = 150 (C), and k = 200 (D), where k is the number of the top most highly significant genes used for various algorithms in each subfigure, on the training and the test sets of breast cancer (BC). For each panel, the x-axis is the sample size used for training, and the y-axis represents the classification accuracy. The blue, red, yellow and green lines represent the K-nearest neighbors (kNNs), Naïve Bayes (NGB), Decision tree (DT) and Support Vector Machines (SVM) method, respectively.
Jpm 11 00061 g002
Figure 3. Average accuracy at varying levels of training sample and gene sizes of Support Vector Machines (SVM) method. The x-axis represents the top number of genes, and the y-axis represents the average accuracy. The top and bottom of the box are the 75th and 25th percentiles, respectively. The top and bottom bar are the maximum and minimum value. The circles are the outliers.
Figure 3. Average accuracy at varying levels of training sample and gene sizes of Support Vector Machines (SVM) method. The x-axis represents the top number of genes, and the y-axis represents the average accuracy. The top and bottom of the box are the 75th and 25th percentiles, respectively. The top and bottom bar are the maximum and minimum value. The circles are the outliers.
Jpm 11 00061 g003
Table 1. Top 30 significantly differentially expressed genes distinguishing TNBC from non-TNBC.
Table 1. Top 30 significantly differentially expressed genes distinguishing TNBC from non-TNBC.
Gene NameChromosomeLog2 Fold Change (logFC)Adjust p-Value
ESR16q25.1-q25.2−8.9660615471.02 × 10−35
MLPH2q37.3−6.2311556111.02 × 10−35
FSIP115q14−6.7856886292.04 × 10−35
C5AR219q13.32−4.9191516243.08 × 10−35
GATA310p14−5.4902215144.68 × 10−35
TBC1D94q31.21−4.7201901218.82 × 10−35
CT6215q23−8.1124126059.86 × 10−35
TFF121q22.3−13.069037192.16 × 10−34
PRR157p14.3−6.252603552.16 × 10−34
CA1215q22.2−6.1685042592.16 × 10−34
AGR37p21.1−11.468738472.38 × 10−34
SRARP1p36.13−12.268070727.31 × 10−34
AGR27p21.1−8.82347081.32 × 10−33
BCAS120q13.2−6.4651400661.34 × 10−33
LINC005044p15.33−7.8469871812.13 × 10−33
THSD415q23−5.07526672.13 × 10−33
CCDC1706q25.1−5.0196579272.13 × 10−33
RHOB2p24.1−2.8284704432.13 × 10−33
FOXA114q21.1−8.2688563172.78 × 10−33
ZNF55219q13.43−3.8139549162.78 × 10−33
SLC16A617q24.2−4.459545052.99 × 10−33
CFAP6120p11.23−3.6806605474.88 × 10−33
GTF2IP77q11.23−6.498290584.98 × 10−33
NEK513q14.3−3.6663102075.90 × 10−33
TTC614q21.1−7.692699931.00 × 10−32
HID117q25.1−3.0696553581.00 × 10−32
ANXA91q21.3−3.7486839281.45 × 10−32
AK89q34.13−3.1347930231.45 × 10−32
FAM198B-AS14q32.1−4.7572939431.63 × 10−32
NAT18p22−6.2789477723.24 × 10−32
Table 2. Performance of classification model for 5502 signature genes.
Table 2. Performance of classification model for 5502 signature genes.
AccuracyRecallSpecificity
K-nearest neighbor (kNN)87%76%88%
Naïve Bayes(NGB)85%68%87%
Decision trees (DT)87%54%91%
Support Vector Machines (SVM)90%87%90%
Table 3. SVM classifier trained on SVM genes obtained with the DE method.
Table 3. SVM classifier trained on SVM genes obtained with the DE method.
Number of GenesTraining SetTest Set
Accuracy Precision Recall Specify F1 Score Accuracy Precision Recall Specify F1 Score
All (5502)0.900.510.870.900.650.820.330.670.800.44
40960.900.520.880.910.650.850.370.580.710.45
20480.920.560.860.920.680.840.380.750.830.50
10240.910.530.870.910.660.860.410.750.810.53
5120.900.510.880.900.650.830.330.580.740.42
2560.910.530.890.910.670.850.380.670.760.48
1280.890.490.870.900.630.820.350.750.850.47
640.870.440.780.880.560.760.260.670.850.37
320.780.270.640.800.380.710.190.500.810.27
160.740.220.630.750.330.690.210.670.890.31
Accuracy = (TP + TN)/(TP + FP + FN + TN). Precision = TP/(TP + FP). Recall = TP/(TP + FN). F1 Score = 2 * (Recall * Precision)/(Recall + Precision). Specificity = TN/(TN + FP).
Publisher’s Note: MDPI stays neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Share and Cite

MDPI and ACS Style

Wu, J.; Hicks, C. Breast Cancer Type Classification Using Machine Learning. J. Pers. Med. 2021, 11, 61. https://doi.org/10.3390/jpm11020061

AMA Style

Wu J, Hicks C. Breast Cancer Type Classification Using Machine Learning. Journal of Personalized Medicine. 2021; 11(2):61. https://doi.org/10.3390/jpm11020061

Chicago/Turabian Style

Wu, Jiande, and Chindo Hicks. 2021. "Breast Cancer Type Classification Using Machine Learning" Journal of Personalized Medicine 11, no. 2: 61. https://doi.org/10.3390/jpm11020061

Note that from the first issue of 2016, this journal uses article numbers instead of page numbers. See further details here.

Article Metrics

Back to TopTop