Issue |
SHS Web Conf.
Volume 144, 2022
2022 International Conference on Science and Technology Ethics and Human Future (STEHF 2022)
|
|
---|---|---|
Article Number | 03012 | |
Number of page(s) | 4 | |
Section | Application of Artificial Intelligence Technology and Machine Learning Algorithms | |
DOI | https://doi.org/10.1051/shsconf/202214403012 | |
Published online | 26 August 2022 |
Optimization of Neural Network Training for Wine Quality Classification Using Simulated Annealing
The Department of Electronic and Information Engineering, The Hong Kong Polytechnic University, Hongkong, China
* Corresponding author. Email: 20099033d@connect.polyu.hk
The backpropagation algorithm is the most used algorithm to train a neural network. However, a simulated annealing algorithm can do that work too. This paper shows the process and results of training neural network by applying simulated annealing algorithm. A Wine Quality Dataset is used to do the experiment. The experiments first extracted the data by feature selection and pre-processing of the dataset. By applying the Principal Component Analysis method, the features in the original data are extracted into a lower-dimensional space. The importance of features will increase significantly, and there is a positive effect on the training of neural networks. Then a variety of neural networks with different structures are constructed and trained with simulated annealing and back propagation respectively. More specifically, neural networks with two-hidden-layer fully-connected neural networks with two, three, and four hidden nodes in each layer are constructed to represent the different architecture of the network. Finally, their respective prediction results are compared to get a conclusion. This paper uses four parameters, accuracy, precision, recall and F1 score respectively, to evaluate the performance of the two target models, in addition to measure their performance in a more holistic way. As a result, the simulated annealing algorithm performs better than the backpropagation method in the context of wine quality classification.
© The Authors, published by EDP Sciences, 2022
This is an Open Access article distributed under the terms of the Creative Commons Attribution License 4.0, which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
Current usage metrics show cumulative count of Article Views (full-text article views including HTML views, PDF and ePub downloads, according to the available data) and Abstracts Views on Vision4Press platform.
Data correspond to usage on the plateform after 2015. The current usage metrics is available 48-96 hours after online publication and is updated daily on week days.
Initial download of the metrics may take a while.