Journal of Jilin University(Engineering and Technology Edition) ›› 2025, Vol. 55 ›› Issue (5): 1705-1713.doi: 10.13229/j.cnki.jdxbgxb.20230812

Previous Articles     Next Articles

Self-selected architecture network for traffic sign classification

Bin WEN1,2(),Yi-fu DING1,Chao YANG1(),Yan-jun SHEN1,Hui LI3   

  1. 1.College of Electrical Engineering and New Energy,China Three Gorges University,Yichang 443002,China
    2.Hubei Provincial Engineering Technology Research Center for Power Transmission Line,Yichang 443002,China
    3.School of Aeronautics and Astronautics,University of Electronic Science and Technology of China,Chengdu 611731,China
  • Received:2023-08-03 Online:2025-05-01 Published:2025-07-18
  • Contact: Chao YANG E-mail:wenbin_08@126.com;yangchao_0305@126.com

Abstract:

The implementation of autonomous driving technology required high-precision recognition of traffic signs. However, due to their high similarity, small size, and vulnerability to outdoor environmental factors, achieving real-time and accurate detection posed significant challenges. In response to the limitations of traditional neural network design approaches, an algorithm based on self-selecting architecture was proposed to automatically adjust the network structure, thereby enhancing model performance and efficiency. The algorithm adopted a two-stage training approach to optimize the selection of network paths. Moreover, gradient propagation was employed to train the hyperparameters for multiple loss functions, replacing the conventional manual tuning with a dynamic loss network scheme. The results demonstrated that the proposed algorithm achieved an accuracy rate of 95.74% and a detection speed of 146.58 frames per second on the GTSRB dataset, while maintaining a model parameter size of only 0.46Mb, enabling deployment on mobile devices. Compared to the traditional manual design of static networks, the adoption of the self-learning architecture module not only reduced experimental costs but also improved accuracy and performance. Furthermore, it enabled superior detection outcomes in various environments and exhibited a noticeable enhancement in loss convergence speed.

Key words: traffic sign classification, deep learning, self-selecting architecture, optimal network path, dynamic loss network

CLC Number: 

  • TP391.4

Fig.1

Neural network architecture"

Fig.2

Optimal path selection in modules"

Fig.3

Inter-node structure"

Fig.4

Two-stage training process"

Fig.5

Loss training process"

Fig.6

GTSRB dataset visualization"

Fig.7

GTSRB dataset analysis"

Fig.8

Two-stage training process"

Fig.9

Path index variation analysis"

Table 1

Path selection experiment"

网络

模块

准确率

/%

帧数

/(帧·s-1

参数量/Mb批显存占用/Mb批训练时间/s
最优路径94.66146.580.46490.5825
路径一93.17146.270.44412.1426
路径二92.49145.440.47124.4124
路径三89.09186.110.2567.4222
路径四86.68195.170.2667.5422
加权路径95.1948.150.751 464.7457

Table 2

Accuracy comparison by loss function"

损失

函数

CrossEntropyLossFocalLossSoftmaxLossLabelSmoothedCrossEntropyLossHybridLoss

准确

率/s

95.4395.4795.6895.1595.74

Fig.10

Loss and accuracy variation by loss function"

Table 3

Algorithm comparison experiment"

实验序号模型名称准确率/%帧数/(帧·s-1参数量/Mb批显存占用/Mb批训练时间/s
10最优路径95.74146.580.46490.5857
1Res2Net5085.0944.1822.64452.4955
2Res2Next5083.9449.2221.66460.2455
4ConvNeXt80.0668.9784.55985.6446
4SwinTransformer89.45105.1926.00401.44942
5EfficientNet-b088.4060.114.88206.4447
6EfficientNet-b490.6829.8816.81607.1078
7GhostNetV290.0451.289.10176.9150
8HorNet82.4129.9420.90488.1090
9FasterNet89.45112.0514.12262.4829

Fig.11

Algorithm accuracy and frame count comparison"

Fig.12

Classification accuracy visualization"

[1] Babi D, Babi D, Fioli M, et al.Analysis of market-ready traffic sign recognition systems in cars: a test field study[J].Energies, 2021, 14: 14123697.
[2] 冯润泽, 江昆, 于伟光, 等. 基于两阶段分类算法的中国交通标志牌识别 [J]. 汽车工程, 2022, 44(3): 434-441.
Feng Run-ze, Jiang Kun, Yu Wei-guang, et al. Chinese traffic sign recognition based on two-stage classification algorithm[J]. Automotive Engineering, 2022, 44(3): 434-441.
[3] Wang M, Liu R, Yang J, et al.Traffic sign three-dimensional reconstruction based on point clouds and panoramic images[J].Photogrammetric Record, 2022, 37(177): 87-110.
[4] Shen L, You L, Peng B, et al.Group multi-scale attention pyramid network for traffic sign detection[J].Neuro Computing, 2021, 452(6): 1-14.
[5] Liu Y, Peng J, Xue J H, et al. TSingNet: scale-aware and context-rich feature learning for traffic sign detection and recognition in the wild[J].Neurocomputing, 2021, 447(4): 10-22.
[6] Vashisht M, Kumar B. Effective implementation of machine learning algorithms using 3D colour texture feature for traffic sign detection for smart cities[J]. Expert Systems, 2021, 39(5): e12781.
[7] Wan H, Gao L, Su M, et al. A novel neural network model for traffic sign detection and recognition under extreme conditions[J]. Journal of Sensors, 2021, 7: 9984787.
[8] Yu L, Xia X, Zhou K. Traffic sign detection based on visual co-saliency in complex scenes [J]. Applied Intelligence, 2019, 49(2): 764-790.
[9] Coţovanu D, Zet C, Foşalău C, et al. Detection of traffic signs based on support vector machine classification using HOG features[C]∥Proceedings of the 2018 International Conference and Exposition on Electrical And Power Engineering, Iasi, Romania, 2018: 8559784.
[10] Lecun Y, Bottou L, Bengio Y, et al. Gradient-based learning applied to document recognition [J]. Proceedings of the IEEE, 1998, 86(11): 2278-2324.
[11] Simonyan K, Zisserman A.Very deep convolutional networks for large-scale image recognition [J/OL].(2015-04-10)[2023-07-30].
[12] He K, Zhang X, Ren S, et al. Deep residual learning for image recognition[C]∥Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition, Las Vegas, USA, 2016: 7780459.
[13] Chollet F. Xception: deep learning with depthwise separable convolutions[C]∥Proceedings of the 2017 IEEE Conference on Computer Vision and Pattern Recognition, Honolulu, USA, 2017: 8099678.
[14] Yu F, Koltun V. Multi-scale context aggregation by dilated convolutions[J/OL].(2016-04-30)[2023-07-30].
[15] Lin T Y, Goyal P, Girshick R, et al. Focal loss for dense object detection[C]∥Proceedings of the 2017 IEEE International Conference on Computer Vision, Venice, Italy, 2017: 8237586.
[16] Liu Z, Lin Y, Cao Y, et al. Swin transformer: hierarchical vision transformer using shifted windows[C]∥Proceedings of the 2021 IEEE/CVF International Conference on Computer Vision, Montreal, Canada, 2021: 10012-10022.
[17] Real E, Liang C, So D R, et al. AutoML-Zero: evolving machine learning algorithms from scratch[C]∥Proceedings of the International Conference on Machine Learning, Vienna, Austria, 2020: 8007-8019.
[18] Liu H, Simonyan K, Yang Y. DARTS: differentiable architecture search[J/OL].(2019-04-23)[2023-07-30].
[19] Gao S H, Cheng M M, Zhao K, et al. Res2Net: A new multi-scale backbone architecture[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2021, 43(2): 652-662.
[20] Liu Z, Mao H, Wu C Y, et al. A convnet for the 2020s[C]∥Proceedings of the 2022 IEEE/CVF Conference on Computer Vision and Pattern Recognition, New Orleans, USA, 2022: 9879745.
[21] Pham H, Guan M Y, Zoph B, et al.Efficient neural architecture search via parameter sharing [J/OL].(2018-02-12)[2023-07-30].
[22] Tang Y, Han K, Guo J, et al. GhostNetV2: enhance cheap operation with long-range attention [J/OL].(2022-11-23)[2023-07-30].
[23] Rao Y, Zhao W, Tang Y, et al.HorNet: efficient high-order spatial interactions with recursive gated convolutions [J/OL].(2022-10-11)[2023-07-30].
[24] Chen J, Kao S H, He H, et al. Run, don't walk: chasing higher flops for faster neural networks [J/OL].(2023-05-21)[2023-07-30].
[1] Jian LI,Huan LIU,Yan-qiu LI,Hai-rui WANG,Lu GUAN,Chang-yi LIAO. Image recognition research on optimizing ResNet-18 model based on THGS algorithm [J]. Journal of Jilin University(Engineering and Technology Edition), 2025, 55(5): 1629-1637.
[2] Ru-bo ZHANG,Shi-qi CHANG,Tian-yi ZHANG. Review on image information hiding methods based on deep learning [J]. Journal of Jilin University(Engineering and Technology Edition), 2025, 55(5): 1497-1515.
[3] Zhen-jiang LI,Li WAN,Shi-rui ZHOU,Chu-qing TAO,Wei WEI. Dynamic estimation of operational risk of tunnel traffic flow based on spatial-temporal Transformer network [J]. Journal of Jilin University(Engineering and Technology Edition), 2025, 55(4): 1336-1345.
[4] Meng-xue ZHAO,Xiang-jiu CHE,Huan XU,Quan-le LIU. A method for generating proposals of medical image based on prior knowledge optimization [J]. Journal of Jilin University(Engineering and Technology Edition), 2025, 55(2): 722-730.
[5] Yuan-ning LIU,Zi-nan ZANG,Hao ZHANG,Zhen LIU. Deep learning-based method for ribonucleic acid secondary structure prediction [J]. Journal of Jilin University(Engineering and Technology Edition), 2025, 55(1): 297-306.
[6] Hui-zhi XU,Shi-sen JIANG,Xiu-qing WANG,Shuang CHEN. Vehicle target detection and ranging in vehicle image based on deep learning [J]. Journal of Jilin University(Engineering and Technology Edition), 2025, 55(1): 185-197.
[7] Lu Li,Jun-qi Song,Ming Zhu,He-qun Tan,Yu-fan Zhou,Chao-qi Sun,Cheng-yu Zhou. Object extraction of yellow catfish based on RGHS image enhancement and improved YOLOv5 network [J]. Journal of Jilin University(Engineering and Technology Edition), 2024, 54(9): 2638-2645.
[8] Lei ZHANG,Jing JIAO,Bo-xin LI,Yan-jie ZHOU. Large capacity semi structured data extraction algorithm combining machine learning and deep learning [J]. Journal of Jilin University(Engineering and Technology Edition), 2024, 54(9): 2631-2637.
[9] Bai-you QIAO,Tong WU,Lu YANG,You-wen JIANG. A text sentiment analysis method based on BiGRU and capsule network [J]. Journal of Jilin University(Engineering and Technology Edition), 2024, 54(7): 2026-2037.
[10] Xin-gang GUO,Ying-chen HE,Chao CHENG. Noise-resistant multistep image super resolution network [J]. Journal of Jilin University(Engineering and Technology Edition), 2024, 54(7): 2063-2071.
[11] Li-ping ZHANG,Bin-yu LIU,Song LI,Zhong-xiao HAO. Trajectory k nearest neighbor query method based on sparse multi-head attention [J]. Journal of Jilin University(Engineering and Technology Edition), 2024, 54(6): 1756-1766.
[12] Ming-hui SUN,Hao XUE,Yu-bo JIN,Wei-dong QU,Gui-he QIN. Video saliency prediction with collective spatio-temporal attention [J]. Journal of Jilin University(Engineering and Technology Edition), 2024, 54(6): 1767-1776.
[13] Yu-kai LU,Shuai-ke YUAN,Shu-sheng XIONG,Shao-peng ZHU,Ning ZHANG. High precision detection system for automotive paint defects [J]. Journal of Jilin University(Engineering and Technology Edition), 2024, 54(5): 1205-1213.
[14] Xiong-fei LI,Zi-xuan SONG,Rui ZHU,Xiao-li ZHANG. Remote sensing change detection model based on multi⁃scale fusion [J]. Journal of Jilin University(Engineering and Technology Edition), 2024, 54(2): 516-523.
[15] Guo-jun YANG,Ya-hui QI,Xiu-ming SHI. Review of bridge crack detection based on digital image technology [J]. Journal of Jilin University(Engineering and Technology Edition), 2024, 54(2): 313-332.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
[1] Yang Fan,Wang Ke,Qian Zhihong. Bluetooth Adaptive Packet Selection Algorithm Based on Signaltonoise Ratio[J]. 吉林大学学报(工学版), 2006, 36(01): 103 -0107 .
[2] CHEN Mian-Shu, FU Ping, LI Yong, ZHANG Hui. Image feature fusion based on scope similarity scores minimization[J]. 吉林大学学报(工学版), 2010, 40(增刊): 365 -0368 .
[3] FU Zhao-yang,GUO Lei,CHANG Wei-wei. Fusion algorithm of hyperspectral images based on wavelet transform and multichannel PCNN[J]. 吉林大学学报(工学版), 2011, 41(03): 838 -843 .
[4] Wang Su, Chen Nan-fei, Zhu Yu-ming, Gao Feng. New approach for modeling objects of functional gradient materials[J]. 吉林大学学报(工学版), 2006, 36(03): 376 -0381 .
[5] . [J]. 吉林大学学报(工学版), 2005, 35(02): 116 -0121 .
[6] WANG Li-juan,LIN Nian-feng,QI Zhong,HUANG Ji-guo,LV Zhen . Pretreatment of pyrimidine wastewater by aerated microelectrolysis and flocculation technique[J]. 吉林大学学报(工学版), 2008, 38(06): 1501 -1504 .
[7] YANG Zhi-Fa, YU Zhuo, GUAN Xin, LI Shi-Wu. Simulation model of road landscape impact on traffic safety[J]. 吉林大学学报(工学版), 2010, 40(增刊): 179 -0182 .
[8] ZHANG Song, MA Lin-hua, TANG Hong, TIAN Yu, ZHA Xiang. Construction of QC-LDPC codes with total linear encoding complexity and eliminating short cycles[J]. 吉林大学学报(工学版), 2013, 43(03): 824 -829 .
[9] CHEN Peng-fei, TIAN Di, YANG Guang. Design and implementation of LIBS software based on MVC architecture[J]. 吉林大学学报(工学版), 2016, 46(1): 242 -245 .
[10] QIN Yu-gang, MA Yong, ZHANG Liang, LI Teng-fei. Parameter identification of ship's maneuvering motion based on improved least square method[J]. 吉林大学学报(工学版), 2016, 46(3): 897 -903 .