Loss Features in TensorFlow


Final Up to date on August 6, 2022

The loss metric is essential for neural networks. As all machine studying fashions are one optimization downside or one other, the loss is the target perform to reduce. In neural networks, the optimization is completed with gradient descent and backpropagation. However what are loss features, and the way are they affecting your neural networks?

On this put up, you’ll be taught what loss features are and delve into some generally used loss features and how one can apply them to your neural networks.

After studying this text, you’ll be taught:

  • What are loss features, and the way they’re completely different from metrics
  • Frequent loss features for regression and classification issues
  • Find out how to use loss features in your TensorFlow mannequin

Let’s get began!

Loss features in TensorFlow
Photograph by Ian Taylor. Some rights reserved.


This text is split into 5 sections; they’re:

  • What are loss features?
  • Imply absolute error
  • Imply squared error
  • Categorical cross-entropy
  • Loss features in apply

What Are Loss Features?

In neural networks, loss features assist optimize the efficiency of the mannequin. They’re normally used to measure some penalty that the mannequin incurs on its predictions, such because the deviation of the prediction away from the bottom fact label. Loss features are normally differentiable throughout their area (however it’s allowed that the gradient is undefined just for very particular factors, reminiscent of x = 0, which is mainly ignored in apply). Within the coaching loop, they’re differentiated with respect to parameters, and these gradients are used on your backpropagation and gradient descent steps to optimize your mannequin on the coaching set.

Loss features are additionally barely completely different from metrics. Whereas loss features can let you know the efficiency of our mannequin, they won’t be of direct curiosity or simply explainable by people. That is the place metrics are available. Metrics reminiscent of accuracy are far more helpful for people to know the efficiency of a neural community though they won’t be good decisions for loss features since they won’t be differentiable.

Within the following, let’s discover some frequent loss features: the imply absolute error, imply squared error, and categorical cross entropy.

Imply Absolute Error

The imply absolute error (MAE) measures absolutely the distinction between predicted values and the bottom fact labels and takes the imply of the distinction throughout all coaching examples. Mathematically, it is the same as $frac{1}{m}sum_{i=1}^mlverthat{y}_i–y_irvert$ the place $m$ is the variety of coaching examples and $y_i$ and $hat{y}_i$ are the bottom fact and predicted values, respectively, averaged over all coaching examples.

The MAE is rarely detrimental and can be zero provided that the prediction matched the bottom fact completely. It’s an intuitive loss perform and may additionally be used as certainly one of your metrics, particularly for regression issues, because you wish to decrease the error in your predictions.

Let’s have a look at what the imply absolute error loss perform seems like graphically:

Imply absolute error loss perform, floor fact at x = 0 and x-axis symbolize the anticipated worth

Just like activation features, you may also be focused on what the gradient of the loss perform seems like since you might be utilizing the gradient later to do backpropagation to coach your mannequin’s parameters.

Imply absolute error loss perform (blue) and gradient (orange)

You may discover a discontinuity within the gradient perform for the imply absolute loss perform. Many are inclined to ignore it because it happens solely at x = 0, which, in apply, not often occurs since it’s the chance of a single level in a steady distribution.

Let’s check out methods to implement this loss perform in TensorFlow utilizing the Keras losses module:

This offers you 2.0 because the output as anticipated, since $ frac{1}{2}(lvert 2-1rvert + lvert 3-0rvert) = frac{1}{2}(4) = 4 $. Subsequent, let’s discover one other loss perform for regression fashions with barely completely different properties, the imply squared error.

Imply Squared Error

One other well-liked loss perform for regression fashions is the imply squared error (MSE), which is the same as $frac{1}{m}sum_{i=1}^m(hat{y}_i–y_i)^2$. It’s just like the imply absolute error because it additionally measures the deviation of the anticipated worth from the bottom fact worth. Nevertheless, the imply squared error squares this distinction (all the time non-negative since squares of actual numbers are all the time non-negative), which provides it barely completely different properties.

One notable one is that the imply squared error favors numerous small errors over a small variety of massive errors, which results in fashions with fewer outliers or at the least outliers which might be much less extreme than fashions skilled with a imply absolute error. It is because a big error would have a considerably bigger affect on the error and, consequently, the gradient of the error when in comparison with a small error.


Imply squared error loss perform, floor fact at x = 0 and x-axis symbolize the anticipated worth

Then, wanting on the gradient,

Imply squared error loss perform (blue) and gradient (orange)

Discover that bigger errors would result in a bigger magnitude for the gradient and a bigger loss. Therefore, for instance, two coaching examples that deviate from their floor truths by 1 unit would result in a lack of 2, whereas a single coaching instance that deviates from its floor fact by 2 items would result in a lack of 4, therefore having a bigger affect.

Let’s have a look at methods to implement the imply squared loss in TensorFlow.

This offers the output 5.0 as anticipated since $frac{1}{2}[(2-1)^2 + (3-0)^2] = frac{1}{2}(10) = 5$. Discover that the second instance with a predicted worth of three and precise worth of 0 contributes 90% of the error underneath the imply squared error vs. 75% underneath the imply absolute error.

Generally, you may even see individuals use root imply squared error (RMSE) as a metric. This can take the sq. root of MSE. From the attitude of a loss perform, MSE and RMSE are equal.

Each MAE and MSE measure values in a steady vary. Therefore they’re for regression issues. For classification issues, you should utilize categorical cross-entropy.

Categorical Cross-Entropy

The earlier two loss features are for regression fashions, the place the output could possibly be any actual quantity. Nevertheless, for classification issues, there’s a small, discrete set of numbers that the output might take. Moreover, the quantity used to label-encode the lessons is bigoted and with no semantic which means (e.g., utilizing the labels 0 for cat, 1 for canine, and a pair of for horse doesn’t symbolize {that a} canine is half cat and half horse). Due to this fact, it mustn’t have an effect on the efficiency of the mannequin.

In a classification downside, the mannequin’s output is a vector of chance for every class. In Keras fashions, this vector is normally anticipated to be “logits,” i.e., actual numbers to be reworked to chance utilizing the softmax perform or the output of a softmax activation perform.

The cross-entropy between two chance distributions is a measure of the distinction between the 2 chance distributions. Exactly, it’s $-sum_i P(X = x_i) log Q(X = x_i)$ for chance $P$ and $Q$. In machine studying, we normally have the chance $P$ offered by the coaching information and $Q$ predicted by the mannequin, through which $P$ is 1 for the proper class and 0 for each different class. The expected chance $Q$, nonetheless, is normally valued between 0 and 1. Therefore when used for classification issues in machine studying, this components might be simplified into: $$textual content{categorical cross entropy} = – log p_{gt}$$ the place $p_{gt}$ is the model-predicted chance of the bottom fact class for that individual pattern.

Cross-entropy metrics have a detrimental signal as a result of $log(x)$ tends to detrimental infinity as $x$ tends to zero. We would like a better loss when the chance approaches 0 and a decrease loss when the chance approaches 1. Graphically,

Categorical cross entropy loss perform, the place x is the anticipated chance of the bottom fact class

Discover that the loss is strictly 0 if the chance of the bottom fact class is 1 as desired. Additionally, because the chance of the bottom fact class tends to 0, the loss tends to optimistic infinity as effectively, therefore considerably penalizing dangerous predictions. You may acknowledge this loss perform for logistic regression, which has similarities besides the logistic regression loss is particular to the case of binary lessons.

Now, wanting on the gradient of the cross entropy loss,

Categorical cross entropy loss perform (blue) and gradient (orange)

Wanting on the gradient, you possibly can see that the gradient is usually detrimental, which can also be anticipated since, to lower this loss, you’d need the chance on the bottom fact class to be as excessive as attainable. Recall that gradient descent goes in the other way of the gradient.

There are two alternative ways to implement categorical cross entropy in TensorFlow. The primary technique takes in one-hot vectors as enter:

This offers the output as 0.2876821 which is the same as $-log(0.75)$ as anticipated. The opposite manner of implementing the explicit cross entropy loss in TensorFlow is utilizing a label-encoded illustration for the category, the place the category is represented by a single non-negative integer indicating the bottom fact class as an alternative.

This likewise provides the output 0.2876821.

Now that you simply’ve explored loss features for each regression and classification fashions, let’s check out how you should utilize loss features in your machine studying fashions.

Loss Features in Observe

Let’s discover methods to use loss features in apply. You’ll discover this via a easy dense mannequin on the MNIST digit classification dataset.

First, obtain the information from the Keras datasets module:

Then, construct your mannequin:

And have a look at the mannequin structure outputted from the above code:

You’ll be able to then compile your mannequin, which can also be the place you introduce the loss perform. Since it is a classification downside, use the cross entropy loss. Particularly, because the MNIST dataset in Keras datasets is represented as a label as an alternative of a one-hot vector, use the SparseCategoricalCrossEntropy loss.

And at last, you prepare your mannequin:

And your mannequin efficiently trains with the next output:

And that’s one instance of methods to use a loss perform in a TensorFlow mannequin.

Additional Studying

Under is a few documentation on loss features from TensorFlow/Keras:


On this put up, you have got seen loss features and the function that they play in a neural community. You may have additionally seen some well-liked loss features utilized in regression and classification fashions, in addition to methods to use the cross entropy loss perform in a TensorFlow mannequin.

Particularly, you realized:

  • What are loss features, and the way they’re completely different from metrics
  • Frequent loss features for regression and classification issues
  • Find out how to use loss features in your TensorFlow mannequin

Develop Deep Studying Initiatives with Python!

Deep Learning with Python

 What If You Might Develop A Community in Minutes

…with just some traces of Python

Uncover how in my new Book:

Deep Studying With Python

It covers end-to-end tasks on subjects like:

Multilayer PerceptronsConvolutional Nets and Recurrent Neural Nets, and extra…

Lastly Carry Deep Studying To

Your Personal Initiatives

Skip the Lecturers. Simply Outcomes.

See What’s Inside


Please enter your comment!
Please enter your name here