Title :
Strategies for addressing class imbalance in ensemble classification of thermography breast cancer features
Author :
Schaefer, Gerald ; Nakashima, Tomoharu
Author_Institution :
Department of Computer Science, Loughborough University, Loughborough, U.K.
Abstract :
Thermography provides an interesting alternative to mammography for diagnosing breast cancer as it is a noncontact, non-invasive and passive technique that is able to detect small tumors and thus can lead to earlier diagnosis. Computer-aided diagnostic approaches based on thermography are typically split into a feature extraction stage that derives useful information from the thermogram images, and a classification stage that distinguishes between malignant and benign cases. The latter is challenging since, as is the case for many medical decision making problems, there are (many) more benign cases available for classifier training compared to malignant cases, leading to an imbalanced classification problem. In this paper, we first perform image analysis to identify features describing bilateral differences in regions of interest in the thermogram. These features then form the input for a pattern classification stage for which we present several strategies to address the existing class imbalance in the context of ensemble classifiers. In particular, we discuss an ensemble constructed of cost-sensitive decision tree classifiers, an ensemble whose base classifiers are trained on balanced subspaces, and an ensemble that is based on the combination of one-class classifiers. All three strategies are evaluated on a challenging dataset of about 150 thermograms and it is shown that they provide very good classification performance and furthermore perform favourably compared to other state-of-the-art classifier ensembles for imbalanced data.
Keywords :
Breast cancer; Feature extraction; Sensitivity; Support vector machines; Training; breast cancer; breast thermogram; ensemble classification; imbalanced classification; machine learning;
Conference_Titel :
Evolutionary Computation (CEC), 2015 IEEE Congress on
Conference_Location :
Sendai, Japan
DOI :
10.1109/CEC.2015.7257177