Ensemble Learning-Based Wine Quality Prediction Using Optimized Feature Selection and XGBoost

被引:0
作者
Tyagi, Sonam [1 ]
Rajput, Ishwari Singh [2 ]
Kumar, Bhawnesh [3 ]
Negi, Harendra Singh [3 ]
机构
[1] ABES Engn Coll, Dept Elect & Commun Engn, Ghaziabad 201009, Uttar Pradesh, India
[2] Graph Era Hill Univ, Dept Comp Sci & Engn, Haldwani 263139, Uttarakhand, India
[3] Graph Era Deemed Be Univ, Dept Comp Sci & Engn, Dehra Dun 248002, Uttarakhand, India
关键词
Wine quality; Feature selection; Machine learning; Classification; Prediction;
D O I
10.33889/IJMEMS.2025.10.5.077
中图分类号
T [工业技术];
学科分类号
08 ;
摘要
Recent years have seen the wine business flourish and become essential to the global economy. Due to rising demand for high-end wines, methods to consistently measure wine quality are needed. In this discipline, machine learning decision-making shows potential. High-dimensional data from several sources may impede processing and classification model performance. Feature selection increases learning and reduces computational costs by picking subsets of features and deleting irrelevant ones. This hybrid architecture uses machine learning to predict wine quality, including a feature selection method to find the most important information. Wrapper-based genetic algorithm (WGA) iteratively removes least significant features and trains a model with the remaining features until the needed number is obtained. We tested the proposed structure on two datasets consisting of 1,599 red wine samples and 4,898 white wine samples, each rated on a scale of 1-10. Additionally, the proposed technique is compared to other cutting-edge machine learning models in the same field. All categorization techniques predicted wine quality well, but WGA-XGB performed best. The study shows how feature selection improves wine quality prediction in different machine learning algorithms. The recommended strategy can be extended to different wine datasets or improved using advanced feature selection or machine learning models to improve forecast performance. Overall, the methodology is designed to be clear, relevant, and reproducible for assessing wine quality in real-world settings.
引用
收藏
页码:1621 / 1639
页数:19
相关论文
共 34 条
[1]   Extended particle swarm optimization for feature selection of high-dimensional biomedical data [J].
Al-Shammary, Dhiah ;
Albukhnefis, Adil L. ;
Alsaeedi, Ali Hakem ;
Al-Asfoor, Muntasir .
CONCURRENCY AND COMPUTATION-PRACTICE & EXPERIENCE, 2022, 34 (10)
[2]   Wrapper feature selection method based differential evolution and extreme learning machine for intrusion detection system [J].
Al-Yaseen, Wathiq Laftah ;
Idrees, Ali Kadhum ;
Almasoudy, Faezah Hamad .
PATTERN RECOGNITION, 2022, 132
[3]  
Anguita D, 2005, STUD FUZZ SOFT COMP, V177, P159
[4]   Boosted decision trees for behaviour mining of concurrent programmes [J].
Avros, R. ;
Dudka, V. ;
Krena, B. ;
Letko, Z. ;
Pluhackova, H. ;
Ur, S. ;
Vojnar, T. ;
Volkovich, Z. .
CONCURRENCY AND COMPUTATION-PRACTICE & EXPERIENCE, 2017, 29 (21)
[5]   A machine learning application in wine quality prediction [J].
Bhardwaj, Piyush ;
Tiwari, Parul ;
Kenneth, Olejar, Jr. ;
Parr, Wendy ;
Kulasiri, Don .
MACHINE LEARNING WITH APPLICATIONS, 2022, 8
[6]   Feature selection in machine learning: A new perspective [J].
Cai, Jie ;
Luo, Jiawei ;
Wang, Shulin ;
Yang, Sheng .
NEUROCOMPUTING, 2018, 300 :70-79
[7]   XGBoost: A Scalable Tree Boosting System [J].
Chen, Tianqi ;
Guestrin, Carlos .
KDD'16: PROCEEDINGS OF THE 22ND ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY AND DATA MINING, 2016, :785-794
[8]  
Chhikara Shruti, 2023, Smart Trends in Computing and Communications: Proceedings of SmartCom 2023. Lecture Notes in Networks and Systems (645), P137, DOI 10.1007/978-981-99-0769-4_14
[9]   A Generalized Wine Quality Prediction Framework by Evolutionary Algorithms [J].
Chiu, Terry Hui-Ye ;
Wu, Chienwen ;
Chen, Chun-Hao .
INTERNATIONAL JOURNAL OF INTERACTIVE MULTIMEDIA AND ARTIFICIAL INTELLIGENCE, 2021, 6 (07) :60-70
[10]   Predictive modeling for wine authenticity using a machine learning approach [J].
Costa, Nattane Luiza da ;
Valentin, Leonardo A. ;
Castro, Inar Alves ;
Barbosa, Rommel Melgaco .
ARTIFICIAL INTELLIGENCE IN AGRICULTURE, 2021, 5 :157-162