Handling imbalanced medical image data: A deep-learning-based one-class classification approach

Long Gao,Lei Zhang,Chang Liu,Shandong Wu
DOI: https://doi.org/10.1016/j.artmed.2020.101935
IF: 7.011
2020-08-01
Artificial Intelligence in Medicine
Abstract:<p>In clinical settings, a lot of medical image datasets suffer from the imbalance problem which hampers the detection of outliers (rare health care events), as most classification methods assume an equal occurrence of classes. In this way, identifying outliers in imbalanced datasets has become a crucial issue. To help address this challenge, one-class classification, which focuses on learning a model using samples from only a single given class, has attracted increasing attention. Previous one-class modeling usually uses feature mapping or feature fitting to enforce the feature learning process. However, these methods are limited for medical images which usually have complex features. In this paper, a novel method is proposed to enable deep learning models to optimally learn single-class-relevant inherent imaging features by leveraging the concept of imaging complexity. We investigate and compare the effects of simple but effective perturbing operations applied to images to capture imaging complexity and to enhance feature learning. Extensive experiments are performed on four clinical datasets to show that the proposed method outperforms four state-of-the-art methods.</p>
engineering, biomedical,computer science, artificial intelligence,medical informatics
What problem does this paper attempt to address?
The problem that this paper attempts to solve is dealing with the class - imbalance problem in medical image datasets. In clinical settings, many medical image datasets have imbalance problems, which hinder the detection of anomalies (rare health events) because most classification methods assume that the frequencies of all classes are equal. Therefore, identifying anomalies in imbalanced datasets has become a crucial issue. For this reason, One - Class Classification (OCC) methods have received increasing attention, which focuses on learning models using samples from only a given single class. However, existing one - class modeling methods usually use feature mapping or feature fitting to strengthen the feature - learning process, and these methods have limited effectiveness for medical images with complex features. To address this challenge, the paper proposes a new deep - learning - based method - Image Complexity based One - Class Classification (ICOCC). By leveraging the concept of image complexity, the deep - learning model can optimally learn the inherent imaging features related to a single class. This method studies and compares simple but effective perturbation operations applied to images to capture imaging complexity and enhance feature learning. Extensive experiments show that this method outperforms four state - of - the - art methods on four imbalanced medical image datasets.