What Is Meta-Learning via Learned Losses (with Python Code)
Facebook AI Research (FAIR) research on meta-learning has majorly classified into two types:  First, methods that can learn representation for generalization. Second, methods that can optimize models. We have thoroughly discussed the type first in our previous article MBIRL. For this post, we are going to give a brief introduction to the second type. Last month, at the International Conference on Pattern Recognition, {ICPR}, Italy, January 10-15, 2021, a group of researchers: S. Bechtle,  A. Molchanov, Y. Chebotar, E. Grefenstette, L. Righetti, G. S. Sukhatme, F. Meier submitted a research paper focussing on the automation of “meta-training” processing: Meta Learning via Learned Loss. Motivation Behind ML3 In meta-learning, the goal is to efficiently optimize the function fθ  which can be a regressor or classifier that finds the optimal value of θ. L is the loss function and h is the gradient transform. The majority of the work in deep learning is associated with learning the f function directly from data and some meta-learning work focuses on the parameter updation. In ML3 approach, the authors have targeted loss learning. Loss functions are architecture independent and widely used for learning problems so learning a loss function doesn’t require any engineering and optimization and allows the addition of extra information during meta-training. About ML3 The key idea of the proposed framework is to develop a pipeline for meta-training that not only can optimize the performance of the model but also generalize for different tasks and model architectures. The proposed framework of learning loss functions efficiently optimize the models for new tasks. The main contribution of the ML3 framework are : i) It is capable of learning adaptive, high-dimensional functions via back propagation and gradient descent. ii) The given framework is very flexible as it is capable of storing additional information at the meta-train time and provides generalization by solving regression, classification, model-based reinforcement learning, model-free reinforcement learning. The Model Architecture of ML3 The task of learning a loss function is based on a bi-level optimization technique i.e., it contains two optimization loops: inner and outer. The inner loop is responsible for training the model or optimizee with gradient descent by using the loss function learners meta-loss function and the outer loop optimized the meta-loss function by minimizing the task loss i.e., regression or classification or reinforcement learning loss. The process contains a function f parameterized by θ that takes a variable x and outputs y. It also learns meta-loss network M parameterized by Φ  that takes the input and output of function f and together with task-specific information g (for example ground truth label for regression or classification, final position in MBIRL or the sample reward from model-free reinforcement learning problems) and outputs the meta- loss function L parameterized by both Φ and θ. So, to update function f, compute the gradient of Meta Loss L with respect to θ and update the gradient using the learned loss function, as shown below : Updation of gradient Now, to update M, the loss network, formulate a task-specific loss that compares the output of the currently optimal f with the target information since f is updated with L, the task is also functional Φ and perform gradient update on Φ to optimize M. This architecture finally forms a fully differential loss learning framework used for training. Updation of loss network M To use the learning loss  at Test time, directly update f by taking the gradient of learned loss L with respect to the parameters of f. Applications of ML3 Regression problems.Classification problems.Shaping Loss during training e.g., Covexifying Loss, exploration signal. ML3 provides a possibility to add additional information during meta-training.Model-based Reinforcement Learning.Model-free Reinforcement Learning. Requirements & Installation Python=3.7Clone the Github repository via git. !git clone https://github.com/facebookresearch/LearningToLearn.git %cd LearningToLearn/ Install all the dependencies of ML3 via : !python setup.py develop Paper Experiment Demos This section contains different experiments mentioned in the research paper. A. Loss Learning for Regression Run Sin function regression experiment by code below: !python ml3/experiments/run_sine_regression_exp.py Now, you can visualize the results by the following code: 2.1 Import the required libraries, packages and modules and specify the path to the saved data during meta-training. The code snippet is available here. 2.2 Load the saved data during the experiment. # load results train_reg = np.zeros((50, len(seeds))) train_ml3 = np.zeros_like(train_reg) test_reg_loss = np.zeros((100, len(seeds))) test_ml3_loss = np.zeros_like(test_reg_loss) test_loss_trace = np.zeros_like(test_reg_loss) n_steps = 1 for j, seed in enumerate(seeds): exp_file = "sine_regression_seed_{}.pt".format(seed) data = torch.load(os.path.join(EXP_FOLDER, exp_file)) train_reg[:, j] = np.asarray([np.vstack(data[i]['train_reg']['loss_trace']).mean(axis=0)[n_steps] for i in range(len(data))]) train_ml3[:, j] =  np.asarray([np.vstack(data[i]['train_ml3']['loss_trace']).mean(axis=0)[n_steps] for i in range(len(data))]) test_reg_loss[:, j] = np.vstack(data[-1]['test_reg']['loss_trace']).mean(axis=0) test_ml3_loss[:, j] = np.vstack(data[-1]['test_ml3']['loss_trace']).mean(axis=0) 2.3 Visualize the performance of the meta loss when used to optimize the meta training tasks, as a function of (outer) meta training iterations. # mean and variance over seeds mu1 = train_reg.mean(axis=1) sigma1 = train_reg.std(axis=1) mu2 = train_ml3.mean(axis=1) sigma2 = train_ml3.std(axis=1) t = np.arange(0, 500, 10) # plot it! fig, ax = plt.subplots(1) ax.plot(t, mu1, lw=2, label='SGD with MSE Loss', color='C1') ax.plot(t, mu2, lw=2, label='SGD with ML3 Loss', color='C0') ax.fill_between(t, mu1+sigma1, mu1-sigma1, facecolor='C1', alpha=0.5) ax.fill_between(t, mu2+sigma2, mu2-sigma2, facecolor='C0', alpha=0.5) ax.set_title(r'MSE on train tasks after 100 gradient steps') #ax.legend(loc='upper left') ax.set_xlabel('meta train iter') ax.set_ylabel('average MSE on meta train tasks') ax.grid() ax.legend(loc='lower left') plt.savefig("{}/sine_meta_training.png".format(EXP_FOLDER)) The output will be : 2.4 Evaluating learned meta loss networks on test tasks. Plot the performance of the final meta loss network when used to optimize the new test tasks at meta test time. Here the x-axis represents the number of gradient descent steps. The code snippet is available here. B. Reward Learning for Model-based RL (MBRL) Reacher For meta learning the reward, run !python ml3/experiments/run_mbrl_reacher_exp.py train For testing the reward, run !python ml3/experiments/run_mbrl_reacher_exp.py test C. Learning with extra information at the meta-train time This demo shows how we can add extra information during meta training in order to shape the loss function. For experiment purposes, we have taken the example of sin function. Now, with the code, the script requires two arguments, first one is train\test, the 2nd one indicates whether to use extra information by setting True\False (with\without extra info). For training, the code is given below !python ml3/experiments/run_shaped_sine_exp.py train True To test the loss with extra information run: !python ml3/experiments/run_shaped_sine_exp.py test True For comparison purposes, we have repeated the above two steps with argument as False. The full code is available here.Comparison of results via visualization. Similarly, the research experiment for meta learning the loss with an additional goal in the mountain car experiment run can be done. The code lines is available here. EndNotes In this write-up we have given an overview Meta Learning via Learned Loss(ML3), a gradient-based bi-level optimization algorithm which is capable of learning any parametric loss function as long as the output is differential with respect to its parameters. These learned loss functions can be used to efficiently optimize models for new tasks. Note : All the figures/images except the output of the code are taken from official sources of ML3. Colab Notebook ML3 Demo Official Code, Documentation & Tutorial are available at: Github Website Research Paper
Facebook AI Research (FAIR) research on meta-learning has majorly classified into two types:  First, methods that can learn representation for generalization. Second, methods that can optimize models. We have thoroughly discussed the type first in our previous article MBIRL. For this post, we are going to give a brief introduction to the second type. Last […]
The key idea of the ML3 is to develop a pipeline for meta-training that not only can optimize the performance of the model but also generalize for different tasks and model architectures.
["Deep Learning", "optimization", "Reinforcement Learning", "reinforcement learning an introduction"]
["Go", "TPU", "Deep Learning", "Reinforcement Learning", "AI", "ML", "RAG", "Colab", "Ray", "Python", "deep learning", "optimization", "R", "reinforcement learning an introduction"]
["AI", "ML", "deep learning", "Ray", "Colab", "RAG", "TPU", "Python", "R", "Go"]
A data science enthusiast and a post-graduate in Big Data Analytics. Creative and organized with an analytical bent of mind.