Sample complexity explained

The sample complexity of a machine learning algorithm represents the number of training-samples that it needs in order to successfully learn a target function.

More precisely, the sample complexity is the number of training-samples that we need to supply to the algorithm, so that the function returned by the algorithm is within an arbitrarily small error of the best possible function, with probability arbitrarily close to 1.

There are two variants of sample complexity:

• The weak variant fixes a particular input-output distribution;
• The strong variant takes the worst-case sample complexity over all input-output distributions.

The No free lunch theorem, discussed below, proves that, in general, the strong sample complexity is infinite, i.e. that there is no algorithm that can learn the globally-optimal target function using a finite number of training samples.

However, if we are only interested in a particular class of target functions (e.g., only linear functions) then the sample complexity is finite, and it depends linearly on the VC dimension on the class of target functions.

Definition

Let

X be a space which we call the input space, and Y be a space which we call the output space, and let Z denote the product X x Y . For example, in the setting of binary classification, X is typically a finite-dimensional vector space and Y is the set \{ -1 , 1 \} .

Fix a hypothesis space

l H of functions h\colon X\to Y . A learning algorithm over l H is a computable map from Z ^{*} to l H . In other words, it is an algorithm that takes as input a finite sequence of training samples and outputs a function from X to Y . Typical learning algorithms include empirical risk minimization, without or with Tikhonov regularization.

Fix a loss function

l{L}\colon Y x Y\to\R _{\geq} , for example, the square loss l{L}(y, y') = (y - y') ^{2} , where h(x) = y' . For a given distribution \rho on X x Y , the expected risk of a hypothesis (a function) h\in l H isl E(h) : = E _{\rho[l{L}(h(x),y)]=\int} _{X x} l{L}(h(x),y)d\rho(x,y)

In our setting, we have

h = l{A}(S _{n)} , where l{A} is a learning algorithm and S _{n} = ((x _{1,y} _{1),\ldots,(x} _{n,y} _{n))\sim} \rho ^{n} is a sequence of vectors which are all drawn independently from \rho . Define the optimal risk$$\backslash mathcal\; E^*\_\backslash mathcal\; =\; \backslash underset\backslash mathcal\; E(h).$$Set h _{n=l{A}(S} _{n)} , for each sample size n . h _{n} is a random variable and depends on the random variable S _{n} , which is drawn from the distribution \rho ^{n} . The algorithm l{A} is called consistent if l E(h _{n)} probabilistically converges to l E ^{*} _{l H} . In other words, for all \epsilon, \delta > 0 , there exists a positive integer N , such that, for all sample sizes n \geq N , we have

The sample complexity of

l{A} is then the minimum N for which this holds, as a function of \rho, \epsilon , and \delta . We write the sample complexity as N(\rho, \epsilon, \delta) to emphasize that this value of N depends on \rho, \epsilon , and \delta . If l{A} is not consistent, then we set N(\rho,\epsilon,\delta) =inf ty . If there exists an algorithm for which N(\rho,\epsilon,\delta) is finite, then we say that the hypothesis space l H is learnable.

In others words, the sample complexity

N(\rho,\epsilon,\delta) defines the rate of consistency of the algorithm: given a desired accuracy \epsilon and confidence \delta , one needs to sample N(\rho,\epsilon,\delta) data points to guarantee that the risk of the output function is within \epsilon of the best possible, with probability at least 1 - \delta .

In probably approximately correct (PAC) learning, one is concerned with whether the sample complexity is polynomial, that is, whether

N(\rho,\epsilon,\delta) is bounded by a polynomial in 1 /\epsilon and 1 /\delta . If N(\rho,\epsilon,\delta) is polynomial for some learning algorithm, then one says that the hypothesis space l H is PAC-learnable. This is a stronger notion than being learnable.

Unrestricted hypothesis space: infinite sample complexity

One can ask whether there exists a learning algorithm so that the sample complexity is finite in the strong sense, that is, there is a bound on the number of samples needed so that the algorithm can learn any distribution over the input-output space with a specified target error. More formally, one asks whether there exists a learning algorithm l{A} , such that, for all \epsilon, \delta > 0 , there exists a positive integer N such that for all n \geq N , we have

where

h _{n=l{A}(S} _{n)} , with S _{n} = ((x _{1,y} _{1),\ldots,(x} _{n,y} _{n))\sim} \rho ^{n} as above. The No Free Lunch Theorem says that without restrictions on the hypothesis space l H , this is not the case, i.e., there always exist "bad" distributions for which the sample complexity is arbitrarily large.

Thus, in order to make statements about the rate of convergence of the quantity$$\backslash sup\_\backslash rho\backslash left(\backslash Pr\_[\backslash mathcal\; E(h\_n)\; -\; \backslash mathcal\; E^*\_\backslash mathcal\{H\}\backslash geq\backslash varepsilon]\backslash right),$$one must either

• constrain the space of probability distributions

\rho , e.g. via a parametric approach, or

• constrain the space of hypotheses

l H , as in distribution-free approaches.

Restricted hypothesis space: finite sample-complexity

The latter approach leads to concepts such as VC dimension and Rademacher complexity which control the complexity of the space

l H . A smaller hypothesis space introduces more bias into the inference process, meaning that l E ^{*} _{l{H}} may be greater than the best possible risk in a larger space. However, by restricting the complexity of the hypothesis space it becomes possible for an algorithm to produce more uniformly consistent functions. This trade-off leads to the concept of regularization.

It is a theorem from VC theory that the following three statements are equivalent for a hypothesis space

l H :l H is PAC-learnable.

1. The VC dimension of

l H is finite.l H is a uniform Glivenko-Cantelli class.This gives a way to prove that certain hypothesis spaces are PAC learnable, and by extension, learnable.

An example of a PAC-learnable hypothesis space

X = \R ^{d,} Y = \{ -1 , 1 \} , and let l H be the space of affine functions on X , that is, functions of the form x\mapsto \langle w,x\rangle + b for some w\in\R ^{d,b\in\R} . This is the linear classification with offset learning problem. Now, four coplanar points in a square cannot be shattered by any affine function, since no affine function can be positive on two diagonally opposite vertices and negative on the remaining two. Thus, the VC dimension of l H is d + 1 , so it is finite. It follows by the above characterization of PAC-learnable classes that l H is PAC-learnable, and by extension, learnable.

Sample-complexity bounds

Suppose l H is a class of binary functions (functions to \{ 0 , 1 \} ). Then, l H is (\epsilon,\delta) -PAC-learnable with a sample of size:^[1] $$N\; =\; O\backslash bigg(\backslash frac\backslash bigg)$$where VC( l H) is the VC dimension of l H .Moreover, any (\epsilon,\delta) -PAC-learning algorithm for l H must have sample-complexity:^[2] $$N\; =\; \backslash Omega\backslash bigg(\backslash frac\backslash bigg)$$Thus, the sample-complexity is a linear function of the VC dimension of the hypothesis space.

Suppose

l H is a class of real-valued functions with range in [ 0 ,T] . Then, l H is (\epsilon,\delta) -PAC-learnable with a sample of size:^{[3] [4]} $$N\; =\; O\backslash bigg(T^2\backslash frac\backslash bigg)$$where PD( l H) is Pollard's pseudo-dimension of l H .

Other settings

In addition to the supervised learning setting, sample complexity is relevant to semi-supervised learning problems including active learning,^[5] where the algorithm can ask for labels to specifically chosen inputs in order to reduce the cost of obtaining many labels. The concept of sample complexity also shows up in reinforcement learning, online learning, and unsupervised algorithms, e.g. for dictionary learning.^[6]

Efficiency in robotics

A high sample complexity means that many calculations are needed for running a Monte Carlo tree search.^[7] It is equivalent to a model-free brute force search in the state space. In contrast, a high-efficiency algorithm has a low sample complexity.^[8] Possible techniques for reducing the sample complexity are metric learning^[9] and model-based reinforcement learning.^[10]

See also

• Active learning (machine learning)

Notes and References

1. The optimal sample complexity of PAC learning. J. Mach. Learn. Res.. 17. 1. 1319–1333. Steve Hanneke. 2016. 1507.00473.
2. 10.1016/0890-5401(89)90002-3. A general lower bound on the number of examples needed for learning. Information and Computation. 82. 3. 247. 1989. Ehrenfeucht. Andrzej. Haussler. David. Kearns. Michael. Valiant. Leslie. free.
3. Book: Martin. Anthony. Peter L.. Bartlett. Neural Network Learning: Theoretical Foundations. 2009. 9780521118620.
4. On the Pseudo-Dimension of Nearly Optimal Auctions. 2015. NIPS. 1506.03684. Morgenstern. Jamie. Roughgarden. Tim. 136–144. Curran Associates.
5. 10.1007/s10994-010-5174-y. The true sample complexity of active learning. Machine Learning. 2010. 80. 2–3. 111–139. Balcan. Maria-Florina. Maria-Florina Balcan. Hanneke. Steve. Wortman Vaughan. Jennifer. free.
6. Vainsencher . Daniel . Mannor . Shie . Bruckstein . Alfred . The Sample Complexity of Dictionary Learning . Journal of Machine Learning Research . 12 . 3259–3281 . 2011 .
7. Monte-carlo tree search by best arm identification . Kaufmann, Emilie and Koolen, Wouter M . Advances in Neural Information Processing Systems . 4897–4906 . 2017 .
8. The chin pinch: A case study in skill learning on a legged robot . Fidelman, Peggy and Stone, Peter . Robot Soccer World Cup . 59–71 . 2006 . Springer .
9. Sample complexity of learning mahalanobis distance metrics . Verma, Nakul and Branson, Kristin . Advances in neural information processing systems . 2584–2592 . 2015 .
10. Model-ensemble trust-region policy optimization . Kurutach, Thanard and Clavera, Ignasi and Duan, Yan and Tamar, Aviv and Abbeel, Pieter . 1802.10592 . 2018 . cs.LG .