Abstract
The Broad Transfer Learning Network (BTLN) model uses only one-third of the parameters as the common neural network does to achieve the similar prediction performance that a Multi-Layer Perceptron (MLP) model has. This brings considerable benefits to AI-related studies that highly rely on computational power. This study also proposed a feature mapping technique that applied a linear transformation to the original features to improve the performance of various learning models. This study combines both broad learning and transfer learning techniques. Data augmentation is used to expand the training dataset. It proves that, under certain conditions, the model with a broader network can perform better. The broad network structure can act as an effective feature extractor. The transform learning algorithm can increase the training efficiency due to decreased trainable parameters. The performance improvement of neural network models is particularly remarkable. The performance improves by 3.5% without any change to the model architecture. In this paper, the performance of the BTLN model proposed based on Root Mean Square Error (RMSE) improves by 18.5%. Compared to common neural network models, the training parameters and overall parameters are lowered to 14.57% and 36.28%, respectively.
Talk to us
Join us for a 30 min session where you can share your feedback and ask us any queries you have
Disclaimer: All third-party content on this website/platform is and will remain the property of their respective owners and is provided on "as is" basis without any warranties, express or implied. Use of third-party content does not indicate any affiliation, sponsorship with or endorsement by them. Any references to third-party content is to identify the corresponding services and shall be considered fair use under The CopyrightLaw.