AIExplainer
Machine Learning Intermediate 2 min read

What is step size?

The amount of change applied to a model's parameters during each iteration of training

In machine learning, a step size determines how quickly a model learns from its data. It controls how much the model's parameters are adjusted during each iteration of training. A large step size can lead to fast learning but may also cause the model to overshoot the optimal solution, while a small step size can lead to more precise learning but may be slower.

Think of the step size like the stride of a hiker. A large stride can cover a lot of ground quickly, but may also cause the hiker to trip or miss important landmarks. A small stride is more cautious and allows the hiker to observe their surroundings more closely, but may take longer to reach the destination.

Imagine a self-driving car learning to navigate a new road. The step size would determine how quickly the car adjusts its steering and speed in response to new data from its sensors. A large step size might cause the car to overcorrect and swerve, while a small step size would allow it to make more precise adjustments.

The step size is used in optimization algorithms, such as gradient descent, to update the model's parameters during training. It is typically set before training begins and may be adjusted during training to achieve better results.

A common misconception is that a larger step size is always better, as it can lead to faster training times. However, this can also lead to overshooting and poor model performance.

The concept of step size has been around since the early days of machine learning and has evolved over time to become a crucial component of optimization algorithms.

learning rate iteration step

Three products for different needs — explore what’s relevant to you.