Optimal Hyperparameters for Deep LSTM-Networks for Sequence Labeling Tasks
Nils Reimers Iryna Gurevych Affiliation: Ubiquitous Knowledge Processing Lab (UKP) and Research Training Group AIPHES Affiliation: Department of Computer Science, Technische Universität Darmstadt Affiliation: Ubiquitous Knowledge Processing Lab (UKP-DIPF) Affiliation: German Institute for Educational Research Affiliation: www.ukp.tu-darmstadt.de
Abstract
Selecting optimal parameters for a neural network architecture can often make the difference between mediocre and state-of-the-art performance. However, little is published which parameters and design choices should be evaluated or selected making the correct hyperparameter optimization often a “black art that requires expert experiences” [Snoek et al. (2012]. In this paper, we evaluate the importance of different network design choices and hyperparameters for five common linguistic sequence tagging tasks (POS, Chunking, NER, Entity Recognition, and Event Detection). We evaluated over 50.000 different setups and found, that some parameters, like the pre-trained word embeddings or the last layer of the network, have a large impact on the performance, while other parameters, for example the number of LSTM layers or the number of recurrent units, are of minor importance. We give a recommendation on a configuration that performs well among different tasks. The optimized implementation of our BiLSTM-CRF architecture is publicly available.11 1 https://github.com/UKPLab/emnlp2017-bilstm-cnn-crf
原文 arXiv:1707.06799;中英对照 + 大白话阅读 https://aha.fim.ai/paper/1707.06799v2