TCN-GAWO: Genetic Algorithm Enhanced Weight Optimization for Temporal Convolutional Network

Shuhuai Gu,Qi Xi,Jing Wang,Peizhen Qiu,Mian Li
DOI: https://doi.org/10.1115/1.4064809
2024-02-19
Journal of Mechanical Design
Abstract:Abstract This paper proposes TCN-GAWO, a Genetic Algorithm enhanced Weight Optimization method for Temporal Convolutional Network (TCN). TCN-GAWO combines the evolutionary process of the genetic algorithm with the gradient-based training, and can achieve higher predication/fitting accuracy than traditional TCN. Performances of TCN-GAWO are also more stable. In TCN-GAWO, multiple TCNs are generated with random initial weights first, then these TCNs are trained individually for given epochs, next the selection-crossover-mutation procedure is applied among TCNs to get the evolved offspring. Gradient-based training and selection-crossover-mutation are taken in turns until convergence. The TCN with the optimal performance is then selected. Performances of TCN-GAWO are thoroughly evaluated using realistic engineering data, including CMPASS dataset provided by NASA and jet engine lubrication oil dataset provided by airlines. Experimental results show that TCN-GAWO outperforms existing methods for both datasets, demonstrating the effectiveness and the wide range applicability of the proposed method in solving time series problems.
engineering, mechanical
What problem does this paper attempt to address?