The amount of data you need depends both on the complexity of your problem and on the complexity of your chosen algorithm.
This is a fact, but does not help you if you are at the pointy end of a machine learning project.
A common question I get asked is:
How much data do I need?
I cannot answer this question directly for you, or for anyone. But I can give you a handful of ways of thinking about this question.
In this post, I lay out a suite of methods that you can use to think about how much training data you need to apply machine learning to your problem.
My hope that one or more of these methods may help you understand the difficulty of the question and how it is tightly coupled with the heart of the induction problem that you are trying to solve.
Let’s dive into it.
Note: Do you have your own heuristic methods for deciding how much data is required for machine learning?
Please share them in the comments.
Why Are You Asking This Question?
It is important to know why you are asking about the required size of the training dataset.
The answer may influence your next step.
- Do you have too much data? Consider developing some learning curves to find out just how big a representative sample is (below). Or, consider using a big data framework in order to use all available data.
- Do you have too little data? Consider confirming that you indeed have too little data. Consider collecting more data, or using data augmentation methods to artificially increase your sample size.
- Have you not collected data yet? Consider collecting some data and evaluating whether it is enough. Or, if it is for a study or data collection is expensive, consider talking to a domain expert and a statistician.
More generally, you may have more pedestrian questions such as:
- How many records should I export from the database?
- How many samples are required to achieve a desired level of performance?
- How large must the training set be to achieve a sufficient estimate of model performance?
- How much data is required to demonstrate that one model is better than another?
- Should I use a train/test split or k-fold cross validation?
It may be these latter questions that the suggestions in this post seek to address.
In practice, I answer this question myself using learning curves (see below), using resampling methods on small datasets (e.g. k-fold cross validation and the bootstrap), and by adding confidence intervals to final results.
What is your reason for asking about the number of samples required for machine learning?
Please let me know in the comments.
So, how much data do you need?
1. It Depends; No One Can Tell You
No one can tell you how much data you need for your predictive modeling problem.
It is unknowable: an intractable problem that you must discover answers to through empirical investigation.
The amount of data required for machine learning depends on many factors, such as:
- The complexity of the problem, nominally the unknown underlying function that best relates your input variables to the output variable.
- The complexity of the learning algorithm, nominally the algorithm used to inductively learn the unknown underlying mapping function from specific examples.
This is our starting point.
And “it depends” is the answer that most practitioners will give you the first time you ask.
2. Reason by Analogy
A lot of people have worked on a lot of applied machine learning problems before you.
Some of them have published their results.
Perhaps you can look at studies on problems similar to yours as an estimate for the amount of data that may be required.
Similarly, it is common to perform studies on how algorithm performance scales with dataset size. Perhaps such studies can inform you how much data you require to use a specific algorithm.
Perhaps you can average over multiple studies.
Search for papers on Google, Google Scholar, and Arxiv.
3. Use Domain Expertise
You need a sample of data from your problem that is representative of the problem you are trying to solve.
In general, the examples must be independent and identically distributed.
Remember, in machine learning we are learning a function to map input data to output data. The mapping function learned will only be as good as the data you provide it from which to learn.
This means that there needs to be enough data to reasonably capture the relationships that may exist both between input features and between input features and output features.
Use your domain knowledge, or find a domain expert and reason about the domain and the scale of data that may be required to reasonably capture the useful complexity in the problem.
4. Use a Statistical Heuristic
There are statistical heuristic methods available that allow you to calculate a suitable sample size.
Most of the heuristics I have seen have been for classification problems as a function of the number of classes, input features or model parameters. Some heuristics seem rigorous, others seem completely ad hoc.
Here are some examples you may consider:
- Factor of the number of classes: There must be x independent examples for each class, where x could be tens, hundreds, or thousands (e.g. 5, 50, 500, 5000).
- Factor of the number of input features: There must be x% more examples than there are input features, where x could be tens (e.g. 10).
- Factor of the number of model parameters: There must be x independent examples for each parameter in the model, where x could be tens (e.g. 10).
They all look like ad hoc scaling factors to me.
Have you used any of these heuristics?
How did it go? Let me know in the comments.
In theoretical work on this topic (not my area of expertise!), a classifier (e.g. k-nearest neighbors) is often contrasted against the optimal Bayesian decision rule and the difficulty is characterized in the context of the curse of dimensionality; that is there is an exponential increase in difficulty of the problem as the number of input features is increased.
Findings suggest avoiding local methods (like k-nearest neighbors) for sparse samples from high dimensional problems (e.g. few samples and many input features).
For a kinder discussion of this topic, see:
5. Nonlinear Algorithms Need More Data
The more powerful machine learning algorithms are often referred to as nonlinear algorithms.
By definition, they are able to learn complex nonlinear relationships between input and output features. You may very well be using these types of algorithms or intend to use them.
These algorithms are often more flexible and even nonparametric (they can figure out how many parameters are required to model your problem in addition to the values of those parameters). They are also high-variance, meaning predictions vary based on the specific data used to train them. This added flexibility and power comes at the cost of requiring more training data, often a lot more data.
In fact, some nonlinear algorithms like deep learning methods can continue to improve in skill as you give them more data.
If a linear algorithm achieves good performance with hundreds of examples per class, you may need thousands of examples per class for a nonlinear algorithm, like random forest, or an artificial neural network.
6. Learning Curves of Training Dataset Size vs Model Skill
It is common when developing a new machine learning algorithm to demonstrate and even explain the performance of the algorithm in response to the amount of data or problem complexity.
These studies may or may not be performed and published by the author of the algorithm, and may or may not exist for the algorithms or problem types that you are working with.
I would suggest performing your own study with your available data and a single well-performing algorithm, such as random forest.
Design a study that evaluates model skill versus the size of the training dataset.
Plotting the result as a line plot with training dataset size on the x-axis and model skill on the y-axis will give you an idea of how the size of the data affects the skill of the model on your specific problem.
This graph is called a learning curve.
From this graph, you may be able to project the amount of data that is required to develop a skillful model, or perhaps how little data you actually need before hitting an inflection point of diminishing returns.
I highly recommend this approach in general in order to develop robust models in the context of a well-rounded understanding of the problem.
7. Naive Guesstimate
You need lots of data when applying machine learning algorithms.
Often, you need more data than you may reasonably require in classical statistics.
I often answer the question of how much data is required with the flippant response:
Get and use as much data as you can.
If pressed with the question, and with zero knowledge of the specifics of your problem, I would say something naive like:
- You need thousands of examples.
- No fewer than hundreds.
- Ideally, tens or hundreds of thousands for “average” modeling problems.
- Millions or tens-of-millions for “hard” problems like those tackled by deep learning.
Again, this is just more ad hoc guesstimating, but it’s a starting point if you need it. So get started!
8. Get More Data (No Matter What!?)
Big data is often discussed along with machine learning, but you may not require big data to fit your predictive model.
Some problems require big data, all the data you have. For example, simple statistical machine translation:
If you are performing traditional predictive modeling, then there will likely be a point of diminishing returns in the training set size, and you should study your problems and your chosen model/s to see where that point is.
Keep in mind that machine learning is a process of induction. The model can only capture what it has seen. If your training data does not include edge cases, they will very likely not be supported by the model.
Don’t Procrastinate; Get Started
Now, stop getting ready to model your problem, and model it.
Do not let the problem of the training set size stop you from getting started on your predictive modeling problem.
In many cases, I see this question as a reason to procrastinate.
Get all the data you can, use what you have, and see how effective models are on your problem.
Learn something, then take action to better understand you have with further analysis, extend the data you have with augmentation, or gather more data from your domain.
This section provides more resources on the topic if you are looking go deeper.
There is a lot of discussion around this question on Q&A sites like Quora, StackOverflow, and CrossValidated. Below are few choice examples that may help.
I expect that there are some great statistical studies on this question; here are a few I could find.
Other related articles.
If you know of more, please let me know in the comments below.
In this post, you discovered a suite of ways to think and reason about the problem of answering the common question:
How much training data do I need for machine learning?
Did any of these methods help?
Let me know in the comments below.
Do you have any questions?
Ask your questions in the comments below and I will do my best to answer.
Except, of course, the question of how much data that you specifically need.