Quick Weighing of Passing Vehicles Using the Transfer-Learning-Enhanced Convolutional Neural Network

Abstract

Transfer learning could reduce the time and resources required for the training of new models and be therefore important in generalized applications of the trained machine learning algorithms. In this study, a transfer learning-enhanced convolutional neural network (CNN) was proposed to identify the gross weight and the axle weight of moving vehicles on the bridge. The proposed transfer learning-enhanced CNN model was expected to weigh different bridges based on a small amount of training datasets and provide high identification accuracy. First of all, a CNN algorithm for bridge weigh-in-motion (B-WIM) technology was proposed to identify the axle weight and the gross weight of the typical two-axle, three-axle, and five-axle vehicles as they crossed the bridge with different loading routes and speeds. Then, the pre-trained CNN model was transferred by fine-tuning to weigh the moving vehicles on another bridge. Finally, the identification accuracy and the amount of training data required were compared between the two CNN models. Results showed that the pre-trained CNN model using transfer learning for B-WIM technology could be successfully used for the identification of the axle weight and the gross weight for moving vehicles on another bridge while reducing the training data by 63%. Moreover, the recognition accuracy of the pre-trained CNN model using transfer learning was comparable to that of the original model, showing its promising potentials in the actual applications.