Automatically Bounding the Taylor Remainder Series: Tighter Bounds and New Applications

Code

• github.com/google/autobound
  OfficialIn paperjax★ 365

Abstract

We present a new algorithm for automatically bounding the Taylor remainder series. In the special case of a scalar function f: R R, our algorithm takes as input a reference point x_0, trust region [a, b], and integer k 1, and returns an interval I such that f(x) - _i=0^k-1 1 i! f^(i)(x_0) (x - x_0)^i I (x - x_0)^k for all x [a, b]. As in automatic differentiation, the function f is provided to the algorithm in symbolic form, and must be composed of known atomic functions. At a high level, our algorithm has two steps. First, for a variety of commonly-used elementary functions (e.g., , ), we use recently-developed theory to derive sharp polynomial upper and lower bounds on the Taylor remainder series. We then recursively combine the bounds for the elementary functions using an interval arithmetic variant of Taylor-mode automatic differentiation. Our algorithm can make efficient use of machine learning hardware accelerators, and we provide an open source implementation in JAX. We then turn our attention to applications. Most notably, in a companion paper we use our new machinery to create the first universal majorization-minimization optimization algorithms: algorithms that iteratively minimize an arbitrary loss using a majorizer that is derived automatically, rather than by hand. We also show that our automatically-derived bounds can be used for verified global optimization and numerical integration, and to prove sharper versions of Jensen's inequality.

Tasks

Reproductions