TITLE:
Traffic Sign Recognition Based on CNN and Twin Support Vector Machine Hybrid Model
AUTHORS:
Yang Sun, Longwei Chen
KEYWORDS:
CNN, Twin Support Vector Machine, Wavelet Kernel Function, Traffic Sign Recognition, Transfer Learning
JOURNAL NAME:
Journal of Applied Mathematics and Physics,
Vol.9 No.12,
December
24,
2021
ABSTRACT: With the progress of deep learning research, convolutional neural networks have become the most important method in feature extraction. How to effectively classify and recognize the extracted features will directly affect the performance of the entire network. Traditional processing methods include classification models such as fully connected network models and support vector machines. In order to solve the problem that the traditional convolutional neural network is prone to over-fitting for the classification of small samples, a CNN-TWSVM hybrid model was proposed by fusing the twin support vector machine (TWSVM) with higher computational efficiency as the CNN classifier, and it was applied to the traffic sign recognition task. In order to improve the generalization ability of the model, the wavelet kernel function is introduced to deal with the nonlinear classification task. The method uses the network initialized from the ImageNet dataset to fine-tune the specific domain and intercept the inner layer of the network to extract the high abstract features of the traffic sign image. Finally, the TWSVM based on wavelet kernel function is used to identify the traffic signs, so as to effectively solve the over-fitting problem of traffic signs classification. On GTSRB and BELGIUMTS datasets, the validity and generalization ability of the improved model is verified by comparing with different kernel functions and different SVM classifiers.