Efficient Natural Language Response Suggestion for Smart Reply


1 Efficient Natural Language Response Suggestion for Smart Reply MATTHEW HENDERSON, RAMI AL-RFOU, BRIAN STROPE, YUN-HSUAN SUNG, ́ ́ ́ L ACS, RUIQI GUO, SANJIV KUMAR, BALINT MIKLOS, and O LUK ASZL Google RAY KURZWEIL, This paper presents a computationally efficient machine-learned method for natural language response suggestion. Feed-forward neural networks using n-gram embedding features encode messages into vectors which are optimized to give message-response pairs a high dot-product value. An optimized search finds response suggestions. The method is evaluated in a large-scale commercial e-mail application, Inbox by Gmail . Compared to a sequence-to-sequence approach, the new system achieves the same quality at a small fraction of the computational requirements and latency. Additional Key Words and Phrases: Natural Language Understanding; Deep Learning; Semantics; Email 1 INTRODUCTION Applications of natural language understanding (NLU) are becoming increasingly interesting with scalable machine learning, web-scale training datasets, and applications that enable fast and nuanced quality evaluations with large numbers of user interactions. Early NLU systems parsed natural language with hand-crafted rules to explicit semantic repre- sentations, and used manually written state machines to generate specific responses from the output 18 ]. Such systems are generally limited to the situations imagined by the designer, and of parsing [ much of the development work involves writing more rules to improve the robustness of semantic parsing and the coverage of the state machines. These systems are brittle, and progress is slow [ 31 ]. Eventually adding more parsing rules and response strategies becomes too complicated for a single designer to manage, and dependencies between the rules become challenging to coordinate across larger teams. Often the best solution is to keep the domains decidedly narrow. Statistical systems can offer a more forgiving path by learning implicit trade-offs, generalizations, and robust behaviors from data. For example, neural network models have been used to learn more robust parsers [ , 24 , 29 ]. In recent work, the components of task-oriented dialog systems have 14 7 been implemented as neural networks, enabling joint learning of robust models [ 26 , 27 ]. However , these methods all rely on either an explicit semantic representation or an explicit representation of the task, always hand-crafted. End-to-end systems avoid using hand-crafted explicit representations, by learning to map to and from natural language via implicit internal vector representations [ 19 , 25 ]. Such systems avoid the unnecessary contraints and bottlenecks inevitably imposed by the system designer. In that context, natural language understanding might be evaluated less in terms of an explicit semantic representation, and more by the utility of the system itself. The system shows evidence of understanding when it offers useful responses. Such end-to-end tasks are difficult: systems not only need to learn language but also must learn to do something useful with it. This paper addresses the task of suggesting responses in human-to- human conversations. There are further challenges that arise when building an end-to-end dialog Corresponding authors: { matthen, rmyeid, bps } @google.com . © 2017 Copyright held by the owner/author(s). Publication rights licensed to ACM.

2 Efficient Natural Language Response Suggestion for Smart Reply Henderson et al. system, i.e. a computer agent that interacts directly with a human user. Dialog systems must learn effective and robust interaction strategies, and goal-oriented systems may need to interact with discrete external databases. Dialog systems must also learn to be consistent throughout the course of a dialog, maintaining some kind of memory from turn to turn. Machine learning requires huge amounts of data, and lots of helpful users to guide development through live interactions, but we also need to make some architectural decisions, in particular how to represent natural language text. Neural natural language understanding models typically represent words, and possibly phrases, , sentences, and documents as implicit vectors. Vector representations of words, or word embeddings have been widely adopted, particularly since the introduction of efficient computational learning algorithms that can derive meaningful embeddings from unlabeled text [15, 17, 20]. Though a simple representation of a sequence of words can be obtained by summing the individual word embeddings, this discards information about the word ordering. The sequence-to-sequence ( ) framework uses recurrent neural networks (RNNs), typically long short-term memory Seq2Seq (LSTM) networks, to encode sequences of word embeddings into representations that depend on the order, and uses a decoder RNN to generate output sequences word by word. This framework provides a direct path for end-to-end learning [ ]. With attention mechanisms and more layers, 23 28 ]. A similar system was initially these systems are revolutionizing the field of machine translation [ Smart Reply system for Inbox by Gmail [11]. used to deployed Google’s While Seq2Seq models provide a generalized solution, it is not obvious that they are maximally efficient, and training these systems can be slow and complicated. Also they are derived as a generative model, and so using them to rank a fixed set of responses (as in the context of Smart Reply) requires extra normalization to bias the system away from common responses. In a broader context, Kurzweil’s work outlines a path to create a simulation of the human neocortex (the outer layer of the brain where we do much of our thinking) by building a hierarchy of similarly structured components that encode increasingly abstract ideas as sequences [ 12 ]. Kurzweil provides evidence that the neocortex is a self-organizing hierarchy of modules, each of which can learn, remember, recognize and/or generate a sequence, in which each sequence consists of a sequential pattern from lower-level modules. Longer relationships (between elements that are far away in time or spatial distance) are modeled by the hierarchy itself. In this work we adopt such a hierarchical structure, representing each sequential model as a feed-forward vector computation (with underlying sequences implicitly represented using n-grams). Whereas a long short-term memory (LSTM) network could also model such sequences, we don’t need an LSTM’s ability to directly encode long-term relationships (since the hierarchy does that) and LSTMs are much slower than feed-forward networks for training and inference since the computation scales with the length of the sequence. Similarly, the work on shows that word embeddings can be back-propagated to paragraph vectors arbitrary levels in a contextual hierarchy [ 13 ]. Machines can optimize sentence vectors, paragraph vectors, chapter vectors, book vectors, author vectors, and so on, with simple back-propagation and computationally efficient feed-forward networks. Putting a few of these ideas together, we wondered if we could predict a sentence using only the sum of its n-gram embeddings. Without the ordering of the words, can we use the limited sequence information from the n-grams, and the redundancy of language, to recreate the original word sequence? With a simple RNN as a decoder, our preliminary experiments showed perplexities of around 1.2 over a vocabulary of hundreds of thousands of words. A lot of the sequence information remains in this simple n-gram sentence representation. As a corollary, a hierarchy built on top of n-gram representations could indeed adequately represent the increasingly abstract sequences underlying natural language. Networks built on n-gram embeddings such as those presented in this

3 Efficient Natural Language Response Suggestion for Smart Reply Henderson et al. paper (see section 4) are computationally inexpensive relative to RNN and convolutional network [6, 30] encoders. To make sure there is enough data and the necessary live feedback from users, we train on the ], and use our models to give anonymized Gmail data that was used in Kannan et al. [ Smart Reply 11 response suggestions to users of Inbox by Gmail (see figure 1). Smart Reply provides a real world application in which we can measure the quality of our response suggestion models. Just as in Kannan et al. [ 11 ], we consider natural language response suggestion from a fixed set of candidates. For efficiency, we frame this as a search problem. Inputs are combined with potential responses using final dot products to enable precomputation of the “response side” of the system. Adding deep layers and delaying combination between input and responses encourages the network to derive implicit semantic representations of the input and responses– if we assume that the best way to predict their relationships is to understand them. We precompute a minimal hierarchy of deep feed-forward networks for all potential responses, and at runtime propagate only the input through the hierarchical network. We use an efficient nearest-neighbor search of the hierarchical embeddings of the responses to find the best suggestions. 2 PROBLEM DEFINITION The Smart Reply system gives short response suggestions to help users respond quickly to emails. Emails are processed by the system according to the pipeline detailed in figure 2. The decision of whether to give suggestions is made by a deep neural network classifier, called the triggering model. This model takes various features of the received email, including a word n-gram representation, and is trained to estimate the probability that the user would type a short reply to the input email, see Kannan et al. [ 11 ]. If the output of the triggering model is above a threshold, then Smart Reply will give (typically 3) short response suggestions for the email. Otherwise no m suggestions are given. As a result, suggestions are not shown for emails where a response is not likely (e.g. spam, news letters, and promotional emails), reducing clutter in the user interface and saving unnecessary computation. The system is restricted to a fixed set of response suggestions, R , selected from millions of common messages. The response selection step involves searching for the top N (typically around 100) scoring responses in according to a response selection model P ( y | x ) . The output of R response selection is a list of suggestions ( y ordered by their probability. , y R , ..., y ∈ ) with y i 2 1 N Kannan et al. [ 11 ] used a sequence-to-sequence model for P ( y | x ) and used a beam search over the Fig. 1. Our natural language understanding models are trained on email data, and evaluated in the context of the Smart Reply feature of Inbox by Gmail , pictured here.

4 Efficient Natural Language Response Suggestion for Smart Reply Henderson et al. x new email Trigger No Smart Reply no ? suggestions suggestions Fig. 2. The Smart Reply pipeline. A re- ceived email is run through the triggering model that decides whether suggestions yes should be given. Response selection searches the response set for good sug- Response Response selection R and set gestions. Finally, diversification ensures ) ,...,y y ( clustering k 1 diversity in the final set shown to the user. This paper focuses on the response se- lection step. Diversification ,...,y ( ) y i i m 1 Smart Reply suggestions are shown prefices in (see section 3). This paper presents a feed-forward neural network model for P ( y | x ) , R including a factorized model where selection can be performed using a highly efficient dot-product and accurate approximate search over a precomputed set of vectors, see section 4. Finally the m response suggestions. A clus- diversification stage ensures diversity in the final R tering algorithm is used to omit redundant suggestions, and a labeling of is used to ensure a negative suggestion is given if the other two are affirmative and vice-versa. Full details are given in Kannan et al. [11]. 3 BASELINE SEQUENCE-TO-SEQUENCE SCORING 11 ] is a long short-term memory (LSTM) The response selection model presented in Kannan et al. [ recurrent neural network [ 8 ] – an application of the sequence-to-sequence learning framework Seq2Seq ( ) [23]. x is tokenized into a word sequence ( The input email and the LSTM computes the , ..., x ) x m 1 conditional probability over a response sequence y = ( y as: , ..., y ) n 1 ( y | x ) = P ( y , ..., y P ) | x , ..., x 1 m n 1 ∏ n ) , ..., y ,y P , ..., x ( y x | = i m LSTM 1 1 i − 1 i =1 P where is the output of the word-level LSTM. The LSTM is trained to maximize the log- LSTM probability according to P y | x ) of the training data (a large collection of emails and responses, see ( section 5.1). At inference time, likely responses from the candidate set R are found using a beam | search that is restricted to the prefix trie of . The time complexity of this search is O ( | x | + b | y R ) | where is the beam width and should be scaled appropriately with | R b . This search dominates the computation of the original Smart Reply system. 4 FEEDFORWARD APPROACH Rather than learning a generative model, we investigate learning a feedforward network to score potential responses.

5 Efficient Natural Language Response Suggestion for Smart Reply Henderson et al. P Recall the goal of response selection is to model | x ) , which is used to rank possible responses ( y x y . This probability distribution can be written as: given an input email ) x,y ( P ∑ (1) | y ) = P x ( x,y ( ) P k k The joint probability of P ( ) is estimated using a learned neural network scoring function, S x,y such that: x,y ( S ) ∝ e x,y ( P (2) ) Note that the calculation of equation 1 requires summing over the neural network outputs for all possible responses . (This is only an issue for training, and not inference since the denominator is y k x and so does not affect the over y ). This is prohibitively expensive a constant for any given arg max P ( x ) by sampling K to calculate, so we will approximate y uniformly from our responses including corpus during training: ( P ) x,y x ) = P ( | y (3) ∑ approx K ) P ( x,y k k =1 Combining equations 2 and 3 gives the approximate probability of the training data used to train the neural networks: S ) x,y ( e ) = x | y ( P (4) ∑ approx K ) S ( x,y k e =1 k The following subsections show several scoring models; how to extend the models to multiple features; how to overcome bias introduced by the sampling procedure; and an efficient search algorithm for response selection. 4.1 N-gram Representation x and responses y as fixed-dimensional input features, we extract n- To represent input emails d -dimensional embedding for each n-gram gram features from each. During training, we learn a jointly with the other neural network parameters. To represent sequences of words, we combine n-gram embeddings by summing their values. We will denote this bag of n-grams representation as d x ) ∈ R Ψ( . This representation is quick to compute and captures basic semantic and word ordering information. 4.2 Joint Scoring Model Figure 3a shows the joint scoring neural network model that takes the bag of n-gram representations x,y and the response y , and produces a scalar score S of the input email x ) . This deep neural ( network can model complex joint interactions between input and responses in its computation of the score. 4.3 Dot-Product Scoring Model Figure 3b shows the structure of the dot-product scoring model, where S ( x,y ) is factorized as a y dot-product between a vector . that depends only on x and a vector h h that depends only on y x This is similar to Deep Structured Semantic Models, which use feedforward networks to project queries and documents into a common space where the relevance of a document given a query is computed as the cosine distance between them [9]. While the interaction between features is not as direct as the joint scoring model (see section 4.2), this factorization allows us to calculate the representation of the input x and possible responses y

6 Efficient Natural Language Response Suggestion for Smart Reply Henderson et al. T h S ) = h ( x, y y x Wh S ( x, y ) = tanh layer tanh layer ReLU layer h y h h x ReLU layer tanh layer tanh layer ReLU layer tanh layer tanh layer ) x y ) Ψ( Ψ( ) Ψ( y ) Ψ( x (b) Dot-product architecture, where a tower (a) A neural network that calculates a score of tanh activation hidden layers encodes x between emails and their responses. Rec- h to h y tified Linear Unit (ReLU) layers are used to and a separate tower encodes to x y )-dimensional concatenation of d 2 reduce the ( ( such that the score S is the dot-product x, y ) T the bag of n-gram representations to a scalar h h . x y S . x, y ( ) Fig. 3. Feedforward scoring models that take the n-gram representation of an email body and a response, and compute a score. R can be precomputed. Then independently. In particular, the representations of the response set searching for response suggestions reduces to encoding a new email x in a simple feed-forward step h , and then searching for high dot-product scoring responses in the precomputed set to the vector x (see section 4.7). ( x It is also efficient to compute the scores S for all pairs of inputs and responses in a training ) ,y j i n examples, as that requires only an additional matrix multiplication after computing the h batch of x i h and vectors. This leads to vastly more efficient training with multiple negatives (see section 4.4) y i than is possible with the joint scoring model. 4.4 Multiple Negatives K possible responses is used to approximate P Recall from section 4 that a set of y | x ) – one ( correct response and K − 1 random negatives . For efficiency and simplicity we use the responses of other examples in a training batch of stochastic gradient descent as negative responses. For a batch of size , there will be K input emails x = ( x K ,...,x and their corresponding responses ) K 1 = ( y y ,...,y . The ) . Every reply y j is effectively treated as a negative candidate for x = if i 6 i K 1 j − 1 negative examples for each x are different at each pass through the data due to shuffling in K stochastic gradient descent. The goal of training is to minimize the approximated mean negative log probability of the data. For a single batch this is: ( x , J ,θ ) y K ∑ 1 ) = − x | log P y ( approx i i K i =1 (5)   K K ∑ ∑ 1 ) ,y S ( x j i   log ,y S ) − ( x e = − i i K i =1 =1 j using equation 4, where represents the word embeddings and neural network parameters used θ to calculate S . Note that this loss function is invariant to adding any function f ( , so ) to S ( x,y ) x

7 Efficient Natural Language Response Suggestion for Smart Reply Henderson et al. T h h x, y ( S ) = y x ( x, y S ) = Wh tanh layer tanh layer h y h x ReLU layer h tanh layer tanh layer ReLU layer tanh layer tanh layer ReLU layer ∑ ∑ M M i i 1 1 h h / / M M x y =1 =1 i i ⊕ M i h i =1 i x S ( ) , y i T i i , y ) x ( S h h = y x i i W = h tanh layer tanh layer ReLU layer i i i h h y h x ReLU layer tanh layer tanh layer ReLU layer tanh layer tanh layer i i Ψ( y ) x ) Ψ( ) y ) x Ψ( Ψ( ∀ i ∀ i (b) Dot-product scoring model with multiple (a) Joint scoring model using multiple features i i input features x of the input email x . This is a novel setup of the . A subnetwork scores multi-loss architecture, whereby the feature- the response using each feature alone, be- i i and the final score h , y fore the top-level hidden representations x ( S level scores ) ⊕ M i are computed as a dot-product be- ( S ) x, y ) and then used are concatenated ( h i =1 tween the parallel input and response sides. to compute the final score. This is an ap- plication of the multi-loss architecture from Al-Rfou et al. [2]. Fig. 4. Scoring models that use multiple features of the input email. ( x,y ) is learned up to an additive term that does not affect the S over y performed in the arg max inference time search. 4.5 Incorporating Multiple Features There is structure in emails that can be used to improve the accuracy of scoring models. We follow the multi-loss architecture of Al-Rfou et al. [ 2 ] to incorporate additional features beyond the message body, for example the subject line. Figure 4 shows the multi-loss architecture applied to both the joint and dot-product scoring models. The multi-loss networks have a sub-network for each feature of the email, which are trained to independently score candidate responses using that feature alone. The highest level hidden layer of the sub-network is used in a final sub-network that is trained to combine the information from all the features and give a final score. This hierarchical structure results in models that learn how to use each feature faster than a network that sees all the features at once, and also allows for learning deeper networks than is otherwise possible [2]. M 1 x , ..., x x features of an input email M Formally, denote the . Then for each i , a sub- as i i h , and a score of the response y using only x , network produces a hidden vector representation i i i i i i ( . Denoting x ) ) ,...,x ,y ,y ) as x x , a loss function J ( x ( , y ,θ ) encourages S ( x S to be high 1 K

8 Efficient Natural Language Response Suggestion for Smart Reply Henderson et al. Message : Did you manage to print the document? Without response bias With response bias – It’s printed. – Yes, I did. – Yes, it’s done. – I have printed it. – Yes, all done. – No, I didn’t. Table 1. Examples of Smart Reply suggestions with and without the response bias. Without biasing, the model prefers responses that are very closely related to the input email, but are less likely to be chosen than the more generic yes/no responses. for corresponding pairs in the training batch, and low for the random pairs. The second stage of the i x,y ) that is a function of all of the h network produces a final score vectors. The network is S ( trained end-to-end with a single loss: M ∑ i ) y , x ,θ ) + J ( x ( , y ,θ J =1 i Note that the final score produced by the multi-loss dot-product model (figure 4b) is a dot-product h that depends only on the input , and a vector h that depends only on the response of a vector x x y , as in the single-feature case. As a result, it can still be used for the fast vector search algorithm y described in section 4.7, and training with multiple negatives remains efficient. For the multi-loss joint scoring model, the input feature vector for the final sub-network is the i vectors and therefore scales with the number of features, leading to a h concatenation of the computational bottleneck. For the dot-product scoring model, the hidden layer representations are learned such that they are meaningful vectors when compared using a dot product. This motivates combining the representations for the final sub-network using vector arithmetic. The features ∑ M i i 1 extracted from the input email, x ), as are the response representations , are averaged ( M h / x i =1 ∑ M i 1 M / ), before being passed to the final neural network learned from different sub-networks ( h y =1 i layers. While this choice may constrain the representations learned by each sub-network, and may limit the ability of the final sub-network to differentiate information from different features, it also encourages them to exist in the same semantic space. 4.6 Response Biasing The discriminative objective function introduced in section 4.4 leads to a biased estimation of (1) . Since our negative examples are sampled from the training data the denominator in equation distribution, common responses with high prior likelihood appear more often as negative examples. In practice, we observed that this bias leads to models that favor specific and long responses instead R of short and generic ones. To encourage more generic responses, we bias the responses in using a score derived from the log likelihood of the response as estimated using a language model. Our final S score ( x,y ) of any input email response pair is calculated as: f S (6) ( x,y ) = S ) ( x,y ) + α log P y ( LM f m where S y is the score calculated by our trained scoring model, P is the probability of ( y ) LM m according to the language model, and α is tuned with online experiments. Note that the additional term is dependent only on y , and so can be precomputed for every response in R prior to inference time. Table 1 demonstrates the effect of including the response bias using an example email.

9 Efficient Natural Language Response Suggestion for Smart Reply Henderson et al. 4.7 Hierarchical Quantization for Efficient Search , we use the dot-product scoring model to find response x At inference time, given an input email R S ( x,y ) , where the scoring function is the dot-product: suggestions ∈ with the highest scores y 1 T h . The problem of finding datapoints with the largest dot-product values is some- h S ) = x,y ( y x times called Maximum Inner Product Search (MIPS). This is a research topic of its own and is also useful for inference in neural networks with a large number of output classes. Maximum Inner Product Search is related to nearest neighbor search (NNS) in Euclidean space, but comes with its own challenges because the dot-product “distance” is non-metric and many classical approaches such as KD-trees cannot be applied directly. For more background, we refer 5 readers to the relevant works of [ 21 , 22 ]. In the Smart Reply system, we need to keep very high 3 , , 99 % in top-30 retrieval). However, many of the existing methods are retrieval recall (for example > not designed to work well in the high recall regime without slowing down the search considerably. To achieve such high recall, hashing methods often require a large number of hash bits and tree methods often need to search a large number of leaves. In this work, we use a hierarchical quantization approach to solve the search problem. For our use R and thus the h y vectors are computed ahead of inference case, the responses come from a fixed set y ], we propose a hierarchical combination of vector quantization, time. Unlike the previous work in [ 5 orthogonal transformation and product quantization of the transformed vector quantization residuals. Our hierarchical model is based on the intuition that data in DNN hidden layers often resemble low dimensional signal with high dimensional residuals. Vector quantization is good at capturing low dimensional signals. Product quantization works by decomposing the high-dimensional vectors into ]. We use a learned rotation before low-dimensional subspaces and then quantizing them separately [ 4 product quantization as it has been shown to improve quantization error [16]. h is approximated by a hierarchical quantization HQ h Specifically, , which is the sum of the ) ( y y V Q ( h is ) and the residuals. A learned orthogonal transformation R vector quantization component y applied to the residual, followed by product quantization. T R h HQ ) = V Q ( h , ) + ( ≈ PQ ( r ) h y y y y = where R ( h r − V Q ( h )) y y y ( k ) C , product quantization codebooks of { C Here, given a vector quantization codebook } VQ PQ d d × k R ∈ , the vector quantization of R , and the learned orthogonal matrix for each of the subspaces 2 V Q ( h is ) = arg min h r || h . The product quantization of the rotated residual − c || is y y y y c ∈ C VQ ( k ) into K subvectors r computed by first dividing the rotated residuals r , and then K , k = 1 , 2 , ··· , y y ( ) k : quantizing the subvectors independently by vector quantizers C PQ k ) 2 ) ( k ) ( k ( − . ( || r || s PQ r ) = arg min y y ) k ( ∈{ } s C PQ PQ r Finally the full product quantization ) is given by the concatenation of the quantization in ( y each subspace:     (1) (1) (1) r PQ ( r ) y y     (2) (2) (2)     r PQ r ) ( y y     = , r PQ ( ) = r y y . .     . .     . . ( K ) ) ( K ) K ( r ) r ( PQ y y 1 The bias term, α log P h ( y ) , can be included in the dot product e.g. by extending the h and the vector with { α } x y LM vector with { log } ) ( y P LM

10 Efficient Natural Language Response Suggestion for Smart Reply Henderson et al. C At training time, the codebook for vector quantization, , codebooks for product quantization VQ · ( ) , and the rotation matrix R are jointly learned by minimizing the reconstruction error of C PQ − h ( h HQ ) with stochastic gradient descent (SGD). At inference time, prediction is made by y y taking the candidates with the highest quantized dot product, i.e. T T r V Q h ) ) + ( Rh ( ) h PQ ( x y y x HQ The distance computation can be performed very efficiently without reconstructing h ) , ( y instead utilizing a lookup table for asymmetric distance computation [ 10 ]. Furthermore, the lookup operation is carried out in register using SIMD (single instruction, multiple data) instructions in our implementation, providing a further speed improvement. 1.00 0.98 0.96 0.94 0.92 0.90 Top-30 recall 0.88 Hierarchical Quantization Clustering [3] 0.86 ALSH [20] 0.84 20 5 15 1 10 30 25 Speed up over exhaustive search Fig. 5. Evaluation of retrieval speed vs. recall of top-30 neighbors with maximum dot product. The curve is produced by varying the number of approximate neighbors retrieved by our hierarchical quantization method and by Asymmetric LSH [ 22 ], and varying the number of leaves searched by the clustering algorithm of [3]. We summarize the speed-recall evaluation of using different approximate MIPS algorithms in figure 5. The y-axis shows the recall of the top-30 retrieved responses, where the ground truth is computed by exhaustive search. The x-axis shows the speed up factor with respect to exhaustive search. Therefore, exhaustive search achieves a recall of 100% with a speed up factor of 1. Our algorithm achieves 99.89% recall with a speed-up factor over 10, outperforming the baselines of [3, 22]. 5 EVALUATION 5.1 Experimental Setup Data . Pairs of emails and their responses are sampled from user data to create datasets for training and testing the feedforward response scoring models. In total around 300M pairs are collected. The data is split uniformly at random into two disjoint sets of 95% and 5%, which constitute the training and test sets respectively. All email data (raw data, preprocessed data and training/evaluation data) is encrypted. Engineers can only inspect aggregated statistics on anonymized sentences that occurred across many users and do not identify any user. Language identification is run on the emails, and only English language emails are kept. The subject lines and message bodies are tokenized into word sequences, from which n-gram features are extracted. Infrequent words, URLs, email addresses, phone numbers etc. are replaced with special

11 Efficient Natural Language Response Suggestion for Smart Reply Henderson et al. tokens. Quoted text arising from replying and forwarding is also removed. We used hundreds of thousands of the most frequent n-grams as features to represent the text. Each of our DNN sub-networks consists of 3 hidden layers of sizes 500, 300, 100 in Training . the case of the joint scoring models and 300, 300, 500 for the dot-product models . The embedding d dimensionality of our n-grams is 320. We train each model for at least 10 epochs. We set the learning rate to 0.01 during the first 40 million batches, after which it is reduced to 0.001. The models are trained on CPUs across 50 machines using a distributed implementation of TensorFlow [1]. 5.2 Offline Evaluation Our models are evaluated offline on their ability to identify the true response to an email in the test data against a set of randomly selected competing responses. In this paper, we score a set of 100 responses that includes the correct response and 99 randomly selected incorrect competitors. We rank responses according to their scores, and report precision at 1 ( ) as a metric of evaluation. [email protected] We found that [email protected] correlates with the quality of our models as measured in online experiments with users (see section 5.3). Batch Size Scoring Model [email protected] Joint 49% 25 25 Dot-product 48% 50 Dot-product 52% Table 2. [email protected] results on the test set for the joint and dot-product multi-loss scoring models. The training objective discriminates against more random negative examples for larger batch sizes. Table 2 presents the results of the offline evaluation for joint and dot-product scoring models. The joint scoring model outperforms the dot-product model trained on the same batch size. This model learns complex cross-features between the input email and the response leading to a better scoring. However, the joint scoring model does not scale well to larger batches, since each possible pairing of input email and response requires a full forward pass through the network. The number of forward passes through the joint scoring model grows quadratically with the batch size. Recall the dot-product scoring model is a lot faster to train with multiple negatives than the joint scoring models K by K matrix multiply to since it requires a linear number of forward passes followed by a single score all possible pairings, where K is the batch size. As a result, the multi-loss dot-product models can be trained on larger batches to produce more accurate models. Note that the models in table 2 are trained with the multiple negatives training loss of section 4.4. It is also possible to train the models as a classifier with a sigmoid loss. We find that multiple negative training results in a consistent 20% reduction in error rate for [email protected] relative to training as a classifier across all of our conversational datasets. For example, in a different version of the Smart Reply data it improved the [email protected] of a dot-product model from 47% to 58%. 5.3 Online Evaluation Though the offline ranking metric gives a useful signal during development, the ultimate proof of a response selection model is how it affects the quality of the suggestions that users see in the end-to-end Smart Reply system. Suggestion quality or usefulness is approximated here by the observed conversion rate , i.e. the percentage of times users click on one of the suggestions when they are shown.

12 Efficient Natural Language Response Suggestion for Smart Reply Henderson et al. h Encode to x new email x DNN gives Dot product new email x score s = i search gives ( y S x, y ) ∀ ∈ Use heap to DNN gives score -best list M M -best list N find top x, y ) = ( s S i Encode to Response y scoring R ∈ ∀ y ∈ R ∀ h y ∈ R set R y Use heap to N find top Response scoring ∈ y precomputed R set -best list M setup first uses a fast (a) The setup scores all exhaustive search (b) The two pass M the examples in the response set using a joint - dot product scoring model to produce an scoring model. The top N best list of response suggestions from the re- scoring responses are then found using a heap. sponse set. The M -best list is then exhaus- tively searched to find the top N scoring re- sponses according to a more accurate joint scoring model. Encode to h x new email x Dot product search gives -best list N Encode to Response ∈ R y h ∀ set R y precomputed single pass (c) The setup uses a dot product scoring model and no joint scoring model. Fig. 6. Online system architectures. This section describes the evolution of our system and shows the effect of each iteration on latency and quality relative to the baseline Seq2Seq system. An outline of this series of online experiments is presented in table 3. 5.3.1 Exhaustive Search. Our initial system scored the input email against every response in R the response set Email body feature alone (see figure 6a). using the joint scoring model and the Given that the joint scoring model requires a forward pass for each response in R , this approach is too computationally expensive for an online experiment, see row 1 of table 3. 5.3.2 Two pass. The computational expense of the initial exhaustive search system motivated the design of a two-stage approach where the first stage is fast and the second is more accurate, as shown in figure 6b. The first stage consists of a dot-product scoring model utilizing the text of the email body alone. As a pre-processing step, all of the responses in the response set = { y R ,...,y are encoded to } n 1 their vector representations to give a matrix R = [ h (see figure 3b). At inference time, a ] ,..., h y y 1 n new input email is encoded to its representation h , and the vector of all scores is calculated as the x dot product with the precomputed matrix: Rh highest scoring . A heap is then used to find the M x responses. The second stage uses the joint scoring model to score the candidates from the first stage. Row 2 of table 3, shows the 50x speedup improvement from using this two pass system. The system tended to suggest overly specific and often long responses because of the biased negative sampling procedure, see section 4.6. Therefore, we added an extra score to boost the scores of more likely responses using a language model. This change significantly improved the quality of

13 Efficient Natural Language Response Suggestion for Smart Reply Henderson et al. Latency Conversion relative to Experiment System rate relative to Seq2Seq Seq2Seq Exhaustive Use a joint scoring model to score (1) 500% – . all responses in search R Two passes: dot-product then joint (2) 67% 10% scoring. Two pass (3) 88% 10% Include response bias. Improve sampling of dataset, and (4) 104% 10% use multi-loss structure. (5) 104% Remove second pass. 2% Single pass Use hierarchical quantization for (6) 104% 1% search. Table 3. Details of several successive iterations of the Smart Reply system, showing the conversion and latency relative to the baseline Seq2Seq rate system of Kannan et al. [11]. the suggestions, see row 3 of table 3, moving the systems toward shorter and more generic responses that users were more likely to find appropriate and click. Improving our dataset sampling, and using the multi-loss structure brought the conversion rate of the system above that of the Seq2Seq system, see row 4 of table 3). 5.3.3 Single pass. To improve the latency, we removed the second pass step and relied solely on the responses found by the first pass dot-product step (see figure 6c). However, to maintain the quality, we had to improve the quality of the dot-product model. Since the dot-product scoring model scales better with more negatives during training, we doubled the number of negatives for training the first pass system. We also applied the multi-loss architecture to the first pass dot-product model, using additional input features (see figure 4b). Together these changes made the dot-product model slightly more accurate than the joint model (see table 2). As a result, the system quality stayed the same while the speed increased 5 times, as shown in row 5 of table 3. So far, we have been computing the dot-product between the new email representation and all the precomputed representations of the responses in the response set, and searching the entire list to find high scoring responses. Switching from this exhaustive search to the hierarchical quantization search described in section 4.7 doubles the speed of the system without compromising quality (see row 6 of table 3). As a result, our final system produces better quality suggestions than the baseline Seq2Seq system with a small percentage of the computation and latency. 6 CONCLUSIONS This paper presents a feed-forward approach for scoring the consistency between input messages and potential responses. A hierarchy of deep networks on top of simple n-gram representations is shown to outperform competitive sequence-to-sequence models in this context.

14 Efficient Natural Language Response Suggestion for Smart Reply Henderson et al. The deep networks use different components for reading inputs and precomputing the representa- tion of possible responses. That architecture enables a highly efficient runtime search. We evaluate the models with the Smart Reply application. Live experiments with production traffic enabled a series of improvements that resulted in a system of higher quality than the original sequence-to-sequence system and a small fraction of the computation and latency. Without addressing the generation of novel responses, this paper suggests a minimal, efficient, and scalable implementation that enables many ranking-based applications. ACKNOWLEDGMENTS ̈ Thanks to Fernando Pereira, Corinna Cortes, Anjuli Kannan, Dilek Hakkani-T ur and Larry Heck for their valuable input to this paper. We would also like to acknowledge the many engineers at Google whose work on the tools and infrastructure made these experiments possible. Thanks especially to the users of Smart Reply. REFERENCES M. Abadi, P. Barham, J. Chen, Z. Chen, A. Davis, J. Dean, M. Devin, S. Ghemawat, G. Irving, M. Isard, et al. Tensorflow: [1] A system for large-scale machine learning. In USENIX Symposium on Operating Systems Design and Implementation (OSDI) , 2016. [2] R. Al-Rfou, M. Pickett, J. Snaider, Y. Sung, B. Strope, and R. Kurzweil. Conversational contextual cues: The case of arXiv preprint arXiv:1606.00372 , 2016. personalization and history for response ranking. A. Auvolat, S. Chandar, P. Vincent, H. Larochelle, and Y. Bengio. Clustering is efficient for approximate maximum [3] arXiv preprint arXiv:1507.05910 , 2015. inner product search. , 1(2):4–29, 1984. [4] R. M. Gray. Vector quantization. ASSP Magazine, IEEE [5] International R. Guo, S. Kumar, K. Choromanski, and D. Simcha. Quantization based fast inner product search. In Conference on Artificial Intelligence and Statistics , 2016. H. He, K. Gimpel, and J. J. Lin. Multi-perspective sentence similarity modeling with convolutional neural networks. In [6] Empirical Methods on Natural Language Processing (EMNLP) , 2015. M. Henderson, B. Thomson, and S. Young. Word-based dialog state tracking with recurrent neural networks. In Special [7] , 2014. Interest Group on Discourse and Dialogue (SIGDIAL) [8] S. Hochreiter and J. Schmidhuber. Long short-term memory. Neural Computation , 9(8), Nov. 1997. [9] P.-S. Huang, X. He, J. Gao, L. Deng, A. Acero, and L. Heck. Learning deep structured semantic models for web search using clickthrough data. 2013. [10] Pattern Analysis and Machine H. Jegou, M. Douze, and C. Schmid. Product quantization for nearest neighbor search. , 33(1), 2011. Intelligence ́ A. Kannan, K. Kurach, S. Ravi, T. Kaufman, B. Miklos, G. Corrado, A. Tomkins, L. Luk acs, M. Ganea, P. Young, and [11] V. Ramavajjala. Smart Reply: Automated response suggestion for email. In Conference on Knowledge Discovery and . ACM, 2016. Data Mining (KDD) [12] R. Kurzweil. . Penguin Books, New York, NY, USA, How to Create a Mind: The Secret of Human Thought Revealed 2013. [13] International Conference on Q. V. Le and T. Mikolov. Distributed representations of sentences and documents. In Machine Learning (ICML) , 2014. ̈ G. Mesnil, Y. Dauphin, K. Yao, Y. Bengio, L. Deng, X. He, L. Heck, G. Tur, D. Hakkani-T ur, D. Yu, and G. Zweig. [14] Using recurrent neural networks for slot filling in spoken language understanding. 2015. [15] arXiv T. Mikolov, K. Chen, G. Corrado, and J. Dean. Efficient estimation of word representations in vector space. preprint arXiv:1301.3781 , 2013. [16] Conference on Computer Vision and Pattern Recognition , pages M. Norouzi and D. J. Fleet. Cartesian k-means. In 3017–3024. IEEE, 2013. [17] J. Pennington, R. Socher, and C. D. Manning. Glove: Global vectors for word representation. In Empirical Methods on Natural Language Processing (EMNLP) , 2014. [18] P. J. Price. Evaluation of spoken language systems: The ATIS domain. In Workshop on Speech and Natural Language , HLT ’90. Association for Computational Linguistics, 1990. [19] I. V. Serban, A. Sordoni, Y. Bengio, A. Courville, and J. Pineau. Building end-to-end dialogue systems using generative hierarchical neural network models. In Conference on Artificial Intelligence . AAAI, 2016.

15 Efficient Natural Language Response Suggestion for Smart Reply Henderson et al. N. Shazeer, R. Doherty, C. Evans, and C. Waterson. Swivel: Improving embeddings by noticing what’s missing. arXiv [20] preprint arXiv:1602.02215 , 2016. [21] F. Shen, W. Liu, S. Zhang, Y. Yang, and H. Tao Shen. Learning binary codes for maximum inner product search. In International Conference on Computer Vision . IEEE, 2015. [22] A. Shrivastava and P. Li. Asymmetric LSH (ALSH) for sublinear time maximum inner product search (MIPS). In Advances in neural information processing systems (NIPS) , 2014. I. Sutskever, O. Vinyals, and Q. V. Le. Sequence to sequence learning with neural networks. In [23] Advances in neural information processing systems (NIPS) , 2014. [24] O. Vinyals, L. Kaiser, T. Koo, S. Petrov, I. Sutskever, and G. Hinton. Grammar as a foreign language. In Advances in neural information processing systems (NIPS) , 2015. [25] O. Vinyals and Q. V. Le. A neural conversational model. In International Conference on Machine Learning (ICML) , 2015. T.-H. Wen, D. Vandyke, N. Mrksic, M. Gasic, L. M. Rojas-Barahona, P.-H. Su, S. Ultes, and S. Young. A network-based [26] end-to-end trainable task-oriented dialogue system. , 2016. arXiv preprint arXiv:1604.04562 J. D. Williams and G. Zweig. End-to-end LSTM-based dialog control optimized with supervised and reinforcement [27] learning. , 2016. arXiv preprint arXiv:1606.01269 Y. Wu, M. Schuster, Z. Chen, Q. V. Le, M. Norouzi, W. Macherey, M. Krikun, Y. Cao, Q. Gao, K. Macherey, J. Klingner, [28] A. Shah, M. Johnson, X. Liu, L. Kaiser, S. Gouws, Y. Kato, T. Kudo, H. Kazawa, K. Stevens, G. Kurian, N. Patil, W. Wang, C. Young, J. Smith, J. Riesa, A. Rudnick, O. Vinyals, G. Corrado, M. Hughes, and J. Dean. Google’s neural machine translation system: Bridging the gap between human and machine translation. arXiv preprint arXiv:1609.08144 , 2016. K. Yao, G. Zweig, M.-Y. Hwang, Y. Shi, and D. Yu. Recurrent neural networks for language understanding. In [29] Interspeech , 2013. ̈ W. Yin, H. Sch [30] utze, B. Xiang, and B. Zhou. ABCNN: Attention-based convolutional neural network for modeling sentence pairs. Transactions of the Association for Computational Linguistics , 4, 2016. [31] S. Young. Talking to machines (statistically speaking). In Interspeech , 2002.

Related documents