Joint Stochastic Approximation and Its Application to Learning Discrete Latent Variable Models

Total Page:16

File Type:pdf, Size:1020Kb

Joint Stochastic Approximation and Its Application to Learning Discrete Latent Variable Models Joint Stochastic Approximation and Its Application to Learning Discrete Latent Variable Models Zhijian Ou Yunfu Song Electronic Engineering Department Electronic Engineering Department Tsinghua University, Beijing, China Tsinghua University, Beijing, China Abstract Currently variational methods are widely used for learning latent variable models, especially those parameterized us- Although with progress in introducing auxil- ing neural networks. In such methods, an auxiliary amor- iary amortized inference models, learning dis- tized inference model q.hðx/ with parameter is intro- crete latent variable models is still challenging. duced to approximate the posterior p.hðx/ (Kingma and In this paper, we show that the annoying diffi- Welling, 2014; Rezende et al., 2014), and some bound of culty of obtaining reliable stochastic gradients the marginal log-likelihood, used as a surrogate objective, for the inference model and the drawback of in- is optimized over both and . Two well-known bounds directly optimizing the target log-likelihood can are the evidence lower bound (ELBO) (Jordan et al., 1999) be gracefully addressed in a new method based and the multi-sample importance-weighted (IW) lower on stochastic approximation (SA) theory of the bound (Burda et al., 2015). Though with progress (as Robbins-Monro type. Specifically, we propose reviewed in Section 4), a difficulty in variational learn- to directly maximize the target log-likelihood ing of discrete latent variable models is to obtain reliable and simultaneously minimize the inclusive di- (unbiased, low-variance) Monte-Carlo estimates of the 1 vergence between the posterior and the infer- gradient of the bound with respect to (w.r.t.) . ence model. The resulting learning algorithm Additionally, a common drawback in many existing meth- is called joint SA (JSA). To the best of our ods for learning latent-variable models is that they indi- knowledge, JSA represents the first method that rectly optimize some bound of the target marginal log- couples an SA version of the EM (expectation- likelihood. This leaves an uncontrolled gap between the maximization) algorithm (SAEM) with an adap- marginal log-likelihood and the bound, depending on the tive MCMC procedure. Experiments on several expressiveness of the inference model. There are hard benchmark generative modeling and structured efforts to develop more expressive but increasingly com- prediction tasks show that JSA consistently out- plex inference models to reduce the gap (Salimans et al., performs recent competitive algorithms, with 2014; Rezende and Mohamed, 2015; Maaløe et al., 2016; faster convergence, better final likelihoods, and Kingma et al., 2017)2. But it is highly desirable that we lower variance of gradient estimates. can eliminate the effect of the gap on model learning or ide- ally directly optimize the marginal log-likelihood, without increasing the complexity of the inference model. 1 INTRODUCTION In this paper, we show that the annoying difficulty of obtaining reliable stochastic gradients for the inference A wide range of machine learning tasks involves observed model and the drawback of indirectly optimizing the tar- and unobserved data. Latent variable models explain ob- get log-likelihood can be gracefully addressed in a new servations as part of a partially observed system and usu- method based on stochastic approximation (SA) theory ally express a joint distribution p.x; h/ over observation x and its unobserved counterpart h, with parameter . 1since the gradient of the ELBO w.r.t. usually can be Models with discrete latent variables are broadly applied, reliably estimated. including mixture modeling, unsupervised learning (Neal, 2Notably, some methods are mainly applied to continuous 1992; Mnih and Gregor, 2014), structured output predic- latent variables, e.g. it is challenging to directly apply normaliz- tion (Raiko et al., 2014; Mnih and Rezende, 2016) and so ing flows to discrete random variables, though recently there are on. some effort (Ziegler and Rush, 2019). Proceedings of the 36th Conference on Uncertainty in Artificial Intelligence (UAI), PMLR volume 124, 2020. of the Robbins-Monro type (Robbins and Monro, 1951). Algorithm 1 The general stochastic approximation (SA) These two seemingly unrelated issues are inherently re- algorithm lated to our choice that we optimizes the ELBO or the for t = 1; 2; 5 do similar IW lower bound. Monte Carlo sampling: Draw a sample z.t/ with a .t*1/ Specifically, we propose to directly maximize w.r.t. the Markov transition kernel K.t*1/ .z ; ⋅/, which 3 .t*1/ marginal log-likelihood and simultaneously minimize starts with z and admits p.t*1/ .⋅/ as the invariant 4 w.r.t. the inclusive divergence KL[p.h x/ q.h x/] distribution. ð ðð ð .t/ .t*1/ .t/ between the posterior and the inference model, and for- SA updating: Set = + tF.t*1/ .z /, where tunately, we can use the SA framework to solve the joint t is the learning rate. optimization problem. The key is to recast the two gra- end for dients as expectations and equate them to zero; then the equations can be solved by applying the SA algorithm, in which the inference model serves as an adaptive proposal of iterative stochastic optimization algorithms, introduced for constructing the Markov Chain Monte Carlo (MCMC) in (Robbins and Monro, 1951) and extensively studied sampler. The resulting learning algorithm is called joint (Benveniste et al., 1990; Chen, 2002). Basically, stochastic SA (JSA), as it couples SA-based model learning and SA- approximation provides a mathematical framework for based adaptive MCMC and jointly finds the two sets of stochastically solving a root finding problem, which has unknown parameters ( and ). the form of expectations being equal to zeros. Suppose that the objective is to find the solution < of f./ = 0 It is worthwhile to recall that there is an another class of with methods in learning latent variable models for maximum f./ = E [F .z/]; (1) likelihood (ML), even prior to the recent development of z∼p.⋅/ variational learning for approximate ML, which consists of where is a d-dimensional parameter vector, and z is the expectation-maximization (EM) algorithm (Dempster an observation from a probability distribution p.⋅/ de- et al., 1977) and its extensions. Interestingly, we show that d pending on . F.z/Ë R is a function of z, providing the JSA method amounts to coupling an SA version of EM d-dimensional stochastic measurements of the so-called (SAEM) (Delyon et al., 1999; Kuhn and Lavielle, 2004) mean-field function f./. Intuitively, we solve a system with an adaptive MCMC procedure. This represents a new of simultaneous equations, f./ = 0, which consists of d extension among the various stochastic versions of EM in constraints, for determining d-dimensional . the literature. This revealing of the connection between JSA and SAEM is important for us to appreciate the new Given some initialization .0/ and z.0/, a general SA al- JSA method. gorithm iterates Monte Carlo sampling and parameter up- dating, as shown in Algorithm 1. The convergence of SA The JSA learning algorithm can handle both continuous has been established under conditions (Benveniste et al., and discrete latent variables. The application of JSA in 1990; Andrieu et al., 2005; Song et al., 2014), including the continuous case is not pursued in the present work, a few technical requirements for the mean-field function and we leave it as an avenue of future exploration. In this .t*1/ f./, the Markov transition kernel K.t*1/ .z ; ⋅/ and paper, we mainly present experimental results for learning the learning rates. Particularly, when f./ corresponds discrete latent variable models with Bernoulli and cate- to the gradient of some objective function, then .t/ will gorical variables, consisting of stochastic layers or neural converge to local optimum, driven by stochastic gradients network layers. Our results on several benchmark genera- F.z/. To speed up convergence, during each SA iteration, tive modeling and structured prediction tasks demonstrate it is possible to generate a set of multiple observations z that JSA consistently outperforms recent competitive al- by performing the Markov transition repeatedly and then gorithms, with faster convergence, better final likelihoods, use the average of the corresponding values of F.z/ for and lower variance of gradient estimates. updating , which is known as SA with multiple moves (Wang et al., 2018), as shown in Algorithm 3 in Appendix. 2 PRELIMINARIES Remarkably, Algorithm 1 shows stochastic approximation with Markovian perturbations (Benveniste et al., 1990). It 2.1 STOCHASTIC APPROXIMATION (SA) is more general than the non-Markovian SA which requires .t/ exact sampling z ∼ p .t*1/ .⋅/ at each iteration and in Stochastic approximation methods are an important family some tasks can hardly be realized. In non-Markovian SA, 3“directly” is in the sense that we set the marginal log- we check that F.z/ is unbiased estimates of f./, while in likelihood as the objective function in stochastic optimization. SA with Markovian perturbations, we check the ergodicity 4 KL[pððq] ≜ ∫ p log p_q property of the Markov transition kernel. 2.2 VARIATIONAL LEARNING METHODS two objectives in Eq. (2) can be derived as follows6: Here we briefly review the variational methods, recently hg ≜ *(KL[ ̃p.x/ððp.x/] developed for learning latent variable models (Kingma n = E ( logp .x; h/ n ̃p.x/p.hðx/ and Welling, 2014; Rezende et al., 2014). Consider a la- (3) l g *( E KL p .h x/ q .h x/ tent variable model p.x; h/ for observation x and latent n ≜ ̃p.x/ ð ðð ð variable h, with parameter . Instead of directly maximiz- n = E ( logq .h x/ j ̃p.x/p.hðx/ ð ing the marginal log-likelihood log p.x/ for the above latent variable model, variational methods maximize the By the following Proposition 1, Eq.(3) can be recast in variational lower bound (also known as ELBO), after intro- the expectation form of Eq.(1).
Recommended publications
  • Trajectory Averaging for Stochastic Approximation MCMC Algorithms
    The Annals of Statistics 2010, Vol. 38, No. 5, 2823–2856 DOI: 10.1214/10-AOS807 c Institute of Mathematical Statistics, 2010 TRAJECTORY AVERAGING FOR STOCHASTIC APPROXIMATION MCMC ALGORITHMS1 By Faming Liang Texas A&M University The subject of stochastic approximation was founded by Rob- bins and Monro [Ann. Math. Statist. 22 (1951) 400–407]. After five decades of continual development, it has developed into an important area in systems control and optimization, and it has also served as a prototype for the development of adaptive algorithms for on-line esti- mation and control of stochastic systems. Recently, it has been used in statistics with Markov chain Monte Carlo for solving maximum likelihood estimation problems and for general simulation and opti- mizations. In this paper, we first show that the trajectory averaging estimator is asymptotically efficient for the stochastic approxima- tion MCMC (SAMCMC) algorithm under mild conditions, and then apply this result to the stochastic approximation Monte Carlo algo- rithm [Liang, Liu and Carroll J. Amer. Statist. Assoc. 102 (2007) 305–320]. The application of the trajectory averaging estimator to other stochastic approximation MCMC algorithms, for example, a stochastic approximation MLE algorithm for missing data problems, is also considered in the paper. 1. Introduction. Robbins and Monro (1951) introduced the stochastic approximation algorithm to solve the integration equation (1) h(θ)= H(θ,x)fθ(x) dx = 0, ZX where θ Θ Rdθ is a parameter vector and f (x), x Rdx , is a density ∈ ⊂ θ ∈X ⊂ function depending on θ. The dθ and dx denote the dimensions of θ and x, arXiv:1011.2587v1 [math.ST] 11 Nov 2010 respectively.
    [Show full text]
  • Stochastic Approximation for the Multivariate and the Functional Median 3 It Can Not Be Updated Simply If the Data Arrive Online
    STOCHASTIC APPROXIMATION TO THE MULTIVARIATE AND THE FUNCTIONAL MEDIAN Submitted to COMPSTAT 2010 Herv´eCardot1, Peggy C´enac1, and Mohamed Chaouch2 1 Institut de Math´ematiques de Bourgogne, UMR 5584 CNRS, Universit´ede Bourgogne, 9, Av. A. Savary - B.P. 47 870, 21078 Dijon, France [email protected], [email protected] 2 EDF - Recherche et D´eveloppement, ICAME-SOAD 1 Av. G´en´eralde Gaulle, 92141 Clamart, France [email protected] Abstract. We propose a very simple algorithm in order to estimate the geometric median, also called spatial median, of multivariate (Small, 1990) or functional data (Gervini, 2008) when the sample size is large. A simple and fast iterative approach based on the Robbins-Monro algorithm (Duflo, 1997) as well as its averaged version (Polyak and Juditsky, 1992) are shown to be effective for large samples of high dimension data. They are very fast and only require O(Nd) elementary operations, where N is the sample size and d is the dimension of data. The averaged approach is shown to be more effective and less sensitive to the tuning parameter. The ability of this new estimator to estimate accurately and rapidly (about thirty times faster than the classical estimator) the geometric median is illustrated on a large sample of 18902 electricity consumption curves measured every half an hour during one week. Keywords: Geometric quantiles, High dimension data, Online estimation al- gorithm, Robustness, Robbins-Monro, Spatial median, Stochastic gradient averaging 1 Introduction Estimation of the median of univariate and multivariate data has given rise to many publications in robust statistics, data mining, signal processing and information theory.
    [Show full text]
  • Simulation Methods for Robust Risk Assessment and the Distorted Mix
    Simulation Methods for Robust Risk Assessment and the Distorted Mix Approach Sojung Kim Stefan Weber Leibniz Universität Hannover September 9, 2020∗ Abstract Uncertainty requires suitable techniques for risk assessment. Combining stochastic ap- proximation and stochastic average approximation, we propose an efficient algorithm to compute the worst case average value at risk in the face of tail uncertainty. Dependence is modelled by the distorted mix method that flexibly assigns different copulas to different regions of multivariate distributions. We illustrate the application of our approach in the context of financial markets and cyber risk. Keywords: Uncertainty; average value at risk; distorted mix method; stochastic approximation; stochastic average approximation; financial markets; cyber risk. 1 Introduction Capital requirements are an instrument to limit the downside risk of financial companies. They constitute an important part of banking and insurance regulation, for example, in the context of Basel III, Solvency II, and the Swiss Solvency test. Their purpose is to provide a buffer to protect policy holders, customers, and creditors. Within complex financial networks, capital requirements also mitigate systemic risks. The quantitative assessment of the downside risk of financial portfolios is a fundamental, but arduous task. The difficulty of estimating downside risk stems from the fact that extreme events are rare; in addition, portfolio downside risk is largely governed by the tail dependence of positions which can hardly be estimated from data and is typically unknown. Tail dependence is a major source of model uncertainty when assessing the downside risk. arXiv:2009.03653v1 [q-fin.RM] 8 Sep 2020 In practice, when extracting information from data, various statistical tools are applied for fitting both the marginals and the copulas – either (semi-)parametrically or empirically.
    [Show full text]
  • An Empirical Bayes Procedure for the Selection of Gaussian Graphical Models
    An empirical Bayes procedure for the selection of Gaussian graphical models Sophie Donnet CEREMADE Universite´ Paris Dauphine, France Jean-Michel Marin∗ Institut de Mathematiques´ et Modelisation´ de Montpellier Universite´ Montpellier 2, France Abstract A new methodology for model determination in decomposable graphical Gaussian models (Dawid and Lauritzen, 1993) is developed. The Bayesian paradigm is used and, for each given graph, a hyper inverse Wishart prior distribution on the covariance matrix is considered. This prior distribution depends on hyper-parameters. It is well-known that the models’s posterior distribution is sensitive to the specification of these hyper-parameters and no completely satisfactory method is registered. In order to avoid this problem, we suggest adopting an empirical Bayes strategy, that is a strategy for which the values of the hyper-parameters are determined using the data. Typically, the hyper-parameters are fixed to their maximum likelihood estimations. In order to calculate these maximum likelihood estimations, we suggest a Markov chain Monte Carlo version of the Stochastic Approximation EM algorithm. Moreover, we introduce a new sampling scheme in the space of graphs that improves the add and delete proposal of Armstrong et al. (2009). We illustrate the efficiency of this new scheme on simulated and real datasets. Keywords: Gaussian graphical models, decomposable models, empirical Bayes, Stochas- arXiv:1003.5851v2 [stat.CO] 23 May 2011 tic Approximation EM, Markov Chain Monte Carlo ∗Corresponding author: [email protected] 1 1 Gaussian graphical models in a Bayesian Context Statistical applications in genetics, sociology, biology , etc often lead to complicated interaction patterns between variables.
    [Show full text]
  • Stochastic Approximation of Score Functions for Gaussian Processes
    The Annals of Applied Statistics 2013, Vol. 7, No. 2, 1162–1191 DOI: 10.1214/13-AOAS627 c Institute of Mathematical Statistics, 2013 STOCHASTIC APPROXIMATION OF SCORE FUNCTIONS FOR GAUSSIAN PROCESSES1 By Michael L. Stein2, Jie Chen3 and Mihai Anitescu3 University of Chicago, Argonne National Laboratory and Argonne National Laboratory We discuss the statistical properties of a recently introduced un- biased stochastic approximation to the score equations for maximum likelihood calculation for Gaussian processes. Under certain condi- tions, including bounded condition number of the covariance matrix, the approach achieves O(n) storage and nearly O(n) computational effort per optimization step, where n is the number of data sites. Here, we prove that if the condition number of the covariance matrix is bounded, then the approximate score equations are nearly optimal in a well-defined sense. Therefore, not only is the approximation ef- ficient to compute, but it also has comparable statistical properties to the exact maximum likelihood estimates. We discuss a modifi- cation of the stochastic approximation in which design elements of the stochastic terms mimic patterns from a 2n factorial design. We prove these designs are always at least as good as the unstructured design, and we demonstrate through simulation that they can pro- duce a substantial improvement over random designs. Our findings are validated by numerical experiments on simulated data sets of up to 1 million observations. We apply the approach to fit a space–time model to over 80,000 observations of total column ozone contained in the latitude band 40◦–50◦N during April 2012.
    [Show full text]
  • Stochastic Approximation Methods and Applications in Financial Optimization Problems
    Stochastic Approximation Methods And Applications in Financial Optimization Problems by Chao Zhuang (Under the direction of Qing Zhang) Abstract Optimizations play an increasingly indispensable role in financial decisions and financial models. Many problems in mathematical finance, such as asset allocation, trading strategy, and derivative pricing, are now routinely and efficiently approached using optimization. Not until recently have stochastic approximation methods been applied to solve optimization problems in finance. This dissertation is concerned with stochastic approximation algorithms and their appli- cations in financial optimization problems. The first part of this dissertation concerns trading a mean-reverting asset. The strategy is to determine a low price to buy and a high price to sell so that the expected return is maximized. Slippage cost is imposed on each transaction. Our effort is devoted to developing a recursive stochastic approximation type algorithm to esti- mate the desired selling and buying prices. In the second part of this dissertation we consider the trailing stop strategy. Trailing stops are often used in stock trading to limit the maximum of a possible loss and to lock in a profit. We develop stochastic approximation algorithms to estimate the optimal trailing stop percentage. A modification using projection is devel- oped to ensure that the approximation sequence constructed stays in a reasonable range. In both parts, we also study the convergence and the rate of convergence. Simulations and real market data are used to demonstrate the performance of the proposed algorithms. The advantage of using stochastic approximation in stock trading is that the underlying asset is model free. Only observed stock prices are required, so it can be performed on line to provide guidelines for stock trading.
    [Show full text]
  • Stochastic Approximation Algorithm
    Learning in Complex Systems Spring 2011 Lecture Notes Nahum Shimkin 5 The Stochastic Approximation Algorithm 5.1 Stochastic Processes { Some Basic Concepts 5.1.1 Random Variables and Random Sequences Let (­; F;P ) be a probability space, namely: { ­ is the sample space. { F is the event space. Its elements are subsets of ­, and it is required to be a σ- algebra (includes ; and ­; includes all countable union of its members; includes all complements of its members). { P is the probability measure (assigns a probability in [0,1] to each element of F, with the usual properties: P (­) = 1, countably additive). A random variable (RV) X on (­; F) is a function X : ­ ! R, with values X(!). It is required to be measurable on F, namely, all sets of the form f! : X(!) · ag are events in F. A vector-valued RV is a vector of RVs. Equivalently, it is a function X : ­ ! Rd, with similar measurability requirement. d A random sequence, or a discrete-time stochastic process, is a sequence (Xn)n¸0 of R -valued RVs, which are all de¯ned on the same probability space. 1 5.1.2 Convergence of Random Variables A random sequence may converge to a random variable, say to X. There are several useful notions of convergence: 1. Almost sure convergence (or: convergence with probability 1): a:s: Xn ! X if P f lim Xn = Xg = 1 : n!1 2. Convergence in probability: p Xn ! X if lim P (jXn ¡ Xj > ²) = 0 ; 8² > 0 : n!1 3. Mean-squares convergence (convergence in L2): 2 L 2 Xn ! X if EjXn ¡ X1j ! 0 : 4.
    [Show full text]
  • Stochastic Approximation: from Statistical Origin to Big-Data, Multidisciplinary Applications
    STOCHASTIC APPROXIMATION: FROM STATISTICAL ORIGIN TO BIG-DATA, MULTIDISCIPLINARY APPLICATIONS By Tze Leung Lai Hongsong Yuan Technical Report No. 2020-05 June 2020 Department of Statistics STANFORD UNIVERSITY Stanford, California 94305-4065 STOCHASTIC APPROXIMATION: FROM STATISTICAL ORIGIN TO BIG-DATA, MULTIDISCIPLINARY APPLICATIONS By Tze Leung Lai Stanford University Hongsong Yuan Shanghai University of Finance and Economics Technical Report No. 2020-05 June 2020 This research was supported in part by National Science Foundation grant DMS 1811818. Department of Statistics STANFORD UNIVERSITY Stanford, California 94305-4065 http://statistics.stanford.edu Submitted to Statistical Science Stochastic Approximation: from Statistical Origin to Big-Data, Multidisciplinary Applications Tze Leung Lai and Hongsong Yuan Abstract. Stochastic approximation was introduced in 1951 to provide a new theoretical framework for root finding and optimization of a regression function in the then-nascent field of statistics. This review shows how it has evolved in response to other developments in statistics, notably time series and sequential analysis, and to applications in artificial intelligence, economics, and engineering. Its resurgence in the Big Data Era has led to new advances in both theory and applications of this microcosm of statistics and data science. Key words and phrases: Control, gradient boosting, optimization, re- cursive stochastic algorithms, regret, weak greedy variable selection. 1. INTRODUCTION The year 2021 will mark the seventieth anniversary of the seminal paper of Robbins and Monro (1951) on stochastic approximation. In 1946, Herbert Robbins entered the then-nascent field of statistics somewhat serendipitously. He had enlisted in the Navy during the Second World War and was demobilized as a lieutenant commander in 1945.
    [Show full text]
  • On Implementation of the Markov Chain Monte Carlo Stochastic Approximation Algorithm
    On Implementation of the Markov Chain Monte Carlo Stochastic Approximation Algorithm Yihua Jiang, Peter Karcher and Yuedong Wang Abstract The Markov Chain Monte Carlo Stochastic Approximation Algorithm (MCMCSAA) was developed to compute estimates of parameters with incomplete data. In theory this algorithm guarantees convergence to the expected fixed points. However, due to its flexibility and complexity, care needs to be taken for imple- mentation in practice. In this paper we show that the performance of MCMCSAA depends on many factors such as the Markov chain Monte Carlo sample size, the step-size of the parameter update, the initial values and the choice of an approxima- tion to the Hessian matrix. Good choices of these factors are crucial to the practi- cal performance and our results provide practical guidelines for these choices. We propose a new adaptive and hybrid procedure which is stable and faster while main- taining the same theoretical properties. 1 Introduction We often face incomplete data such as censored and/or truncated data in survival analysis, and unobservable latent variables in mixed effects and errors-in-variables models. It is difficult to compute estimates of parameters such as maximum like- lihood estimates (MLE) with incomplete data because the likelihood usually in- volves intractable integrals. For example, the likelihood function of the generalized linear mixed models (GLMM) involves an integral with respect to the random ef- Yihua Jiang Capital One Financial Corp., 15000 Capital One Dr, Richmond, VA 23238 e-mail: Yi- [email protected] Peter Karcher e-mail: [email protected] Yuedong Wang Department of Statistics & Applied Probability, University of California, Santa Barbara, CA 93106 e-mail: [email protected] 1 2 Jiang, Karcher, Wang fects which usually does not have a closed form for non-Gaussian observations.
    [Show full text]
  • 1 Multidimensional Stochastic Approximation: Adaptive Algorithms and Applications
    1 Multidimensional Stochastic Approximation: Adaptive Algorithms and Applications Mark Broadie, Columbia University Deniz M. Cicek, Columbia University Assaf Zeevi, Columbia University We consider prototypical sequential stochastic optimization methods of Robbins-Monro (RM), Kiefer- Wolfowitz (KW) and simultaneous perturbations stochastic approximation (SPSA) varieties and propose adaptive modifications for multidimensional applications. These adaptive versions dynamically scale and shift the tuning sequences to better match the characteristics of the unknown underlying function, as well as the noise level. We test our algorithms on a variety of representative applications in inventory manage- ment, health care, revenue management, supply chain management, financial engineering and queueing theory. ACM Reference Format: Mark Broadie, Deniz M. Cicek, and Assaf Zeevi, 2013. Multidimensional Stochastic Approximation: Adap- tive Algorithms and Applications ACM Trans. Embedd. Comput. Syst. 1, 1, Article 1 (January 2014), 28 pages. DOI:http://dx.doi.org/10.1145/0000000.0000000 1. INTRODUCTION 1.1. Background and overview of main contributions Stochastic approximation is an iterative procedure which can be used to estimate the root of a function or the point of optimum, where the function values cannot be calcu- lated exactly. This can happen, for example, when function values are estimated from noisy samples in a Monte Carlo simulation procedure. The first type of stochastic approximation algorithm dates back to the work of [Rob- bins and Monro 1951] and focuses on estimating the root of an unknown function. Following this, [Kiefer and Wolfowitz 1952] introduced an algorithm to find the point d E of maximum of a function observed with noise, i.e., to solve maxx∈R f(x) = [f(x)] where f is a noisy observation of a strongly concave function f : Rd R.
    [Show full text]
  • Performance of a Distributed Stochastic Approximation Algorithm
    1 Performance of a Distributed Stochastic Approximation Algorithm Pascal Bianchi, Member, IEEE, Gersende Fort, Walid Hachem, Member, IEEE Abstract In this paper, a distributed stochastic approximation algorithm is studied. Applications of such algorithms include decentralized estimation, optimization, control or computing. The algorithm consists in two steps: a local step, where each node in a network updates a local estimate using a stochastic approximation algorithm with decreasing step size, and a gossip step, where a node computes a local weighted average between its estimates and those of its neighbors. Convergence of the estimates toward a consensus is established under weak assumptions. The approach relies on two main ingredients: the existence of a Lyapunov function for the mean field in the agreement subspace, and a contraction property of the random matrices of weights in the subspace orthogonal to the agreement subspace. A second order analysis of the algorithm is also performed under the form of a Central Limit Theorem. The Polyak-averaged version of the algorithm is also considered. I. INTRODUCTION Stochastic approximation has been a very active research area for the last sixty years (see e.g. [1], [2]). The pattern for a stochastic approximation algorithm is provided by the recursion d θn = θn−1 + γnYn, where θn is typically a R -valued sequence of parameters, Yn is a sequence arXiv:1203.1505v2 [math.OC] 2 Dec 2013 of random observations, and γn is a deterministic sequence of step sizes. An archetypal example of such algorithms is provided by stochastic gradient algorithms. These are characterized by the fact that Yn = g(θn−1) + ξn where g is the gradient of a function g to be minimized, and −∇ r where (ξn)n≥0 is a noise sequence corrupting the observations.
    [Show full text]
  • Arxiv:1909.08749V2 [Stat.ML] 15 Sep 2020 Learning [Ber95a, Ber95b, SB18]
    Instance-dependent `1-bounds for policy evaluation in tabular reinforcement learning Ashwin Pananjady? and Martin J. Wainwrighty;z ?Simons Institute for the Theory of Computing, UC Berkeley yDepartments of EECS and Statistics, UC Berkeley zVoleon Group, Berkeley September 17, 2020 Abstract Markov reward processes (MRPs) are used to model stochastic phenomena arising in opera- tions research, control engineering, robotics, and artificial intelligence, as well as communication and transportation networks. In many of these cases, such as in the policy evaluation problem encountered in reinforcement learning, the goal is to estimate the long-term value function of such a process without access to the underlying population transition and reward functions. Working with samples generated under the synchronous model, we study the problem of esti- mating the value function of an infinite-horizon, discounted MRP on finitely many states in the `1-norm. We analyze both the standard plug-in approach to this problem and a more robust variant, and establish non-asymptotic bounds that depend on the (unknown) problem instance, as well as data-dependent bounds that can be evaluated based on the observations of state- transitions and rewards. We show that these approaches are minimax-optimal up to constant factors over natural sub-classes of MRPs. Our analysis makes use of a leave-one-out decoupling argument tailored to the policy evaluation problem, one which may be of independent interest. 1 Introduction A variety of applications spanning science and engineering use Markov reward processes as models for real-world phenomena, including queuing systems, transportation networks, robotic exploration, game playing, and epidemiology.
    [Show full text]