Hierarchical Attention: What Really Counts in Various NLP Tasks
Attention mechanisms in sequence to sequence models have shown great ability and wonderful performance in various natural language processing (NLP) tasks, such as sentence embedding, text generation, machine translation, machine reading comprehension, etc. Unfortunately, existing attention mechanisms only learn either high-level or low-level features. In this paper, we think that the lack of hierarchical mechanisms is a bottleneck in improving the performance of the attention mechanisms, and propose a novel Hierarchical Attention Mechanism (Ham) based on the weighted sum of different layers of a multi-level attention. Ham achieves a state-of-the-art BLEU score of on Chinese poem generation task and a nearly averaged improvement compared with the existing machine reading comprehension models such as BIDAF and Match-LSTM. Furthermore, our experiments and theorems reveal that Ham has greater generalization and representation ability than existing attention mechanisms.
Hierarchical Attention: What Really Counts in Various NLP Tasks
Zehao Dou School of Mathematical Science Peking University Beijing, 100871 email@example.com Zhihua Zhang School of Mathematical Science Deep Learning Lab Peking University Beijing, 100871 firstname.lastname@example.org
noticebox[b]Preprint. Work in progress.\end@float
In recent years, long short-term memory, convolutional and recurrent networks with attention mechanisms have been very successful on a variety of NLP tasks. Most of the tasks In NLP can be formulated as sequence to sequence problems. Moreover, encoders and decoders are vital components in sequence to sequence models while processing the inputs and outputs.
An Attention mechanism works as a connector between the encoders and decoders and help the decoder to decide which parts of the source text to pay attention to. Thus an attention mechanism can integrates sequence models and transduction models, so it is able to connect two words in a single passage or paragraph without regard to their positions.
Attention mechanisms have become integral components in various NLP models. For example, in machine translation tasks, attention mechanism-based models [1, 2, 3] have ever been the state-of-the-art; in sentence embedding, self attention based model is now the state-of-the-art ; in machine reading comprehension, almost every recently-published model, such as BIDAF, Match-LSTM, Reinforcement Ranker Reader, and R-NET, contains attention mechanism; in abstractive summarization model  which has also once been the state-of-the-art, attention mechanism is very necessary, and in poem generation , attention mechanism is also widely used. More surprisingly, Vaswani, et al. (2017) showed that their model Transformer which relies solely on the attention mechanisms can outperform those RNN or LSTM-based existing models in machine translation tasks. Thus, they stated that "Attention is all you need".
However, we note that the potential issue with the existing attention mechanisms is that the basic attention mechanism learns only the low-level features while the multi-level attention mechanism learns only the high-level features. This may make it difficult for the model to capture the intermediate feature information, especially when the source texts are long. In order to address this issue, we present Ham which introduces a hierarchical mechanism into the existing multi-level attention mechanisms. Each time when we perform a multi-level attention, instead of using the result of the last attention level only, we use the weighted sum of the results of all the attention levels as the final output.
We show that Ham can learn all levels of features among the tokens in the input sequence and give a proof of its monotonicity and convergency. This work presents the design and implementation of Ham and our implementation performs well on a range of tasks by replacing the existing attention mechanisms in different models of different tasks.
We are able to achieve results comparable to or better than existing state-of-the-art model. On Chinese poem generation, our model scores BLUE, an improvement of from a RNN-based Poem Generator model. On machine reading comprehension task, our implementation is more effective, model with Ham has achieved an average improvement of compared to previous models.
The implementation of the Hierarchical Attention Mechanism is not difficult and the code will be available on http://github.com after the acceptance.
2 Attention mechanisms
The attention mechanism can be described as a function whose input is a query and a set of key-value pairs, where the query and keys are vectors with the same dimension (denoted ), and the values are defined as -vectors. Note that in most types of attention mechanisms, the values are equal to the keys. Through the mapping of the attention mechanism, the input can be mapped to a single vector, which is as the output.
2.1 The Vanilla Attention Mechanism(VAM)
Given a query and an input sequence where denotes the word embedding of the -th word of the sequence, the vanilla attention mechanism aims at using a compatibility function to compute a relativity score between the query and each word . This score is treated as the attention value of to . Then we have attention scores for . Now we apply the softmax function to define a categorical distribution:
Futher, we compute the output which is represented as the weighted sum of the input sequence:
The attention mechanism above is the original version which was firstly proposed by Bahdanau, et al. (2014). In the Scaled Dot-product Attention Mechanism, the compatibility function is defined as the scaled dot product function . Here the scaling factor is used to prevent the dot product from growing too large in magnitude.
2.2 Soft, Hard and Local Attention Mechanisms
There are three different types of mechanisms: soft, hard and local. The main difference between them is the region where attention function is calculated. The VAM belongs to soft attention where the categorical distribution is computed over the whole input sequence of words. Thus it is also referred to as global attention. The resulting distribution can reflect the relatedness or importance between the query and every word in the input sequence, and we use these importance scores as the weights of the output sum. Soft attention takes every word in the input sequence, no matter what kind of word it is, into consideration. Soft attention is differentiable and parameter-free, but is computationally expensive and less accurate.
In order to overcome the weakness of soft attention, hard attention is a natural alternative. Contrary to the widely-studied soft attention, hard attention locates accurately to only one key . In other words, the probability of getting the special key is 1 and others be 0. This implies that the choice of the one key means everything to the performance of the model. The action of choosing is not differentiable, so one uses reinforcement learning methods instead, such as policy gradient method.
As we have seen, soft and hard attentions are two extreme cases. Xu, et al. (2015) proposed a hybrid attention mechanism. Instead of choosing every key or only one key, one chooses a subset of all the keys from the input sequence. When computing the attention, one can just focus on the important part of the keys and discard the rest, thus it is also referred to as local attention. This attention mechanism combines the wideness and accuracy when choosing keys. The subset-choosing process is non-differentiable and reinforcement learning methods are also needed.
2.3 Multi-Head Attention and Multi-Level Attention Mechanisms
The multi-head attention mechanism proposed by Vaswani, et al. (2017) plays an important role in the Transformer model which is state-of-the-art in neural machine translation. Instead of calculating a single attention function with queries, keys and values, it linearly projects the queries, keys and values times to , and dimensions, respectively. On each version of linear projections, the attention function is performed in parallel and yields several versions of -dimensional scaled dot-product attentions. Subsequently, these attention values are concatenated and once again projected, resulting in the final value as the output of the multi-head attention mechanism. That is,
where and are all projection matrices.
The multi-level attention mechanism is another variety of attention mechanisms. Instead of increasing the number of heads, the multi-level attention increases the number of levels. For example, in a two-level attention mechanism, we calculate the attention value of the query and the keys, which is represented as . This output has the same dimension as the query, giving the first level. In the second level, we treat the output as the new query and calculate the attention value with the input sequence (or keys) again. The result can be represented as . The second attention can learn a higher level of internal features among the words of the input text.
Based on the long line of previous attempt, Cui et al. (2017) proposed a novel way of treating various documents in neural machine translation. They used a self-attention mechanism to encode the words in every document and then used a second attention over different documents to learn a higher level of features among the words of different documents. Yang, et al. (2016) also proposed a 2-level hierarchical attention network for document classification task. In recently proposed Transformer mode , the authors repeated the attention mechanism times over the input sequence in order to learn the higher level feature. This is an -level attention mechanism through which the input sequences can be changed again and again into sequences much more suitable for feature extraction and decoder input. This is why so-called Transformer.
2.4 Self Attention Mechanism
In the self attention mechanism the query and key are the same. In other words, the query stems from the input sequence itself. Using self attention mechanism, we are able to learn the relatedness of different parts of the input sequence, no matter what their distance is. With self-attention, a long text can be encoded to a more suitable input for the decoder. Similar to the original attention mechanism, self-attentions also have three different types: soft, hard and local, as well as have multi-head version and multi-level version.
3 Hierarchical Attention Mechanism (Ham)
In this section we present two Hierarchical Attention models built on the vanilla attention and self attention, respectively.
3.1 Hierarchical Vanilla Attention Mechanism (Ham-V)
We have mentioned above that multi-level attention mechanisms can learn a deeper level of features among all the tokens of the input sequence and the query. In our model, we use multi-level for reference but different from the existing Multi-Level Attention Mechanisms. Our Ham-V focus on all the intermediate attention results rather than just the result of the last attention level. As shown in Figure1, given the query and the input sequence which consists of keys, we calculate the Vanilla Attention Mechanism result of them and get Query 1. And then we continue to calculate the attention result of Query 1 and the keys and get Query 2. Repeat this calculation times. Thus, we form a -depth attention. Finally, the output of our Ham-V is the weighted sum of the above attention results, where the weights are the softmax values of trainable parameters. The softmax is used to convert these weights into the probabilities. These weights can tell us the relative importance of the intermediate attention results Query . In other words, the relative importance of the attention levels.
3.2 Hierarchical Self Attention Mechanism (Ham-S)
In Self Attention Mechanisms, the query stems from the input sequence itself. So it can be treated as a special case of attention mechanisms. Similarly, the self-version of Hierarchical Attention Mechanisms, which is shown in Figure 2, takes only the sequence as input. We calculate the self-attention results of the input sequence for times consecutively. Finally, the output of our Ham-S is the weighted sum of these attention results, where the weights are the softmax values of trainable parameters which is the same as Ham-V. Through the levels of self-attention, our model can learn different levels of deep features among all the tokens of the input sequence, and through the trainable parameters and the weighted sum mechanism, our model can learn the relative importance of the self-attention levels.
We present theoretical analysis of our Ham mainly in two aspects: representation ability and convergence. The representation ability of Ham is obviously higher than Vanilla Attention Mechanisms and Multi-Level Attention Mechanisms. Because the two attention mechanisms are just two extreme cases of Ham. When we set , , our Ham model is equivalent to the former. When we set , our model is equivalent to the latter. Thus Ham is much more general and the representation ability is much higher. Using the weighted linear combination of these intermediate attention results, our model takes every level of features into consideration.
We are going to prove that the global minimum value of our loss function of the whole Ham model will decrease monotonically and converges finally as the increase of hierarchical attention depth . Here, denotes a Ham with the attention depth and trainable parameters , denotes all the input data including the queries and keys, and denotes all the parameters in the other part of the whole model. It is also worth emphasizing that all loss functions in NLP tasks have positive values.
In Vanilla Attention Mechanism, we have that
This result is very obvious according to the definition of Vanilla Attention Mechanism, is the weighted sum of and the weights are nonnegative with their sum 1. Denote the weights as . Then
The left-hand side of the inequality can be proved similarly. This theorem tells us that through multiple attention layers, the vectors of intermediate attention levels will neither explode nor vanish as the increase of attention depth .
Let be the global minimal value of the loss function. Then is a monotonically decreasing sequence and it will converge.
It is easy to note that:
This means the monotonicity of the sequence . On the other hand, since our loss function always has positive values, sequence has 0 as its lower bound. Therefore, the monotonically decreasing sequence will converge.
Attention mechanisms are widely used in various NLP tasks. In our experiments, we would like to replace existing attention mechanisms with our novel Ham-V and replace existing self-attention mechanisms with our Ham-S to show the powerfulness and generalization ability of our model. We conduct our experiments on two different NLP tasks, Chinese Poem Generation and Machine Reading Comprehension.
5.1 Machine Reading Comprehension
The first NLP task we used to test our Ham is the machine reading comprehension (MRC). We conduct our experiment on both English MRC and Chinese MRC. The baseline models we use include BIDAF, Match-LSTM, R-NET. Here, BIDAF is for both Chinese and the other two are for English. They have a major similarity that all of them use attention mechanism as the connection between their encoders and decoders and they are all open source. Their code is available on http://github.com/baidu/DuReader, https://github.com/MurtyShikhar/Question-Answering and https://github.com/NLPLearn/R-net. What we will do is to replace their attention mechanism with Ham and compare the difference of their performance. Specially, the R-NET model contains two attention mechanisms when doing question-passage matching and passage self-matching. Here, we replace them both with Ham and Ham-S.
The Chinese dataset we use for MRC experiments is DuREADER which is introduced by He, et al. (2017) and the English dataset we use include SQUAD which can be downloaded from https://rajpurkar.github.io/SQuAD-explorer and MS-MARCO which can be downloaded from http://www.msmarco.org. We randomly choose 10 percent of question-answer data as testing set and the rest as training set. The evaluation method we use is BLEU-4 and ROUGE-L for Chinese, ExactMatch(EM) and F1 score for English, where EM measures the percentage of how much the prediction of the model matches ground truth exactly and F1 measures the overlap between prediction and ground truth. During our experiments, we set also different attention depths to show the influence of attention depth.
5.2 Chinese Poem Generation
In this work we generate Chinese quatrains, each of whose lines has the same length of 5 or 7 characters. The baseline model we use is Planning based Poetry Generation (PPG) proposed by Wang, et al. (2016), which generates Chinese poetries with a planning based neural network. Once we input a Chinese text, this model will generate a highly-related Chinese quatrains as follows.
Firstly, the model extracts keywords from this input text with TextRank algorithm proposed by Mihalcea, et al. (2004). Next, if the number of extracted keywords is not enough for a whole quatrain, more keywords will be created by Knowledge-based method. Then it comes to the final step, poem generation. The quatrain is generated line by line and each line corresponds a keyword. When generating a single line, one uses a bi-directional Gated Recurrent Unit (GRU) model proposed by Cho, et al. (2014 ) as encoder and another GRU model as decoder. Between encoder and decoder, an attention mechanism is used for connection.
It is worth emphasizing that the dataset used by PPG consists of 76,859 quatrains from the Internet and PPG randomly chooses 2000 quatrains for testing, 2000 for validation and the rest for training. In the encoder part of PPG, the word embedding dimensionality is set as 512 and initialized by word2vec (Mikolov, et al. (2013)). In both GRU models, the hidden layers also contain 512 hidden units but they are initialized randomly. For more details, please read Wang, et al. (2016). The code and dataset of PPG model can be found from https://github.com/Disiok/poetry-seq2seq.
In our experiment, we replace the attention part of PPG from a Vanilla Attention Mechanism to our Ham-V and set the compatibility function to be the scaled dot product function, while keeping other parts and dataset unchanged as PPG except the evaluation part. The evaluation of poem generation in PPG is done by experts and we can not keep evaluation method unchanged since it is not convincing to find experts for evaluation. Our evaluation algorithm is based on BLEU-2 score which is calculated as
where denotes the BLEU-2 score computed for the next th lines given the previous goldstandard lines. This averaged can tell us how much correlated the lines of a generated quatrain are. We will show some quatrains generated by our Ham-based PPG in the appendix.
6 Experimental Results and Qualitative Analysis
6.1 Machine Reading Comprehension
As clearly visible in Table 1, the proposed model is much better than conventional model at Chinese and English machine reading comprehension. This is likely due to the fact that our human language has a kind of hierarchical relationship within itself both structurally and semantically. And our hierarchical attention mechanism is much easier to capture the inherent structural and semantical hierarchical relationship in the source texts because of their innate similarity. We also set the attention depths to be 1 (which is equivalent to ordinary models without using Ham), 2, 5, 10 and 20.
From Table 1, we can find that our Ham plays a significant role of the whole model. With the increase of attention depth , the performance rises quickly at first and starts to converge when grows larger. The biggest improvements on these three models are , and respectively. Their average is over which is a huge progress.
|BIDAF with 2-level Ham||DuReader||34.02||44.39|
|BIDAF with 5-level Ham||DuReader||34.79||45.10|
|BIDAF with 10-level Ham||DuReader||35.96||47.33|
|BIDAF with 20-level Ham||DuReader||35.79||47.41|
|Match-LSTM with 2-level Ham||SQUAD||54.41||66.99|
|Match-LSTM with 5-level Ham||SQUAD||55.29||69.03|
|Match-LSTM with 10-level Ham||SQUAD||58.37||70.70|
|Match-LSTM with 20-level Ham||SQUAD||58.47||70.81|
|R-NET with 2-level Ham||MSMARCO||41.37||43.89|
|R-NET with 5-level Ham||MSMARCO||43.13||45.67|
|R-NET with 10-level Ham||MSMARCO||43.48||45.75|
|R-NET with 20-level Ham||MSMARCO||43.62||45.78|
6.2 Chinese Poem Generation
The results of our BLEU-based evaluation are summarized in Table 2. We compare our Ham-based PPG with several relevant baselines like Statistical Machine Translation (SMT) proposed by He, et al. (2012) and RNN-based Poem Generator (RNNPG) proposed by Zhang, et al. (2014). In the former model, a poem is generated iteratively by translating the previous line into the next line. In the latter model, all the lines are generated based on a context vector encoded from the previous lines. We also set different attention depths in order to learn the relationship between overall performance and the attention depth .
|5-level Ham PPG||0.063||0.210||0.070||0.237||0.075||0.226||0.070||0.224|
|10-level Ham PPG||0.062||0.217||0.075||0.267||0.076||0.259||0.072||0.244|
|20-level Ham PPG||0.062||0.221||0.074||0.258||0.076||0.260||0.071||0.246|
When attention depth is not large enough, the larger is, the better performance our model will achieve. PPG with 10-level Ham has almost of improvement on the averaged BLEU score compared with ordinary PPG. On the other hand, through the last two rows of the table above and Theorem 2, we know that with the continuing increase of , our performance will converge sooner or later. So in order to balance between performance and training cost, we suggest attention depth to be between 5 and 10.
7 Concluding Remarks
In this paper we have developed Hierarchical Attention Mechanism (Ham). So far as we know, This is the first attention mechanism which introduces hierarchical mechanisms into attention mechanisms and takes the weighted sum of different attention levels as the output, so it combines low-level features and high-level features of input sequences to output a more suitable intermediate result for decoders.
We tested the proposed model Ham on the task of Chinese poem generation and machine reading comprehension. The experiment revealed that the proposed Ham outperforms the conventional models significantly, achieving the state-of-the-art results. In the future, we would like to study more applications of Ham on other NLP tasks such as neural machine translation, abstractive summarization, paraphrase generalization and so on.
Recall that Ham belongs to soft attention where every token of input sequences is calculated by attention function. We will extend Ham to hard attention and local attention, to show whether the performance can be better and whether Ham can fit reinforcement learning environment better. Also, we will attempt to extend Multi-head Attention Mechanism of Vaswani, et al. (2017) to its hierarchical version and apply to neural machine translation.
 Vaswani, Ashish, Noam Shazeer, Niki Parmar, Jakob Uszkoreit, Llion Jones, Aidan N. Gomez, Lukasz Kaiser, and Illia Polosukhin. "Attention is all you need." In Advances in Neural Information Processing Systems, pp. 6000-6010. 2017.
 Bahdanau, Dzmitry, Kyunghyun Cho, and Yoshua Bengio. "Neural machine translation by jointly learning to align and translate." arXiv preprint arXiv:1409.0473 (2014).
 Shenjian Zhao and Zhihua Zhang. Attention-via-Attention Neural Machine Translation. In Proceedings of the Thirty-Second AAAI Conference on Artificial Intelligence (AAAI’18) , 2018.
 Lin, Zhouhan, Minwei Feng, Cicero Nogueira dos Santos, Mo Yu, Bing Xiang, Bowen Zhou, and Yoshua Bengio. "A structured self-attentive sentence embedding." arXiv preprint arXiv:1703.03130 (2017).
 Seo, Minjoon, Aniruddha Kembhavi, Ali Farhadi, and Hannaneh Hajishirzi. "Bidirectional attention flow for machine comprehension." arXiv preprint arXiv:1611.01603 (2016).
 Wang, Shuohang, and Jing Jiang. "Machine comprehension using match-lstm and answer pointer." arXiv preprint arXiv:1608.07905 (2016).
 Wang, Shuohang, Mo Yu, Xiaoxiao Guo, Zhiguo Wang, Tim Klinger, Wei Zhang, Shiyu Chang, Gerald Tesauro, Bowen Zhou, and Jing Jiang. "R : Reinforced Reader-Ranker for Open-Domain Question Answering." arXiv preprint arXiv:1709.00023 (2017).
 Wang, Wenhui, Nan Yang, Furu Wei, Baobao Chang, and Ming Zhou. "Gated self-matching networks for reading comprehension and question answering." In Proceedings of the 55th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers), vol. 1, pp. 189-198. 2017.
 Wang, Zhe, Wei He, Hua Wu, Haiyang Wu, Wei Li, Haifeng Wang, and Enhong Chen. "Chinese poetry generation with planning based neural network." arXiv preprint arXiv:1610.09889 (2016).
 Xu, Kelvin, Jimmy Ba, Ryan Kiros, Kyunghyun Cho, Aaron Courville, Ruslan Salakhudinov, Rich Zemel, and Yoshua Bengio. "Show, attend and tell: Neural image caption generation with visual attention." In International Conference on Machine Learning, pp. 2048-2057. 2015.
 Yiming Cui, Zhipeng Chen, Si Wei, Shijin Wang, Ting Liu and Guoping Hu. "Attention-over-Attention Neural Networks for Reading Comprehension" arXiv preprint arXiv:1607.04423v4 (2017).
 Rush, Alexander M., Sumit Chopra, and Jason Weston. "A neural attention model for abstractive sentence summarization." arXiv preprint arXiv:1509.00685 (2015).
 Mihalcea, Rada, and Paul Tarau. "Textrank: Bringing order into text." In Proceedings of the 2004 conference on empirical methods in natural language processing. 2004.
 Mikolov, Tomas, Kai Chen, Greg Corrado, and Jeffrey Dean. "Efficient estimation of word representations in vector space." arXiv preprint arXiv:1301.3781 (2013).
 He, Jing, Ming Zhou, and Long Jiang. "Generating Chinese Classical Poems with Statistical Machine Translation Models." In AAAI. 2012.
 Zhang, Xingxing, and Mirella Lapata. "Chinese poetry generation with recurrent neural networks." In Proceedings of the 2014 Conference on Empirical Methods in Natural Language Processing (EMNLP), pp. 670-680. 2014.
 He, Kaiming, Xiangyu Zhang, Shaoqing Ren, and Jian Sun. "Deep residual learning for image recognition." In Proceedings of the IEEE conference on computer vision and pattern recognition, pp. 770-778. 2016.
 He, Wei, Kai Liu, Yajuan Lyu, Shiqi Zhao, Xinyan Xiao, Yuan Liu, Yizhong Wang et al. "DuReader: a Chinese Machine Reading Comprehension Dataset from Real-world Applications." arXiv preprint arXiv:1711.05073 (2017).
 Yang, Zichao, Diyi Yang, Chris Dyer, Xiaodong He, Alex Smola, and Eduard Hovy. "Hierarchical attention networks for document classification." In Proceedings of the 2016 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies, pp. 1480-1489. 2016.