The 10th International Conference on Next Generation Computing 2024 (2024.11)바로가기
페이지
pp.378-381
저자
Won-Young Jo, Chan-Uk Yeom, Keun-Chang Kwak
언어
영어(ENG)
URL
https://www.earticle.net/Article/A468890
원문정보
초록
영어
This study compares and analyzes the performance of LIME-based machine learning methods (Gaussian Naive Bayes (GNB), Highly-Efficient Logistic Regression (LR), Linear Support Vector Machine (SVM), and Triple-layer Neural Network (TNN)) using three medical datasets. High-dimensional data increases the likelihood of overfitting in learning algorithms due to the curse of dimensionality. To address this, LIME is utilized to compute the importance of key features contributing to the model's predictions. Based on this, features are selected. The LIME technique generates multiple samples by perturbing the data in the local region. Subsequently, a simple linear model is used to evaluate the impact of each feature on the predictions. Features with high importance derived from this process are selected for model retraining. As a result, it was confirmed that learning time could be reduced while maintaining or even improving performance with a smaller number of features. Consequently, by selecting necessary features, the curse of dimensionality issue is alleviated, and accuracy can be maintained or improved using fewer features in the Hepatitis C Prediction Dataset, Breast Cancer Wisconsin (Prognostic) Dataset, and Glioma Grading Clinical and Mutation Features Dataset.
목차
Abstract I. INTRODUCTION II. LIME III. MACHINE LEARNING MODELS AND LIME-BASED FEATURE SELECTION METHODS A. Gaussian Naive Bayes (GNB) B. Highly-Efficient Logistic Regression (LR) C. Linear Support Vector Machine (SVM) D. Triple-layer Neural Network (TNN) E. LIME-based Machine Learning Method IV. EXPERIMENTS AND RESULTS ANALYSIS V. CONCLUSION ACKNOWLEDGMENT REFERENCE
키워드
LIMEfeature selectionvalidationaccuracyefficiency
저자
Won-Young Jo [ Department of Electronics Engineering, Chosun University Gwangju, South Korea ]
Chan-Uk Yeom [ Division of AI Convergence College Chosun University Gwangju, South Korea ]
Keun-Chang Kwak [ Department of Electronics Engineering, Chosun University Gwangju, South Korea ]