CAO Xuefei, LI Jihong, WANG Ruibo, NIU Qian, WANG Yu. A Tuning Method for Bi-directional Long Short-Term Memory Neural Network Model Based on Robust Design[J]. Chinese Journal of Applied Probability and Statistics, 2022, 38(3): 317-332. DOI: 10.3969/j.issn.1001-4268.2022.03.001
Citation: CAO Xuefei, LI Jihong, WANG Ruibo, NIU Qian, WANG Yu. A Tuning Method for Bi-directional Long Short-Term Memory Neural Network Model Based on Robust Design[J]. Chinese Journal of Applied Probability and Statistics, 2022, 38(3): 317-332. DOI: 10.3969/j.issn.1001-4268.2022.03.001

A Tuning Method for Bi-directional Long Short-Term Memory Neural Network Model Based on Robust Design

  • The bi-directional long short-term memory neural network model is widely used in natural language processing, but hyperparameter tuning of the model is difficult in practice. In this paper, we take the semantic role recognition task as an example, consider four candidate features (word, part of speech, target word and position) and two hyperparameters (the number of layers of the network and whether CRF classifier is used) as factors in robust design, and select the optimal combination of features and hyperparameters by setting levels of each factor and performing experiments. In particular, we perform 32 cross validation on a small datasets to select the optimal configuration combination of the model based on the SNR of robust design. Then, we analyze the influence of each factor on the performance of the model by quantitatively analyze so that the model has a certain degree of interpretability. Moreover, in order to verify the superiority of our tuning method, we use the standard segmentation of natural language processing on a big dataset, adopt the traditional greedy strategy to select the optimal configuration combination, and compare with our method on the test set. The results show that our method is better than the traditional tuning method.
  • loading

Catalog

    Turn off MathJax
    Article Contents

    /

    DownLoad:  Full-Size Img  PowerPoint
    Return
    Return