Abstract:
Class-imbalance is common occurrence in the machine learning and data mining community. Many classification algorithms often fail to learn the minority class information ...Show MoreMetadata
Abstract:
Class-imbalance is common occurrence in the machine learning and data mining community. Many classification algorithms often fail to learn the minority class information when encountering imbalanced data, resulting in the model being biased towards the majority class. Most importantly, existing learning algorithms mainly concern the class-imbalance distribution, and do not take into account the small number of samples, high feature dimensions, and other issues. To this end, this paper presents a new learning paradigm for solving above problems. To be specific, we first utilize Autoencoder to achieve dimensionality reduction and extract most discriminative features. Next, we explore focal loss to solve the class-imbalance issue and obtain better classification capability. Finally, we tackle the black-box optimization problem via Bayesian Optimization and find a suitable parameters for classification. Extensive experiments performed on three real-world datasets show our method achieves better performance.
Published in: 2022 12th International Conference on Information Technology in Medicine and Education (ITME)
Date of Conference: 18-20 November 2022
Date Added to IEEE Xplore: 04 April 2023
ISBN Information: