Goto

Collaborating Authors

 Deep Learning


ZeRO: Memory Optimization Towards Training A Trillion Parameter Models

arXiv.org Machine Learning

Training large DL models with billions and potentially trillions of parameters is challenging. Existing solutions exhibit fundamental limitations to obtain both memory and scaling (computation/communication) efficiency together. Data parallelism does not help reduce memory footprint per device: a model with 1.5 billion parameters or more runs out of memory. Model parallelism hardly scales efficiently beyond multiple devices of a single node due to fine-grained computation and expensive communication. We develop a novel solution, Zero Redundancy Optimizer (ZeRO), to optimize memory, achieving both memory efficiency and scaling efficiency. Unlike basic data parallelism where memory states are replicated across data-parallel processes, ZeRO partitions model states instead, to scale the model size linearly with the number of devices. Furthermore, it retains scaling efficiency via computation and communication rescheduling and by reducing the model parallelism degree required to run large models. Our analysis on memory requirements and communication volume demonstrates: ZeRO has the potential to scale beyond 1 Trillion parameters using today's hardware (e.g., 1024 GPUs, 64 DGX-2 nodes). To meet near-term scaling goals and serve as a demonstration of ZeRO's capability, we implemented stage-1 optimizations of ZeRO (out of 3 stages in total described in the paper) and tested this ZeRO-OS version. ZeRO-OS reduces memory and boosts model size by 4x compared with the state-of-art, scaling up to 100B parameters. Moving forward, we will work on unlocking stage-2 optimizations, with up to 8x memory savings per device, and ultimately stage-3 optimizations, reducing memory linearly with respect to the number of devices and potentially scaling to models of arbitrary size. We are excited to transform very large models from impossible to train to feasible and efficient to train!


Advantage-Weighted Regression: Simple and Scalable Off-Policy Reinforcement Learning

arXiv.org Machine Learning

In this paper, we aim to develop a simple and scalable reinforcement learning algorithm that uses standard supervised learning methods as subroutines. Our goal is an algorithm that utilizes only simple and convergent maximum likelihood loss functions, while also being able to leverage off-policy data. Our proposed approach, which we refer to as advantage-weighted regression (AWR), consists of two standard supervised learning steps: one to regress onto target values for a value function, and another to regress onto weighted target actions for the policy. The method is simple and general, can accommodate continuous and discrete actions, and can be implemented in just a few lines of code on top of standard supervised learning methods. We provide a theoretical motivation for AWR and analyze its properties when incorporating off-policy data from experience replay. We evaluate AWR on a suite of standard OpenAI Gym benchmark tasks, and show that it achieves competitive performance compared to a number of well-established state-of-the-art RL algorithms. AWR is also able to acquire more effective policies than most off-policy algorithms when learning from purely static datasets with no additional environmental interactions. Furthermore, we demonstrate our algorithm on challenging continuous control tasks with highly complex simulated characters.


Policies Modulating Trajectory Generators

arXiv.org Artificial Intelligence

Abstract: We propose an architecture for learning complex controllable behaviors by having simple Policies Modulate Trajectory Generators (PMTG), a powerful combination that can provide both memory and prior knowledge to the controller. The result is a flexible architecture that is applicable to a class of problems with periodic motion for which one has an insight into the class of trajectories that might lead to a desired behavior. We illustrate the basics of our architecture using a synthetic control problem, then go on to learn speed-controlled locomotion for a quadrupedal robot by using Deep Reinforcement Learning and Evolutionary Strategies. We demonstrate that a simple linear policy, when paired with a parametric Trajectory Generator for quadrupedal gaits, can induce walking behaviors with controllable speed from 4 -dimensional IMU observations alone, and can be learned in under 1000 rollouts. We also transfer these policies to a real robot and show locomotion with controllable forward velocity. Keywords: Reinforcement Learning, Control, Locomotion 1 Introduction The recent success of Deep Learning (DL) on simulated robotic tasks has opened an exciting research direction. Nevertheless, many robotic tasks such as locomotion still remain an open problem for learning-based methods due to their complexity or dynamics. From a Deep Learning (DL) perspective, one way to tackle these complex problems is by using more and more complex policies (such as recurrent networks). Unfortunately, more complex policies are harder to train and require even more training data which is often problematic for robotics.


DSTL: Solution to Limitation of Small Corpus in Speech Emotion Recognition

Journal of Artificial Intelligence Research

Traditional machine learning methods share a common hypothesis: training and testing datasets must be in a common feature space with the same distribution. However, in reality, the labeled target data may be rare, so that target space does not share the same feature space or distribution as an available training set (source domain). To address the mismatch of domains, we propose a Dual-Subspace Transfer Learning (DSTL) framework that considers both the common and specific information of the two domains. In DSTL, a latent common subspace is first learned to preserve the data properties and reduce the discrepancy of domains. Then, we propose a mapping strategy to transfer the sourcespecific information to the target subspace. The integration of the domain-common and specific information constructs the proposed DSTL framework. In comparison to the stateart-of works, the main contribution of our work is that the DSTL framework not only considers the commonalities, but also exploits the specific information. Experiments on three emotional speech corpora verify the effectiveness of our approach. The results show that the methods which include both domain-common and specific information perform better than the baseline methods which only exploit the domain commonalities.


How Do You Know You Have Enough Training Data?

#artificialintelligence

There is some debate recently as to whether data is the new oil [1] or not [2]. Whatever the case, acquiring training data for our machine learning work can be expensive (in man-hours, licensing fees, equipment run time, etc.). Thus, a crucial issue in machine learning projects is to determine how much training data is needed to achieve a specific performance goal (i.e., classifier accuracy). In this post, we will do a quick but broad in scope review of empirical and research literature results, regarding training data size, in areas ranging from regression analysis to deep learning. The training data size issue is also known in the literature as sample complexity.


Overcoming Deep Learning Stumbling Blocks - KDnuggets

#artificialintelligence

Recently at the 6th annual Deep Learning Summit in London, attendees congregated to hear from industry leaders, academics, researchers, and innovative startups, presenting both the latest cross-industry technological advancements and industry application methods. Running in parallel was the AI Assistant Summit and AI in Retail and Advertising Summit, resulting in some of the world's leading experts from Universities, brands, and emerging startups coming together. We began the day with Huma Lodhi, Data Scientist at BP, discussing some of the tricks and tips she has picked up during her work in Deep Learning, with intelligent methodologies using structured and unstructured data as the focal point. "We need to find better methods to use this data for our real-world applications. Examples of this can be Noisy Data, Missing data, or unstructured data. This gives us the principle problem for data, Quantity vs Quality."


Monitoring Sun's ultraviolet emission using deep learning

#artificialintelligence

It is possible to monitor Sun's extreme ultraviolet (EUV) irradiance using deep learning. Scientists at the NASA Frontier Development Lab (FDL) have shown that deep learning can help get more value out of our current ability to monitor the Sun by providing virtual instruments to supplement physical devices. The Sun is vital for survival, yet solar flares, which typically occurs a few times a year, can cause severe interruptions in space and on Earth. These interruptions can affect rocket, satellites, and even frameworks here on Earth, including GPS route, radio interchanges, and the power grid. FDL team member and co-author Alexander Szenicer (Oxford University) said, "Our research shows how a deep neural network can be trained to mimic an instrument on the Solar Dynamics Observatory (SDO). By inferring what ultraviolet radiation levels that sensor would have detected based on what the other instruments on SDO are observing at any given time, we demonstrate it is possible to increase the scientific productivity of NASA missions and to increase our capability to monitor solar sources of space weather."


What Are Editable Neural Networks & Can They Simplify Deep Learning?

#artificialintelligence

Deep Learning is a computational heavy process. Cutting down the costs is one major challenge along with data curation. Their power hungry training processes had garnered such reputation that researchers have published works reporting the carbon footprint of training networks. As things can only get complicated from here on, heading into a future with a deluge of machine learning applications, we see new strategies being invented to make training neural networks as efficient as problem. Updating a neural network to change its predictions on a single input can decrease performance across other inputs. While being simple, this approach is not robust to minor changes in the input.


What Are The Alternatives To Batch Normalization In Deep Learning?

#artificialintelligence

In the original BatchNorm paper, the authors Sergey Ioffe and Christian Szegedy of Google introduced a method to address a phenomenon called internal covariate shift. This occurs because the distribution of each layer's inputs changes during training, as the parameters of the previous layers change. This slows down the training by requiring lower learning rates and careful parameter initialisation. This makes training the models harder. The introduction of batch normalized networks helped achieve state-of-the-art accuracies with 14 times fewer training steps.


What is TensorFlow? The machine learning library explained

#artificialintelligence

Machine learning is a complex discipline. But implementing machine learning models is far less daunting and difficult than it used to be, thanks to machine learning frameworks--such as Google's TensorFlow--that ease the process of acquiring data, training models, serving predictions, and refining future results. Created by the Google Brain team, TensorFlow is an open source library for numerical computation and large-scale machine learning. TensorFlow bundles together a slew of machine learning and deep learning (aka neural networking) models and algorithms and makes them useful by way of a common metaphor. It uses Python to provide a convenient front-end API for building applications with the framework, while executing those applications in high-performance C .