Sample complexity

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.[1]

Definition[edit]

Let be a space which we call the input space, and be a space which we call the output space, and let denote the product . For example, in the setting of binary classification, is typically a finite-dimensional vector space and is the set .

Fix a hypothesis space of functions . A learning algorithm over is a computable map from to . In other words, it is an algorithm that takes as input a finite sequence of training samples and outputs a function from to . Typical learning algorithms include empirical risk minimization, without or with Tikhonov regularization.

Fix a loss function , for example, the square loss , where . For a given distribution on , the expected risk of a hypothesis (a function) is

In our setting, we have , where is a learning algorithm and is a sequence of vectors which are all drawn independently from . Define the optimal risk

Set , for each sample size . is a random variable and depends on the random variable , which is drawn from the distribution . The algorithm is called consistent if probabilistically converges to . In other words, for all , there exists a positive integer , such that, for all sample sizes , we have

The sample complexity of is then the minimum for which this holds, as a function of , and . We write the sample complexity as to emphasize that this value of depends on , and . If is not consistent, then we set . If there exists an algorithm for which is finite, then we say that the hypothesis space is learnable.

In others words, the sample complexity defines the rate of consistency of the algorithm: given a desired accuracy and confidence , one needs to sample data points to guarantee that the risk of the output function is within of the best possible, with probability at least .[2]

In probably approximately correct (PAC) learning, one is concerned with whether the sample complexity is polynomial, that is, whether is bounded by a polynomial in and . If is polynomial for some learning algorithm, then one says that the hypothesis space is PAC-learnable. This is a stronger notion than being learnable.

Unrestricted hypothesis space: infinite sample complexity[edit]

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 , such that, for all , there exists a positive integer such that for all , we have

where , with as above. The No Free Lunch Theorem says that without restrictions on the hypothesis space , this is not the case, i.e., there always exist "bad" distributions for which the sample complexity is arbitrarily large.[1]

Thus, in order to make statements about the rate of convergence of the quantity

one must either

  • constrain the space of probability distributions , e.g. via a parametric approach, or
  • constrain the space of hypotheses , as in distribution-free approaches.

Restricted hypothesis space: finite sample-complexity[edit]

The latter approach leads to concepts such as VC dimension and Rademacher complexity which control the complexity of the space . A smaller hypothesis space introduces more bias into the inference process, meaning that 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.[2]

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

  1. is PAC-learnable.
  2. The VC dimension of is finite.
  3. 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[edit]

, and let be the space of affine functions on , that is, functions of the form for some . 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 is , so it is finite. It follows by the above characterization of PAC-learnable classes that is PAC-learnable, and by extension, learnable.

Sample-complexity bounds[edit]

Suppose is a class of binary functions (functions to ). Then, is -PAC-learnable with a sample of size: [3]

where is the VC dimension of . Moreover, any -PAC-learning algorithm for must have sample-complexity:[4]
Thus, the sample-complexity is a linear function of the VC dimension of the hypothesis space.

Suppose is a class of real-valued functions with range in . Then, is -PAC-learnable with a sample of size: [5][6]

where is Pollard's pseudo-dimension of .

Other Settings[edit]

In addition to the supervised learning setting, sample complexity is relevant to semi-supervised learning problems including active learning,[7] 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,[8] online learning, and unsupervised algorithms, e.g. for dictionary learning.[9]

Efficiency in robotics[edit]

A high sample complexity means that many calculations are needed for running a Monte Carlo tree search.[10] 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.[11] Possible techniques for reducing the sample complexity are metric learning[12] and model-based reinforcement learning.[13]

See also[edit]

References[edit]

  1. ^ a b Vapnik, Vladimir (1998), Statistical Learning Theory, New York: Wiley.
  2. ^ a b Rosasco, Lorenzo (2014), Consistency, Learnability, and Regularization, Lecture Notes for MIT Course 9.520.
  3. ^ Steve Hanneke (2016). "The optimal sample complexity of PAC learning". J. Mach. Learn. Res. 17 (1): 1319–1333. arXiv:1507.00473.
  4. ^ Ehrenfeucht, Andrzej; Haussler, David; Kearns, Michael; Valiant, Leslie (1989). "A general lower bound on the number of examples needed for learning". Information and Computation. 82 (3): 247. doi:10.1016/0890-5401(89)90002-3.
  5. ^ Anthony, Martin; Bartlett, Peter L. (2009). Neural Network Learning: Theoretical Foundations. ISBN 9780521118620.
  6. ^ Morgenstern, Jamie; Roughgarden, Tim (2015). On the Pseudo-Dimension of Nearly Optimal Auctions. NIPS. Curran Associates. pp. 136–144. arXiv:1506.03684.
  7. ^ Balcan, Maria-Florina; Hanneke, Steve; Wortman Vaughan, Jennifer (2010). "The true sample complexity of active learning". Machine Learning. 80 (2–3): 111–139. doi:10.1007/s10994-010-5174-y.
  8. ^ Kakade, Sham (2003), On the Sample Complexity of Reinforcement Learning (PDF), PhD Thesis, University College London: Gatsby Computational Neuroscience Unit.
  9. ^ Vainsencher, Daniel; Mannor, Shie; Bruckstein, Alfred (2011). "The Sample Complexity of Dictionary Learning" (PDF). Journal of Machine Learning Research. 12: 3259–3281.
  10. ^ Kaufmann, Emilie and Koolen, Wouter M (2017). Monte-carlo tree search by best arm identification. Advances in Neural Information Processing Systems. pp. 4897–4906.{{cite conference}}: CS1 maint: multiple names: authors list (link)
  11. ^ Fidelman, Peggy and Stone, Peter (2006). The chin pinch: A case study in skill learning on a legged robot. Robot Soccer World Cup. Springer. pp. 59–71.{{cite conference}}: CS1 maint: multiple names: authors list (link)
  12. ^ Verma, Nakul and Branson, Kristin (2015). Sample complexity of learning mahalanobis distance metrics. Advances in neural information processing systems. pp. 2584–2592.{{cite conference}}: CS1 maint: multiple names: authors list (link)
  13. ^ Kurutach, Thanard and Clavera, Ignasi and Duan, Yan and Tamar, Aviv and Abbeel, Pieter (2018). "Model-ensemble trust-region policy optimization". arXiv:1802.10592 [cs.LG].{{cite arXiv}}: CS1 maint: multiple names: authors list (link)