A Lightweight Method for Vehicle Classification Based on Improved Binarized Convolutional Neural Network
Vehicle classification is an important part of intelligent transportation. Owing to the development of deep learning, better vehicle classification can be achieved compared to traditional methods. Contemporary deep network models have huge computational scales and require a large number of parameter...
| Main Authors: | , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
MDPI AG
2022-06-01
|
| Series: | Electronics |
| Subjects: | |
| Online Access: | https://www.mdpi.com/2079-9292/11/12/1852 |
| _version_ | 1797487971421650944 |
|---|---|
| author | Bangyuan Zhang Kai Zeng |
| author_facet | Bangyuan Zhang Kai Zeng |
| author_sort | Bangyuan Zhang |
| collection | DOAJ |
| description | Vehicle classification is an important part of intelligent transportation. Owing to the development of deep learning, better vehicle classification can be achieved compared to traditional methods. Contemporary deep network models have huge computational scales and require a large number of parameters. Binarized convolutional neural networks (CNNs) can effectively reduce model computational size and the number of parameters. Most contemporary lightweight networks are binarized directly on a full-precision model, leading to shortcomings such as gradient mismatch or serious accuracy degradation. To addresses the inherent defects of binarization networks, herein, we adjust and improve residual blocks and propose a new pooling method, which is called absolute value maximum pooling (Abs-MaxPooling). The information entropy after weight binary quantization is used to propose a weight distribution binary quantization method. A binarized CNN-based vehicle classification model is constructed, and the weights and activation values of the model are quantified to 1 bit, which saves data storage space and improves classification accuracy. The proposed binarized model performs well on the BIT-Vehicle dataset and outperforms some full-precision models. |
| first_indexed | 2024-03-09T23:56:24Z |
| format | Article |
| id | doaj.art-5b01fb34c17c4fe39e94c520396b3057 |
| institution | Directory Open Access Journal |
| issn | 2079-9292 |
| language | English |
| last_indexed | 2024-03-09T23:56:24Z |
| publishDate | 2022-06-01 |
| publisher | MDPI AG |
| record_format | Article |
| series | Electronics |
| spelling | doaj.art-5b01fb34c17c4fe39e94c520396b30572023-11-23T16:24:44ZengMDPI AGElectronics2079-92922022-06-011112185210.3390/electronics11121852A Lightweight Method for Vehicle Classification Based on Improved Binarized Convolutional Neural NetworkBangyuan Zhang0Kai Zeng1Faculty of Information Engineering and Automation, Kunming University of Science and Technology, Kunming 650500, ChinaFaculty of Information Engineering and Automation, Kunming University of Science and Technology, Kunming 650500, ChinaVehicle classification is an important part of intelligent transportation. Owing to the development of deep learning, better vehicle classification can be achieved compared to traditional methods. Contemporary deep network models have huge computational scales and require a large number of parameters. Binarized convolutional neural networks (CNNs) can effectively reduce model computational size and the number of parameters. Most contemporary lightweight networks are binarized directly on a full-precision model, leading to shortcomings such as gradient mismatch or serious accuracy degradation. To addresses the inherent defects of binarization networks, herein, we adjust and improve residual blocks and propose a new pooling method, which is called absolute value maximum pooling (Abs-MaxPooling). The information entropy after weight binary quantization is used to propose a weight distribution binary quantization method. A binarized CNN-based vehicle classification model is constructed, and the weights and activation values of the model are quantified to 1 bit, which saves data storage space and improves classification accuracy. The proposed binarized model performs well on the BIT-Vehicle dataset and outperforms some full-precision models.https://www.mdpi.com/2079-9292/11/12/1852lightweightbinary neural networksdeep learningvehicle classification |
| spellingShingle | Bangyuan Zhang Kai Zeng A Lightweight Method for Vehicle Classification Based on Improved Binarized Convolutional Neural Network Electronics lightweight binary neural networks deep learning vehicle classification |
| title | A Lightweight Method for Vehicle Classification Based on Improved Binarized Convolutional Neural Network |
| title_full | A Lightweight Method for Vehicle Classification Based on Improved Binarized Convolutional Neural Network |
| title_fullStr | A Lightweight Method for Vehicle Classification Based on Improved Binarized Convolutional Neural Network |
| title_full_unstemmed | A Lightweight Method for Vehicle Classification Based on Improved Binarized Convolutional Neural Network |
| title_short | A Lightweight Method for Vehicle Classification Based on Improved Binarized Convolutional Neural Network |
| title_sort | lightweight method for vehicle classification based on improved binarized convolutional neural network |
| topic | lightweight binary neural networks deep learning vehicle classification |
| url | https://www.mdpi.com/2079-9292/11/12/1852 |
| work_keys_str_mv | AT bangyuanzhang alightweightmethodforvehicleclassificationbasedonimprovedbinarizedconvolutionalneuralnetwork AT kaizeng alightweightmethodforvehicleclassificationbasedonimprovedbinarizedconvolutionalneuralnetwork AT bangyuanzhang lightweightmethodforvehicleclassificationbasedonimprovedbinarizedconvolutionalneuralnetwork AT kaizeng lightweightmethodforvehicleclassificationbasedonimprovedbinarizedconvolutionalneuralnetwork |