Title: ROLE OF TWO LEARNING RATES IN CONVERGENCE OF MODEL-AGNOSTIC META-LEARNING Anonymous authors Paper under double-blind review

Abstract: Model-agnostic meta-learning (MAML) is known as a powerful meta-learning method. However, MAML is notorious for being hard to train because of the existence of two learning rates. Therefore, in this paper, we derived a sufficinent condition of inner learning rate α and meta-learning rate β for a simplified MAML to locally converge to local minima from any point in the vicinity of the local minima. We find that the upper bound of β depends on α. Moreover, we show that the threshold of β increases as α approaches its own upper bound. This result is verified by experiments on various few-shot tasks and architectures; specifically, we perform sinusoid regression and classification of Omniglot and MiniImagenet datasets with a multilayer perceptron and a convolutional neural network. Based on this outcome, we present a guideline for determining the learning rates: first, search for the largest possible α; next, tune β based on the chosen value of α.

1. INTRODUCTION:

A pillar of human intelligence is the ability to learn and adapt to unseen tasks quickly and based on only a limited quantity of data. Although machine learning has achieved remarkable results, many recent models require massive quantities of data and are designed for solving particular tasks. Meta-learning, one of the ways of tackling this problem, tries to develop a model that can adapt to new tasks quickly by learning to learn new concepts from few data points (Schmidhuber, 1987; Thrun & Pratt, 1998) .

Among meta-learning algorithms, model-agnostic meta-learning (MAML), a gradient-based metalearning method proposed by Finn et al. (2017) , has recently been extensively studied. For example, MAML is used for continual learning (Finn et al., 2019; Jerfel et al., 2019; Spigler, 2019; Al-Shedivat et al., 2018) , reinforcement learning (Finn et al., 2017; Al-Shedivat et al., 2018; Gupta et al., 2018; Deleu & Bengio, 2018; Liu & Theodorou, 2019) and probablistic inference (Finn et al., 2018; Yoon et al., 2018; Grant et al., 2018) . The reason why MAML is widely used is because MAML is simple but efficient and applicable to a wide range of tasks independent of model architecture and the loss function. However, MAML is notorious for being hard to train (Antoniou et al., 2019) . One of the reasons why training MAML is hard is the existence of two learning rates in MAML: the inner learning rate α and meta-learning rate β. A learning rate is known to be one of the most important parameters, and tuning this parameter may be challenging even if the simple gradient descent (GD) method is used. Nevertheless, we do not yet know the relationship between these two learning rates and have little guidance on how to tune them. Hence, guidelines for choosing these parameters are urgently needed.

In this paper, we investigate the MAML algorithm and propose a guideline for selecting the learning rates. First, in Section 2 we briefly explain by using an approximation how MAML can be regarded as optimization with the negative gradient penalty. Because the gradient norm is related to the shape of the loss surface, a bias towards a larger gradient norm can make training unstable. Next, based on the approximation explained in Section 2, in Section 3, we derive a sufficinent condition of α and β for a simplified MAML to locally converge to local minima from any point in the neighborhood of the local minima. Furthermore, by removing a constraint, we derive a sufficient condition for local convergence with fewer simplifications as well. We find that the upper bound β c of meta-learning rate depends on inner learning rate α. In particular, β c of α ≈ α c is larger than that of α = 0, where α c is the upper bound of α. This is verified by experiments in Section 5. These results imply a guideline for selecting the learning rates: first, search for the largest possible α; next, tune β.

2. MAML AS OPTIMIZATION WITH NEGATIVE GRADIENT PENALTY:



2.1. MAML:

The goal of MAML is to find a representation that can rapidly adapt to new tasks with a small quantity of data. In other words, MAML performs optimization for parameters θ ∈ R d that the optimizer can use to quickly reach the optimal parameter θ * τ for task τ with few data points. To this end, MAML takes the following steps to update θ. First, it samples a batch of tasks from task distribution P (τ ) and updates θ for each task τ with stochastic gradient descent (SGD). Although MAML allows multiple-step being taken to update θ, we will consider the case only one step being taken for simplicity. The update equation is as follows:

EQUATION

)

where α is a step size referred to as the inner learning rate, L τ (θ) is the loss of τ , and ∇ θ L τ (θ) is an estimate of the true gradient. The data used for this update is called training data. Next, MAML resamples data, referred to as test data, from each τ and computes the loss at the updated parameters θ ′ τ , obtaining L τ (θ ′ τ ) for each task. Finally, to determine θ that can be adapted to θ ′ τ for all tasks, θ is updated with the gradient of a sum of loss values L τ (θ ′ τ ) over all tasks. In other words,

EQUATION

)

where β is the learning rate called the meta-learning rate and

∇ θ ∑ τ ∼P (τ ) L τ (θ ′ τ

) is an estimate of the true gradient by using the test data. Though learning rates α and β can be tuned during training or different for each task in practice, we will think them as fixed scalar hyperparameters.

2.2. NEGATIVE GRADIENT PENALTY:

Unless otherwise noted, we will consider the case of only one step being made per update, and the data are not resampled to compute the loss for updating θ. The case of multiple steps and that of training data and test data being separated are considered in Appendix A. The gradient of the loss at

θ ′ τ is g τ (θ ′ τ ) = ∇ θ L τ (θ ′ τ ) = ∇ θ θ ′ τ ∂Lτ ∂θ ′ τ

, where g(•) is the gradient of L(•) with respect to θ. If α is small, we can assume that I ∂Lτ ∂θ ′ τ = g τ (θ); this seems to hold since α is usually small (Finn et al., 2017) . Then,

EQUATION

The result but not the procedure with the approximation is the same as that with the well-known firstorder approximation as long as data are not resampled and only one step being taken. The first order approximation has been mentioned by Finn et al. (2017) and extensively studied by Nichol et al. (2018) and Fallah et al. (2019) . It is known that the error induced by the first-order approximation does not degrade the performance so much in practice (Finn et al., 2017) . Also, Fallah et al. (2019) theoretically proved that the first-order approximation of MAML does not affect convergence result when each task is similar to each other or α is small enough.

For simplicity, we will assume that only one task is considered during training, omitting task index τ . Therefore, instead of ∑ τ ∼P (τ ) L τ (θ ′ τ ), we will consider L(θ ′ ) as the loss of the simplified MAML. Since the MAML loss is just a sum of task-specific loss, extension to the case of multiple tasks being considered is straightforward, as provided in Appendix A.1.1. Because ∇ θ (g(θ

EQUATION

)

The above means that the simplified MAML can be regarded as optimization with the negative gradient penalty. We will analyze this simplified MAML loss in Section 3. It can also be interpreted as a Taylor series expansion of the simplified MAML loss for the first-order term, up to scale:

EQUATION

)

The fact that the simplified MAML is optimization with the negative gradient penalty is worth keeping in mind. Because the goal of gradient-based optimization is to find a point where the gradient is zero, a bias that favors a larger gradient is highly likely to make training unstable; this can be a cause of instability of MAML (Antoniou et al., 2019) . In fact, as shown in Fig. 1 , the gradient norm becomes larger during training, as do the gradient inner products, as Guiroy et al. (2019) observed. 

3. LEARNING RATE FOR LOCAL CONVERGENCE:

In this section, we will derive a sufficient condition of learning rate α and β for local convergence from any point in the vicinity of the local minima. To this end, we will assume that only one step is taken for update and training data and test data are not distinguished as we did in Section 2. Also, we do not consider SGD but steepest GD to derive the condition. In other words, same training data are assumed to be used continuously for updating parameters during training. First, we will consider the case of only single task being considered. Next, we will consider the case of multiple tasks being considered.

3.1 SINGLE TASK

3.1.1. CONDITION FOR INNER LEARNING RATE α:

First, we derive the sufficient condition of learning rate α. To this end, we will consider the sufficinet condition that a fixed point is a local minimum. Taking the Taylor series for the second-order term at a fixed point θ * , the simplified MAML loss is

EQUATION

where

H = H -α ( Tg + H 2 ) is the Hessian matrix of L(θ) at θ * and g = ∇ θ L(θ * ) ∈ R d , H = ∇ 2 θ L(θ * ) ∈ R d×d , and T = ∇ 3 θ L(θ * ) ∈ R d×d×d .

The calculation of H is presented in Appendix B. We calculated the magnitudes of Tg and H 2 numerically and observed that Tg was much smaller than H 2 in practice. Hence, we will ignore Tg while deriving the condition and will thus assume that H = H -αH 2 . Further details are provided in Appendix C, and the case of Tg being considered is provided in Appendix D. Since P Λ H P ⊤ = P [Λ H -αΛ 2 H ]P ⊤ where Λ H is a diagonal matrix with entries that are eigenvalues of H and P is a matrix with rows that are eigenvectors of H, the sufficinet condition of α for θ * to be a local minimum is

EQUATION

)

Note that λ(A) i represents the ith eigenvalue of matrix A. Hence, the sufficient condition of α for θ * to be a local minimum is

EQUATION

)

Therefore, α c is the inverse of the largest eigenvalue of H.

3.1.2. CONDITION FOR META-LEARNING RATE β:

Next, we derive the sufficient condition of meta-learning rate β for the simplified MAML to locally converge to the local minima discussed above from any point in the vicinity of the local minimum. This is an extension of research of LeCun et al. (1998) . Since P P ⊤ = I, the simplified MAML loss can be written as

EQUATION

By using the simplified loss defined in Eq. 8, the update equation of the parameter θ with GD is

EQUATION

)

where t (t = 0, ..., M ) is iteration and M is the total number of iterations. Hence, θ(t + 1) -θ * = (I -β H)(θ(t) -θ * ). If we denote P ⊤ (θ -θ * ) by v, the simplified MAML loss in Eq. 12 is

EQUATION

where v(t) is the value of v during iteration t. Assuming that Eq. 11 holds, the sufficinet condition of β is as follows: for all i,

EQUATION

)

Consequently, the sufficient condition for the simplified MAML to locally converge to local minima from any point in the vicinity of the local minima is as follows:

EQUATION

)

Vanilla GD with learning rate β corresponds to MAML if α = 0. In this case, β < 2 λmax is the condition of β, where λ max is the largest eigenvalue of H, because 2/λ max is smaller than any other 2/λ i (LeCun et al., 1998) . Though this holds for the simplified MAML as well, this is not the case if α is close to α c . The reason is that β c diverges as α approaches 1 λ(H)i , or α c as Eq. 18 indicates. Hence, for the simplified MAML we must consider not only the largest but also other eigenvalues and in particular, the second-largest eigenvalue. In short, unlike when vanilla GD is employed, β c depends on α in the case of the simplified MAML, and β c is expected to be larger if α is close to α c , as shown in Fig. 2 . This finding is validated by experiments presented in Section 5.

3.2. MULTIPLE TASKS:

We discussed the case of only one task being available during training in Section 3.1. In this section, we derive upper bounds of α and β that apply if multiple tasks τ ∼ P (τ ) are considered. Assumptions, except that multiple tasks are considered, are the same as those in Section 3.1.

3.2.1. CONDITION FOR α:

Now, we define the simplified meta-objective as a sum of task-specific objectives:

EQUATION

)

where Ĥ is the Hessian matrix of L(θ) at θ. Note that we ignore T τ g τ as we did in Section 3.1.1.

Since H τ -αH 2 τ is not necessarily simultaneously diagonalizable for each task τ , we cannot exactly express eigenvalues of Ĥ as function of λ(H τ ) and α. Therefore, instead of the exact value of α c , we will derive an upper bound of α c .

If θ * is a local minimum for all Lτ (θ * ), θ * is a local minimum for L(θ) as well. Hence, if all eigenvalues of Hτ = H τ -2αH 2 τ are positive for all tasks, those of Ĥ are positive as well. Therefore, if the inequality

∀τ, i, α < 1 λ(H τ ) i (23)

holds, it guarantees that the condition that θ * is a local minimum of L(θ) is satisfied. Note that this is a sufficient condition if θ * is a local minimum for all Lτ (θ * ). Since θ * can be a local minimum of L(θ) even when it is not a local minimum for all Lτ (θ * ), the upper bound of α seems to be larger than that in Eq. 23 in practice.

3.2.2. CONDITION FOR β:

The analysis for β is also basically the same as that performed in Section 3.1.2. Denoting

EQUATION

)

Since Hτ is not always simultaneously diagonalizable for each task, as mentioned in Section 3.2.1, P τ differs from task to task in general. Hence, we will consider an upper bound of β c as we did for α c in Section 3.2.1. Accordingly, if both Eq. 23 and

|1 -β(λ(H τ ) i -αλ(H τ ) 2 i )| < 1 (25)

hold for any eigenvalue λ i of any task τ , it guarantees that β satisfies the condition for local convergence. Therefore, a sufficient for the simplified MAML to locally converge to local minima from any point in the neighborhood of the local minima in the case of multiple tasks is as follows:

∀τ, i, α < 1 λ(H τ ) i ∧ β < 2 λ(H τ ) i -αλ(H τ ) 2 i (26)

4. RELATED WORKS:

Sevral papers have investigated MAML and proposed various algorithms (Nichol et al., 2018; Guiroy et al., 2019; Eshratifar et al., 2018; Antoniou et al., 2019; Fallah et al., 2019; Khodak et al., The area colored in black is when the loss is below 1e-2, and that in gray is when the loss is over 1e-2. Uncolored region is not considered. βc : λi shows βc of λi, where λ1 < λ2. The dashed line is αc. Theoretical βc and αc correspond to empirical ones. 2019; Vuorio et al., 2018; Finn et al., 2019; Deleu & Bengio, 2018; Liu & Theodorou, 2019; Deleu & Bengio, 2018; Grant et al., 2018) . Nichol et al. (2018) studied the first-order MAML family in detail and showed that the MAML gradient could be decomposed into two terms: a term related to joint training and a term responsible for increasing the inner product between gradients for different tasks. Guiroy et al. (2019) investigated the generalization ability of MAML. The researchers observed that generalization was correlated with the average gradient inner product and that flatness of the loss surface, often thought to be an indicator of strong generalizability in normal neural network training, was not necessarily related to generalizability in the case of MAML. Eshratifar et al. (2018) also noted that the average gradient inner product was important. Hence, the authors proposed an algorithm that considered the relative importance of each parameter based on the magnitude of the inner product between the task-specific gradient and the average gradient. Although the above studies were cognizant of the importance of the inner product of the gradients, they did not explicitly insert the negative gradient inner product, which is the negative squared gradient norm with simplifications, as a regularization term. To consider the simplified MAML as optimization with a regularization term is a contribution of our study. Antoniou et al. (2019) enumerated five factors that could cause training MAML to be difficult. Then, they authors proposed an algorithm to address all of these problems and make training MAML easier and more stable. Behl et al. (2019) , like us, pointed out that tuning the inner learning rate α and meta-learning rate β was troublesome. The authors approached this problem by proposing an algorithm that tuned learning rates automatically during training. Fallah et al. (2019) studied the convergence theory of MAML. They proposed a method for selecting meta-learning rate by approximating smoothness of the loss. Based on this result, they proved that MAML can find an ε-first-order stationary point after sufficient number of iterations. On the other hand, we studied the relationship between the sufficient conditions of inner learning rate α and meta-learning rate β and showed that how the largest possible β is affected by the value of α.

5. EXPERIMENTS:

In this section, we will present the results of experiments to confirm our expectation that MAML allows larger β if α is close to its upper bound. First, we will show the result of linear regression with simplifications used in Section 3.1. Because linear regression is convex optimization, the result is expected to exactly match the theory presented in Section 3.1. Second, to check if our expectation is confirmed in practice as well, we will present results of the practical case without any simplification. In particular, we conducted sinusoid regression and classification of Omniglot and MiniImagenet datasets with a multilayer perceptron and a convolutional neural network (CNN). Note that the meta-objective used for experiments is not a sum of task-specific objectives but a mean of them. 

5.1. LINEAR REGRESSION:

We performed a linear regression, where the task is to regress a linear function with scale parameter in the range of [0, 5.0] and bias parameter in the range of [0, 5.0] based on data points in the range of [-5.0, 5.0]. The true function has the same architecture as that of the model. We employed the steepest gradient descent method to minimize the mean squared loss, where 1 step was taken during update. Only one task was considered during training and the same data was used to update the task-specific parameter and the meta parameter as we did in Section 3.1. Using these settings, we computed the training loss after 500 iterations with α in the range of [1e-5, 9e-2] and β in the range of [5e-3, 9e+0]. The eigenvalues are those of the Hessian matrix of the training loss at the end of the training, where α = 5e-2 and β = 7e-1. We chose this training loss because it was thought to be the closest to minima. Fig. 3 shows the training losses at various values of α and β. Horizontal axis indicates α and vertical axis indicates β. The curves are β c of two eigenvalues and the dashed line shows α c . In the case of linear regression with simplifications used in Section 3.1, the result of numerical experiment shows good agreement with upper bounds of β c and α c that we derived in Section 3.1, as shown in Fig. 3 .

5.2. SINUSOID REGRESSION:

We conducted a sinusoid regression, where each task is to regress a sine wave with amplitude in the range of [0.1, 5.0] and phase in the range of [0, π] based on data points in the range of [-5.0, 5.0] . A multilayer perceptron with two hidden units of size 40 and ReLU was trained with SGD. The batch size of data was 10, the number of tasks was 100, and 1 step was taken for update. Using these settings, we computed the training loss after 500 iterations with α in the range of α is close to the value above which the losses diverge, a larger β can be used. As explained above, we did not put any simplification as we did in Section 3 and 5.1 in the experiment, meaning that we used different data for updating task-specific parameter and meta parameter, considered multiple tasks, and employed not steepest GD but SGD as optimizer. Despite simplifications, surprisingly, this result confirms the expectation that MAML allows larger β if α is close to α c .

5.3. CLASSIFICATION:

We performed classification of the Omniglot and MiniImagenet datasets (Lake et al., 2011; ravi & Larochelle, 2017) , which are benchmark datasets for few-shot learning. The model used was essentially the same as that Finn et al. (2017) , and hence, Vinyals et al. (2016) used. The task is a five-way one-shot classification, where the query size is 15, the number of update steps is two, and the task batch size is 32 for Omniglot and four for MiniImagenet. In this setup, we computed c), the maximum β is larger at large α. Even though the model architecture is composed of convolutional layer, max-pooling, and batch normalization (Ioffe & Szegedy, 2015) and practical dataset is used for training, our expectation is confirmed in this case as well. This result confirms that our theory is applicable in practice.

Our experimental result confirms that larger α is good for stabilizing MAML training. According to Fig. 4 (a), (b) and (c), moreover, while large β does not necessarily make training loss smaller, employing large α leads to smaller training loss comparatively. These result has a practical implication for tuning the learning rates: first, the largest possible α should be identified, and β may be subsequently tuned based on the value of α. Once you identify α c , MAML is likely to work well even if meta-learning rate is roughly chosen. Taking large α is desirable for the goal of MAML as well. The aim of MAML is to find a good initial parameter which quickly adapt to new tasks and the quickness is determined by inner learning rate α because α determines the step size from initial parameter to task-specific parameter when model is fine-tuned. Therefore, identifying α c is not only good for robustifying the model against divergence but also good for finding good initial parameter.

6. CONCLUSIONS:

We regard a simplified MAML as training with the negative gradient penalty. Based on this formulation, we derived the sufficient condition of the inner learning rate α and the meta-learning rate β for the simplified MAML to locally converge to local minima from any point in the vicinity of the local minima. We showed that the upper bound of β required for the simplified MAML to locally converge to local minima depends on α. Moreover, we found that if α is close to its upper bound α c , the maximum possible meta-learning rate β c is larger than that used while training with ordinary SGD. This finding is validated by experiments, confirming that our theory is applicable in practice. 

D CONVERGENCE CONDITION WHEN T g IS CONSIDERED:

We assumed that Tg was negligible in Section 3. In this section, we derive a sufficient condition of α and β for the simplified MAML to locally converge to local minima from any point in the vicinity of the local minima under some assumptions when Tg is considered.

D.1 CONDITION FOR INNER LEARNING RATE α TO SATISFY:

When Tg is considered, the Hessian matrix of the simplified MAML loss L is H = Hα ( Tg + H 2 ) . Because H -α ( Tg + H 2 ) is a real symmetric matrix, it can be diagonalized. Then, the sufficient condition that a fixed point θ * is a local minimum is ∀i, λ(

EQUATION

Note that Tg and H are not simultaneously diagonalizable, so we cannot decompose λ( H) i into eigenvalues of each matrix as we did in Section 3. Therefore, we have to consider the relationship among λ( H) i , λ(H) i and λ(Tg) i . In general, it is known that n × n Hermitian matrices A and B satisfy the following equation (Bhatia, 2001) :

EQUATION

)

where λ(A) represents a vector with elements that are eigenvalues of A, ↑ indicates the operation of sorting a vector in the ascending order, and ↓ indicates that in the descending order. If two real vectors x, y ∈ R d are related in the following way, x is said to be majorized by y and the relationship is written as x ≺ y: 

EQUATION

EQUATION

)

:



Figures and Tables:

Figure1. Gradient norm during training. We compute the norm per task and subsequently compute their average. Joint training shows when α = 0, and MAML is when α = 1e-2. These results are computed using training data, but those determined using test data behave similarly. The total number of iterations is 50000, β = 1e-3 and the Adam optimizer is used. Other settings are the same as those in Section 5.2.

Figure2. Curves of βc as a function of α for eigenvalues of the Hessian, λ0 < ... < λ4. Parameter β is supposed to be smaller than βc for both λ4 and λ3. Hence, β should be chosen from the colored area. Since α must satisfy α < 1 λ i , α should also be in the colored region. The dashed line shows βc if α = 0. If α ≈ αc, βc is larger than that at α = 0.

Figure 4. Training losses for (a) sinusoid regression, (b) Omniglot classification, and (c) MiniImagenet classification at various values of α and β after a fixed number of iterations. The area with no color represents the diverged losses, and the dashed line indicates the values of β above which the loss diverges for α = 0. The maximum possible β is larger if α is close to the value above which the losses diverge than that at α = 0.

[1e-4, 9e-1] and β in the range of [1e-2, 9e-0]. Fig. 4 (a) shows the training losses with various values of α and β. The dashed line indicates β of α = 0 over which training loss diverges. According to Fig. 4 (a), if

the training losses after 100 iterations for the Omniglot dataset and one epoch for the MiniImagenet dataset with various values of α and β; for Omniglot, α was in the range of [1e-3, 9e-0] and β was in the range of [1e-1, 9e+1], and for MiniImagenet, α was in the range of [1e-4, 9e-1] and β was in the range of [1e-2, 9e-0]. Fig. 4 (b) and (c) show the training losses of classification task at various values of α and β. The dashed line indicates β of α = 0 above which training loss diverges. As shown in Fig. 4 (b) and (

Figure 6. Training losses for (a) the sinusoid regression, (b) Omniglot classification and (c) MiniImagenet classification for various values of the inner learning rate α and meta-learning rate β after a fixed number of iterations. The area with no color represents the diverged losses.

(H 2 ) max , and λ(Tg) max is by far smaller than them. Therefore, ignoring λ(Tg) max is reasonable when the conditions are derived. Furthermore, we calculate the Frobenius norm of Tg and H 2 . As Fig.5(b) indicates, the Frobenius norm of Tg is much smaller than that of H 2 , meaning that Tg is negligible in the sense of the magnitude of the norm as well. These results confirm that we can neglect Tg when considering H. The maximum eigenvalues of Tg, H 2 and Tg + H 2 . It is clear that the maximum eigenvalue of + H 2 is almost the same as that of H 2 , while that of Tg is much smaller than them. (b): The second-largest eigenvalues of Tg, H 2 and Tg + H 2 . Like (a), the second-largest eigenvalue of Tg + H 2 is almost equal to that of H 2 . (c): The Frobenius norm of Tg and H 2 . The Frobenius norm of Tg is much smaller than that of H 2 .

Table content: <table><tr><td colspan="2">2 × 10</td><td>4</td><td/><td/><td>10</td><td>2</td><td/><td>3 × 10</td><td>4</td></tr><tr><td>10</td><td colspan="2">4</td><td/><td/><td>10</td><td>1</td><td/><td>2 × 10</td><td>4</td></tr><tr><td/><td/><td>Tg</td><td>H 2</td><td>Tg + H 2</td><td/><td>Tg</td><td>H 2</td><td>Tg + H 2</td><td>Tg</td><td>H 2</td></tr><tr><td colspan="5">(a) Maximum eigenvalue</td><td colspan="4">(b) Second largest eigenvalue</td><td>(c) Frobenius norm</td></tr><tr><td colspan="3">Figure 5. (a):</td><td/><td/><td/><td/><td/></tr></table>

Peer review (iclr2020):
In this paper, the authors tackled one important research problem in MAML, i.e., the optimization instability, by investigating the two learning rates. Though I appreciate the theoretic contribution this work makes, I am not sure with the practical significance of it. Below please find my detailed comments. Pros: - In this work, the authors focused on an important problem – training MAML is kind of unstable and tricky, so that developing guidelines that stabilize MAML or its first-order approximations is of significance. - This work theoretically discusses the relationship between the inner loop learning rate and the outer one, under a set of assumptions. - The paper is well written and easy to follow. Cons: - Some of the simplifications for proving are empirical, so that the proof itself is not that rigorous. o For example, In Section 3.1.1, the authors ignored Tg based on an observation that Tg is small. However, even in the Appendix, the authors did not explicate the experimental setting where they reach such a conclusion. Will Tg be always small then, in any task and any dataset? - The take-away of this work is not clear, in other words, it makes small contribution to the practical training of MAML. o Practically, we often train MAML in 5 or 10 steps instead of only one step. Will the conclusion apply to such setups? Or will the model itself diverge during the inner loop as more steps are taken, provided with a larger value of alpha? o Practically, we merely use vanilla SGD. The figure in the Appendix shows that the main conclusion actually does not apply to Adam, which disempower the practicability of the theoretical guideline. o How can we search the “largest possible” inner loop learning rate? Should we still use grid-search or heuristic search? In that case, what is the implication/shortcut that this proposed guideline brings? We still reach a stable training process by tuning the hyperparameters in a brute-force fashion. - The baselines should be compared, to support the effectiveness of this proposed algorithm. For example, Behl et al. (2019) automatically tuning the learning rates during training definitely needs to be compared, since both aim to stabilize the training of MAML.