Title: ALPHA-DIVERGENCE BRIDGES MAXIMUM LIKELI-HOOD AND REINFORCEMENT LEARNING IN NEURAL SEQUENCE GENERATION

Abstract: Neural sequence generation is commonly approached by using maximumlikelihood (ML) estimation or reinforcement learning (RL). However, it is known that they have their own shortcomings; ML presents training/testing discrepancy, whereas RL suffers from sample inefficiency. We point out that it is difficult to resolve all of the shortcomings simultaneously because of a tradeoff between ML and RL. In order to counteract these problems, we propose an objective function for sequence generation using α-divergence, which leads to an ML-RL integrated method that exploits better parts of ML and RL. We demonstrate that the proposed objective function generalizes ML and RL objective functions because it includes both as its special cases (ML corresponds to α → 0 and RL to α → 1). We provide a proposition stating that the difference between the RL objective function and the proposed one monotonically decreases with increasing α. Experimental results on machine translation tasks show that minimizing the proposed objective function achieves better sequence generation performance than ML-based methods.

1. INTRODUCTION:

Neural sequence models have been successfully applied to various types of machine learning tasks, such as neural machine translation (Cho et al., 2014; Sutskever et al., 2014; Bahdanau et al., 2015) , caption generation (Xu et al., 2015; Chen & Lawrence Zitnick, 2015) , conversation (Vinyals & Le, 2015) , and speech recognition (Chorowski et al., 2014; 2015; Bahdanau et al., 2016) . Therefore, developing more effective and sophisticated learning algorithms can be beneficial.

Popular objective functions for training neural sequence models include the maximum-likelihood (ML) and reinforcement learning (RL) objective functions. However, both have limitations, i.e., training/testing discrepancy and sample inefficiency, respectively. Bengio et al. (2015) indicated that optimizing the ML objective is not equal to optimizing the evaluation metric. For example, in machine translation, maximizing likelihood is different from optimizing the BLEU score (Papineni et al., 2002) , which is a popular metric for machine translation tasks. In addition, during training, ground-truth tokens are used for the predicting the next token; however, during testing, no ground-truth tokens are available and the tokens predicted by the model are used instead. On the contrary, although the RL-based approach does not suffer from this training/testing discrepancy, it does suffer from sample inefficiency. Samples generated by the model do not necessarily yield high evaluation scores (i.e., rewards), especially in the early stage of training. Consequently, RL-based methods are not self-contained, i.e., they require pre-training via ML-based methods. As discussed in Section 2, since these problems depend on the sampling distributions, it is difficult to resolve them simultaneously.

Our solution to these problems is to integrate these two objective functions. We propose a new objective function α-DM (α-divergence minimization) for a neural sequence generation, and we demonstrate that it generalizes ML-and RL-based objective functions, i.e., α-DM can represent both functions as its special cases (α → 0 and α → 1). We also show that, for α ∈ (0, 1), the gradient of the α-DM objective is a combinations of the ML-and RL-based objective gradients. We apply the same optimization strategy as Norouzi et al. (2016) , who useed importance sampling, to optimize this proposed objective function. Consequently, we avoid on-policy RL sampling which suffers from sample inefficiency, and optimize the objective function closer to the desired RL-based objective than the ML-based objective.

The experimental results for a machine translation task indicate that the proposed α-DM objective outperforms the ML baseline and the reward augmented ML method (RAML; Norouzi et al., 2016) , upon which we build the proposed method. We compare our results to those reported by Bahdanau et al. (2017) , who proposed an on-policy RL-based method. We also confirm that α-DM can provide a comparable BLEU score without pre-training.

The contributions of this paper are summarized as follows.

• We propose the α-DM objective function using α-divergence and demonstrate that it can be considered a generalization of the ML-and RL-based objective functions (Section 4). • We prove that the α-DM objective function becomes closer to the desired RL-based objectives as α increases in the sense that the upper bound of the maximum discrepancy between ML-and RL-based objective functions monotonically decreases as α increases.

• The results of machine translation experiments demonstrate that the proposed α-DM objective outperforms the ML-baseline and RAML (Section 7).

2. COMPARING OBJECTIVE FUNCTIONS:

In this section, we introduce ML-based and RL-based objective functions and the problems in association with learning neural sequence models using them. We also explain why it is difficult to resolve these problems simulataneously.

Maximum-likelihood An ML approach is typically used to train a neural sequence model. Given a context (or input sequence) x ∈ X and a target sequence y = (y 1 , . . . , y T ) ∈ Y, ML minimizes the negative log-likelihood objective function

EQUATION

where q(y|x) denotes the true sampling distribution. Here, we assume that x is uniformly sampled from X and omit the distribution of x from Eq. (1) for simplicity. For example, in machine translation, if a corpus contains only a single target sentence y * for each input sentence x, then q(y|x) = δ(y -y * ) and the objective becomes L(θ) = -x∈X log p θ (y * |x).

ML does not directly optimize the final performance measure; that is, training/testing discrepancy exists. This arises from at leset these two problems:

(i) Objective score discrepancy. The reward function is not used while training the model; however, it is the performance measure in the testing (evaluation) phase. For example, in the case of machine translation, the popular evaluation measures such as BLEU or edit rate (Snover et al., 2006) differ from the negative likelihood function. (ii) Sampling distribution discrepancy. The model is trained with samples from the true sampling distribution q(y|x); however, it is evaluated using samples generated from the learned distribution p θ (y|x).

Reinforcement learning In most sequence generation task, the optimization of the final performance measure can be formulated as the minimization of the negative total expected rewards expressed as follows:

EQUATION

where r(y, y * |x) is a reward function associated with the sequence prediction y, i.e., the BLEU score or the edit rate in machine translation. RL is an approach to solve the above problems. The objective function of RL is L * in Eq. ( 2), which is a reward-based objective function; thus, there is no objective score discrepancy, thereby resolbing problem (i). Sampling from p θ (y|x) and taking the expectation with p θ (y|x) in Eq. ( 2) also resolves problem (ii). Ranzato et al. ( 2016) and Bahdanau et al. (2017) directly optimized L * using policy gradient methods (Sutton et al., 2000) . A sequence prediction task that selects the next token based on an action trajectory (y 1 , . . . , y t-1 ) can be considered to be an RL problem. Here the next token selection corresponds to the next action selection in RL. In addition, the action trajectory and the context x correspond to the current state in RL.

RL can suffer from sample inefficiency; thus, it may not generate samples with high rewards, particularly in the early learning stage. By definition, RL generates training samples from its model distribution. This means that, if model p θ (y|x) has low predictive ability, only a few samples will exist with high rewards.

(iii) Sample inefficiency. The RL model may rarely draw samples with high rewards, which hinders to find the true gradient to optimize the objective function.

Machine translation suffers from this problem because the action (token) space is vast (typically >10, 000 dimensions) and rewards are sparse, i.e., positive rewards are observed only at the end of a sequence. Therefore, the RL-based approach usually requires good initialization and thus is not self-contained. Previous studies have employed pre-training with ML before performing on-policy RL-based sampling (Ranzato et al., 2016; Bahdanau et al., 2017) .

Entropy regularized RL To prevent the policy from becoming overly greedy and deterministic, some studies have used the following entropy-regularized version of the policy gradient objective function (Mnih et al., 2016) :

EQUATION

Note that lim τ →0 L * (τ ) = L * holds.

Reward augmented ML Norouzi et al. (2016) proposed RAML, which solves problems (i) and (iii) simultaneously. RAML replaces the sampling distribution of ML, i.e., q(y|x) in Eq. ( 1), with a reward-based distribution q (τ ) (y|x) ∝ exp {r(y, y * |x)/τ }. In other words, RAML incorporates the reward information into the ML objective function. The RAML objective function is expressed as follows:

EQUATION

However, problem (ii) remains.

Despite these various attempts, a fundamental technical barrier exists. This barrier prevents solving the three problems using a single method. The barrier originates from a trade-off between sampling distribution discrepancy (ii) and sample inefficiency (iii), because these issues are related to the sampling distribution. Thus, our approach is to control the trade-off of the sampling distributions by combining them.

3. α-DIVERGENCE:

The proposed method utilizes α-divergence D (α)

A (p q), which measures the asymmetric distance between two distributions p and q (Amari, 1985) . A prominent feature of α-divergence is that it can behave as D KL (p q) or D KL (q p) depending on the value of α, i.e., D

(1)

EQUATION

where

log (α) (•) is the generalized logarithm log (α) (x) := (1 -α) -1 (x 1-α -1)

. Furthermore, α-divergence becomes a Hellinger distance when α equals to 1/2.

4. PROPOSED OBJECTIVE FUNCTION: α-DM:

In this section, we describe the proposed objective function α-DM and its gradient. Furthermore, we demonstrate that it can smoothly bridge both ML-and RL-based objective functions. We define the α-DM objective function as the α-divergence between p θ and q (τ ) :

4.1. OBJECTIVE FUNCTION:

α → 0 α → 1 0 < α < 1 τ = 0 τ > 0

EQUATION

)

This α-divergence is equal to L * (τ ) in Eq. ( 3) or L (τ ) in Eq. ( 4) by employing α → 1 or α → 0 limits, respectively (up to constant).

EQUATION

EQUATION

)

Figure 1 illustrates how the α-DM objective bridges the ML-and RL-based objective functions.

Although the objectives L * (α,τ ) (θ), L * (τ ) (θ), and L (τ ) (θ) have the same global minimizer p θ (y|x) = q (τ ) (y|x), empirical solutions often differ.

4.2. OBJECTIVE FUNCTION GRADIENT:

To train neural network or other machine learning models via α-divergence minimization, one can use the gradient of α-DM objective function. The gradient of Eq. ( 6) can be expressed as

EQUATION

where p

EQUATION

is a weight that mixes sampling distributions p θ and q (τ ) . This weight makes it clear that the α-DM objective can be considered as a mixture of ML-and RL-based objective functions. See Appendix A for the derivation of this gradient. It converges to the gradient of entropy regularized RL or RAML by taking α → 1 or α → 0 limits, respectively (up to constant); i.e.,

lim α→1 ∇ θ L (α,τ ) = ∇ θ L * (τ )

and

lim α→0 ∇ θ L (α,τ ) = τ ∇ θ L (τ ) .

In Appendix C, we summarize all of the objective functions, gradients, and their connections.

5. α-DM ANALYSIS:

In this section, we characterize the difference between α-DM objective function L (α,τ ) and the desired RL-based objective function L * (τ ) with respect to sup-norm. Our main claim is that, with respect to sup-norm, the discrepancy between L (α,τ ) and L * (τ ) decreases linearly as α increases to 1. We utilize this analysis to motivate our α-DM objective function with larger α if there are no concerns about the sampling inefficiency.

Proposition 1 Assume that p θ has the same finite support S as that of q (τ ) , and that for any s ∈ S, there exists δ > 0 such that p θ (s) > δ holds. For any α ∈ (0, 1), the following holds.

EQUATION

)

where L(α,τ) := αL (α,τ ) . Here, C 1 , C 2 is universal constants irrelevant to α.

The following proposition immediately proves the theorem above.

Proposition 2 Assume that probability distribution p has the same finite support S as that of q, and that for any s ∈ S there exists δ > 0 such that p(s) > δ holds. For any α ∈ (0, 1), the following holds.

EQUATION

)

Here, C = max sup p p log 2 (q/p) , sup p q log 2 (q/p) .

For the proof of the Proposition 1 and Proposition 2, see Appendix B.

6. OPTIMIZATION OF α-DM OBJECTIVE FUNCTION:

In this paper, we employed the optimization strategy which is similar to that of RAML. We sample target sentence y for each x from another data augmentation distribution q 0 (y|x), and then estimate the gradient by importance sampling (IS). For example, we add some noise to the ground truth target sentence y * by insertion, substitution, or deletion, and the distribution p 0 (y|x) assigns some probability to each modified target sentence. Given samples from this proposal ditribution p 0 (y|x), we update the parameter using the following IS estimator

EQUATION

-

EQUATION

Here, {(x 1 , y 1 ), . . . , (x N , y N )} are the N samples from the proposal distribution q 0 (y|x), and w i is the importance weight which is proportional to p (α,τ ) θ (y i |x i ):

EQUATION

Note that the difference betweene RAML and α-DM is only this importance weight w i . In RAML, w i depends only on q (τ ) (y i |x i ) but not on p θ (y i |x i ). We normalize w i in each minibatch in order to use same hyperparameter (e.g., learning rate) as ML baseline. Thus, this estimator becomes a weighted IS estimator. A weighted IS estimator is not unbiased, yet but it has smaller variance. Also, we found that normalizing q (τ ) (y i |x i ) and p θ (y i |x i ) in each minibatch leads to good results.

7. NUMERICAL EXPERIMENTS:

We evaluate the effectiveness of α-DM experimentally using neural machine translation tasks. We compare the BLEU scores of ML, RAML, and the proposed α-DM on the IWSLT'14 German-English corpus (Cettolo et al., 2014) . In order to evaluate the impact of training objective function, we train the same attention-based encoder-decoder model (Bahdanau et al., 2015; Luong et al., 2015) for each objective function. Furthermore, we use the same hyperparameter (e.g., learning rate, dropout rate, and temperature τ ) between all the objective functions. For RAML and α-DM, we employ a data augmentation procedure similar to that of Norouzi et al. (2016) , and thus we generate samples from a data augmentation distribution q 0 (y|x). Note that the difference between RAML and α-DM is only the weight w i of Eq. ( 14). The details of data augmentation distribution are described in Section 7.2.

Figure 2 : Performance on different hyperparameters α. The α-DM approach with larger α performs better than that with smaller α. Bahdanau et al. (2017) . Specifically, we trained attention-based encoder-decoder model with the encoder of a bidirectional LSTM with 256 units and the LSTM decoder with the same number of layers and units. We exponentially decay the learning rate, and the initial learning rate is chosen using grid search to maximize the BLEU performance of ML baseline on development dataset. The important hyperparameter τ of RAML and α-DM is also determined to maximize the BLEU performance of RAML baseline on development dataset. As a result, the initial learning rate of 0.5 and τ of 1.0 were used. Our α-DM used the same hyperparameters as ML and RAML including the initial learning rate, τ , and so on. Details about the models and parameters are discussed in Section 7.2.

7.1. RESULTS:

To investigate the impact of hyperparameter α, we train the neural sequence models using α-DM 5 times for each fixed α ∈ {0.0, 0.1, . . . , 0.9}, and then reported the BLEU score of test dataset. Moreover, assuming that the underfitted model prevents the gradient from being stable in the early stage of training, we train the same models with α being linearly annealed from 0.0 to larger values; we increase the value of α by adding 0.03 at each epoch. Here, the beam width k was set to 1 or 10. All BLEU scores and their averages are plotted in Figure 2 . The results show that for both k = 1, 10, the models performance are better than smaller or larger α when α is around 0.5 (α = 0.5). However, for larger fixed α, the performance was worse than RAML and ML baselines. On the other hand, we can see that the annealed versions of α-DM improve the performance of the corresponding fixed versions in relatively larger α. As a result, in the annealed scenario, α-DM with wide range of α ∈ (0, 1) improves on the performance consistently. This implies that the underfitted model makes the performance worse.

We summarize the average BLEU scores and their standard deviation of ML, RAML, and α-DM with α ∈ {0.3, 0.4, 0.5} in Table . 1. The result shows that the BLEU score (k = 10) of our α-DM outperforms ML and RAML baseline. Furthermore, although the ML baseline performances differ between our results and those of Bahdanau et al. (2017) , the proposed α-DM performance with α = 0.5 without pre-training is comparable with the on-policy RL-based methods (Bahdanau et al., 2017) . We believe that these results come from the fact that α-DM with α > 0 has smaller bias than that of α = 0 (i.e., RAML).

7.2. DETAILS:

We utilized a stochastic gradient descent with a decaying learning rate. The learning rate decays from the initial learning rate to 0.05 with dev-decay (Wilson et al., 2017) , i.e., after training each epoch, we monitored the perplexity for the development set and reduced the learning rate by multiplying it with δ = 0.5 only when the perplexity for the development set does not update the best perplexity. The mini-batch size is 128. We used the dropout with probability 0.3. Gradients are rescaled when the norms exceed 5. In addition, if an unknown token, i.e., a special token representing a word that is not in the vocabulary, is generated in the predicted sentence, it was replaced by the token with the highest attention in the source sentence (Jean et al., 2015) . We implemented our models using a fork from the PyTorch1 version of the OpenNMT toolkit (Klein et al., 2017) . We calculated the BLEU scores with multi-bleu.perl2 script for both the development and test sets.

We obtained augmented data in the same manner as the RAML framework (Norouzi et al., 2016) . For each target sentence, some tokens were replaced by other tokens in the vocabulary and we used the negative Hamming distance as reward. We assumed that Hamming distance e for each sentence is less than [m × 0.25], where m is the length of the sentence and [a] denotes the maximum integer which is less than or equal to a ∈ R. Moreover, the Hamming distance for a sample is uniformly selected from 0 to [m × 0.25]. One can also use BLEU or another machine translation metric for this reward. However, we assumed proposal distribution q 0 (y|x) different from that of RAML. We assumed the simplified proposal distribution q 0 (y|x), which is a discrete uniform distribution over [0, m × 0.25]. This results in hyperparameter τ used in this experiment being different from that of RAML. We search the τ , which maximize the BLEU score of RAML on the development set. As a results, τ = 1.0 was chosen, and α-DM also uses this fixed τ in all the experiments.

8. RELATED WORKS:

From the RL literature, reward-based neural sequence model training can be separated into on-policy and off-policy approaches, which differ in the sampling distributions. The proposed α-DM approach can be considered an off-policy approach with importance sampling.

Recently, on-policy RL-based approaches for neural sequence predictions have been proposed. Ranzato et al. (2016) proposed a method that uses the REINFORCE algorithm (Williams, 1992) . Based on Ranzato et al. ( 2016), Bahdanau et al. (2017) proposed a method that estimates a critic network and uses it to reduce the variance of the estimated gradient. Bengio et al. (2015) proposed a method that replaces some ground-truth tokens in an output sequence with generated tokens. Yu et al. ( 2017), Lamb et al. (2016), and Wu et al. (2017) proposed methods based on GAN (generative adversarial net) approaches (Goodfellow et al., 2014) . Note that on-policy RL-based approaches can directly optimize the evaluation metric. Degris et al. (2012) proposed off-policy gradient methods using importance sampling, and the proposed α-DM off-policy approach utilizes importance sampling to reduce the difference between the objective function and the evaluation measure when α > 0.

As mentioned previously, the proposed α-DM can be considered an off-policy RL-based approach in that the sampling distribution differs from the model itself. Thus, the proposed α-DM approach has the same advantages as off-policy RL methods compared to on-policy RL methods, i.e., computational efficiency during training and learning stability. On-policy RL approaches must generate samples during training, and immediately utilize these samples. This property leads to high computational costs during training and if the model falls into a poor local minimum, it is difficult to recover from this failure. On the other hand, by exploiting data augmentation, the proposed α-DM can collect samples before training. Moreover, because the sampling distribution is a stationary distribution independent of the model, one can expect that the learning process of α-DM is more stable than that of on-policy RL approaches. Several other methods that compute rewards before training can be considered off-policy RL-based approaches, e.g., minimum risk training (MRT; Shen et al., 2016 , RANDOMER (Guu et al., 2017) , and Google neural machine translation (GNMT; Wu et al., 2016) .

While the proposed approach is a mixture of ML-and RL-based approaches, this attempt is not unique. The sampling distribution of scheduled sampling (Bengio et al., 2015) is also a mixture of ML-and RL-based sampling distributions. However, the sampling distributions of scheduled sampling can differ even in the same sentence, whereas ours are sampled from a stationary distribution. To bridge the ML-and RL-based approaches, Guu et al. (2017) considered the weights of the gradients of the ML-and RL-based approaches by directly comparing both gradients. In contrast, the weights of the proposed α-DM approach are obtained as the results of defining the α-divergence objective function. GNMT (Wu et al., 2016) considered a mixture of ML-and RL-based objective functions by the weighted arithmetic sum of L and L * . Comparing this weighted mean objective function and α-DM's objective function could be an interesting research direction in future.

9. CONCLUSION:

In this study, we have proposed a new objective function as α-divergence minimization for neural sequence model training that unifies ML-and RL-based objective functions. In addition, we proved that the gradient of the objective function is the weighted sum of the gradients of negative loglikelihoods, and that the weights are represented as a mixture of the sampling distributions of the ML-and RL-based objective functions. We demonstrated that the proposed approach outperforms the ML baseline and RAML in the IWSLT'14 machine translation task.

In this study, we focus our attention on the neural sequence generation problem, but we expect our framework may be useful to broader area of reinforcement learning. The sample inefficiency is one of major problems in reinforcement learning, and people try to mitigiate this problem by using several type of supervised learning frameworks such as imitation learning or apprenticisip learning. This alternative approaches bring another problem similar to the neural sequence generaton problem that is originated from the fact that the objective function for training is different from the one for testing. Since our framework is general and independent from the task, our approach may be useful to combine these approaches.

Lantao Yu, Weinan Zhang, Jun Wang, and Yong Yu. SeqGAN: Sequence generative adversarial nets with policy gradient. In Proceedings of AAAI Conference on Artificial Intelligence (AAAI), 2017.

A GRADIENT OF α-DM OBJECTIVE

The gradient of α-DM can be obtained as follows:

EQUATION

where

EQUATION

In Eq. ( 18), we used the so-called log-trick: ∇ θ p θ (y|x) = p θ (y|x)∇ θ log p θ (y|x).

B PROOFS OF PREPOSITIONS:

Proposition 1 Assume that probability distribution p has the same finite support S as that of q, and that for any s ∈ S there exists δ > 0 such that p(s) > δ holds. For any α ∈ (0, 1) the following holds.

EQUATION

Here, C = max sup p p log 2 (q/p) , sup p q log 2 (q/p) .

Proof. By Taylor's theorem, there is an α ∈ (α, 1) such that

EQUATION

A (p q) := -

EQUATION

Therefore, by Jensen's inequality we have

EQUATION

≤ (1 -α) sup p (α p + (1 -α )q) log 2 q p (30)

≤ (1 -α) α sup p p log 2 q p + (1 -α ) sup p q log 2 q p (31)

EQUATION

where C = max sup p p log 2 (q/p) , sup p q log 2 (q/p) .

Proposition 2 Assume that p θ has the same finite support S as that of q (τ ) , and that for any s ∈ S, there exists δ > 0 such that p θ (s) > δ holds. For any α ∈ (0, 1), the following holds.

EQUATION

)

where L(α,τ) := αL (α,τ ) . Here, C 1 , C 2 is universal constants irrevant to α. 

EQUATION

where C 1 = τ max sup θ x∈X y∈Y p θ log 2 (q (τ ) /p θ ) , sup θ x∈X y∈Y q (τ ) log 2 (q (τ ) /p θ ) and C 2 = |Z(τ )|.

C CATALOG OF OBJECTIVE FUNCTIONS AND THEIR GRADIENTS:

In this section, we summarize the objective functions of

• ML (Maximum Likelihood),

• RL (Reinforcement Learning),

• RAML (Reward Augmented Maximum Likelihood; Norouzi et al., 2016) ,

• EnRL (Entropy regularized Reinforcement Learning), and 

EQUATION

where q (τ ) (y|x) ∝ exp {r(y, y * |x)/τ }. Typically, q(y|x) = δ(y, y * |x) where y * is the target with the highest reward.

We can rewrite some of these functions using KL or α-divergences: A (p θ q (τ ) ).

EQUATION

(46)

In the limits, there are the following connections between the objectives. 

:

http://pytorch.org

https://github.com/moses-smt/mosesdecoder/blob/master/scripts/generic/ multi-bleu.perl

Figures and Tables:

Figure 1: α-DM objective bridges ML-and RL-based objectives.

(p θ |q (τ ) ) -Z(τ ) where Z(τ ) := x∈X y∈Y exp(r(y, y * |x)/τ ). By Proposition 1 we have sup KL (p θ |q (τ ) ) -Z(τ ) -ατ x D α A (p θ |q (τ ) ) KL (p θ |q (τ ) ) -αD α A (p θ |q (τ ) ) + |Z(τ )| (37)

α-DM (α-Divergence Minimization Training).Objectives. The objective functions of ML, RL, RAML, EnRL, and α-DM are as followsL(θ) = -x∈X y∈Y q(y|x) log p θ (y|x),(39)L * (θ) = -x∈X y∈Y p θ (y|x)r(y, y * ),(40)L (τ ) (θ) = -x∈X y∈Yq (τ ) (y|x) log p θ (y|x), y|x) r(y, y * |x) -τ log p θ (y|x) ,

L * (τ ) (θ) = τ x∈X D KL (p θ q (τ ) ) + constant,(45)L (α,τ ) (θ) = τ x∈X D (α)

(α,τ ) (θ) = τ L (τ ) (θ) + constant, (49) lim α→1

BLEU on IWSLT'14 German-English.

Table content: <table><tr><td/><td>k = 1</td><td>k = 10</td></tr><tr><td>ML</td><td colspan="2">27.59 (±0.18) 28.15 (±0.16)</td></tr><tr><td>RAML</td><td colspan="2">27.74 (±0.06) 28.44 (±0.13)</td></tr><tr><td>α-DM, α = 0.3 fixed</td><td colspan="2">27.92 (±0.10) 28.51 (±0.10)</td></tr><tr><td>α-DM, α = 0.4 fixed</td><td colspan="2">28.05 (±0.07) 28.65 (±0.06)</td></tr><tr><td>α-DM, α = 0.5 fixed</td><td colspan="2">28.01 (±0.16) 28.52 (±0.10)</td></tr><tr><td colspan="3">α-DM, α = 0.3 annealed 27.99 (±0.11) 28.60 (±0.09)</td></tr><tr><td colspan="3">α-DM, α = 0.4 annealed 27.93 (±0.11) 28.51 (±0.15)</td></tr><tr><td colspan="3">α-DM, α = 0.5 annealed 27.98 (±0.11) 28.48 (±0.15)</td></tr><tr><td colspan="3">IWSLT'14 German-English. The training data comprised approximately 153K German-English</td></tr><tr><td colspan="3">sentence pairs and 7K development/test sentence pairs. The vocabulary size for the source/target</td></tr><tr><td colspan="3">were 32 009 and 22 822, respectively. The model architecture and parameters follow that of Ranzato</td></tr><tr><td>et al. (2016) and</td><td/></tr></table>

Peer review (iclr2018):
 The paper proposes another training objective for training neural sequence-to-sequence models. The objective is based on alpha-divergence between the true input-output distribution q and the model distribution p. The new objective generalizes Reward-Augmented Maximum Likelihood (RAML) and entropy-regularized Reinforcement Learning (RL), to which it presumably degenerates when alpha goes to 1 or to 0 respectively. The paper has significant writing issues. In Paragraph “Maximum Likelihood”, page 2, the formalization of the studied problem is unclear. Do X and Y denote the complete input/output spaces, or do they stand for the training set examples only? In the former case, the statement “x is uniformly sampled from X” does not make sense because X is practically infinite. Same applies to the dirac distribution q(y|x), the true conditional distribution of outputs given inputs is multimodal even for machine translation. If X and Y were meant to refer to the training set, it would be worth mentioning the existence of the test set. Furthermore, in the same Section 2 the paper fails to mention that reinforcement learning training also does not completely correspond to the evaluation approach, at which stage greedy search or beam search is used. The proposed method is evaluated on just one dataset. Crucially, there is no comparison to a trivial linear combination of ML and RL, which in one way or another was used in almost all prior work, including GNMT, Bahdanau et al, Ranzato et al. The paper does not argue why alpha divergence is better that the aforementioned combination method and also does not include it in the comparison. To sum up, I can not recommend the paper to acceptance, because (a) an important baseline is missing (b) there are serious writing issues.