Enhancing Imbalanced Binary Classification in Employment Big Data Using TabNet-Driven Generative Adversarial Networks
0
조회
0
다운로드
0
인용
0
참고문헌
키워드
초록
Handling imbalanced datasets in binary classification, especially in employment big data, is challenging. Traditional methods like oversampling and undersampling have limitations. This paper integrates TabNet and Generative Adversarial Networks (GANs) to address class imbalance. The generator creates synthetic samples for the minority class, and the discriminator, using TabNet, ensures authenticity. Evaluations on benchmark datasets show significant improvements in accuracy, precision, recall, and F1-score for the minority class, outperforming traditional methods. This integration offers a robust solution for imbalanced datasets in employment big data, leading to fairer and more effective predictive models.
Abstract
Handling imbalanced datasets in binary classification, especially in employment big data, is challenging. Traditional methods like oversampling and undersampling have limitations. This paper integrates TabNet and Generative Adversarial Networks (GANs) to address class imbalance. The generator creates synthetic samples for the minority class, and the discriminator, using TabNet, ensures authenticity. Evaluations on benchmark datasets show significant improvements in accuracy, precision, recall, and F1-score for the minority class, outperforming traditional methods. This integration offers a robust solution for imbalanced datasets in employment big data, leading to fairer and more effective predictive models.
참고문헌 (21)
- "A Novel approach to Handle Imbalanced Dataset in Machine Learning" (2023)[URL]
- Arafat, M. Y., "Machine Learning for Mining Imbalanced Data", ~ (2018)
- "Rethinking Class Imbalance in Machine Learning" (2023)[URL]
- "Research Review on Imbalanced Learning Models in Embedded Intelligent Computing" (2020)[URL]
- "The class imbalance problem in deep learning" (2022)[URL]
- "Addressing the class imbalance in tabular datasets from a generative adversarial network approach in supervised machine learning" (2023)[URL]
- Krawczyk, B., "Learning from imbalanced data: open challenges and future directions", Progress in Artificial Intelligence, 5(4), pp.221~232 (2016)[DOI]
- Fernandez, A., "SMOTE for Learning from Imbalanced Data: Progress and Challenges, Marking the 15-year Anniversary", Journal of Artificial Intelligence Research, 61, pp.863~905 (2018)[DOI]
- Tan, M., "Time-consistent forecasting of unemployment rates with labor force data from Google Trends", Economic Modelling, 93, pp.352~369 (2020)[DOI]
- Chawla, N. V., "SMOTE: Synthetic Minority Over-sampling Technique", Journal of Artificial Intelligence Research, 16, pp.321~357 (2002)[DOI]
- Goodfellow, I., "Generative Adversarial Nets", Advances in Neural Information Processing Systems, 27 (2014)
- Creswell, A., "Generative Adversarial Networks: An Overview", IEEE Signal Processing Magazine, 35(1), pp.53~65 (2018)[DOI]
- "TabNet: Attentive Interpretable Tabular Learning" (2019)[URL]
- Fawaz, H. I., "Deep learning for time series classification: a review", Data Mining and Knowledge Discovery, 33(4), pp.917~963 (2019)[DOI]
- Shamsolmoali, P., "GAN-based Synthetic Brain PET Image Generation", Brain Sciences, 10(2), pp.84~ (2020)[DOI]
- Arik, S.O., "TabNet: Attentive Interpretable Tabular Learning", Proceedings of the AAAI Conference on Artificial Intelligence (2021)
- Zhang, Z., "Adversarial Attacks on Deep Learning Models in Natural Language Processing: A Survey", ACM Transactions on Intelligent Systems and Technology, 10(3), pp.26~ (2019)[DOI]
- Firmani, D., "Data Synthesis in Deep Learning: Techniques and Challenges", Information, 11(4), pp.215~ (2020)[DOI]
- Jiang, H., "To Trust Or Not To Trust A Classifier", NeurIPS (2019)
- Yang, J., "Oversampling method based on GAN for tabular binary classification problems", International Journal of Approximate Reasoning (2023)[DOI]
- Ali-Gombe, A., "MFC-GAN: Class- imbalanced dataset classification using Multiple Fake Class Generative Adversarial Network", Neurocomputing (2019)[DOI]