// ---
// jupyter:
//   jupytext:
//     text_representation:
//       extension: .java
//       format_name: light
//       format_version: '1.5'
//       jupytext_version: 1.14.4
//   kernelspec:
//     display_name: Java
//     language: java
//     name: java
// ---

// # Optimization and Deep Learning
//
// In this section, we will discuss the relationship between optimization and deep learning as well as the challenges of using optimization in deep learning. For a deep learning problem, we will usually define a loss function first. Once we have the loss function, we can use an optimization algorithm in attempt to minimize the loss. In optimization, a loss function is often referred to as the objective function of the optimization problem. By tradition and convention most optimization algorithms are concerned with *minimization*. If we ever need to maximize an objective there is a simple solution: just flip the sign on the objective.
//
// ## Optimization and Estimation
//
// Although optimization provides a way to minimize the loss function for deep
// learning, in essence, the goals of optimization and deep learning are
// fundamentally different. The former is primarily concerned with minimizing an
// objective whereas the latter is concerned with finding a suitable model, given a
// finite amount of data.  In :numref:`sec_model_selection`,
// we discussed the difference between these two goals in detail. For instance,
// training error and generalization error generally differ: since the objective
// function of the optimization algorithm is usually a loss function based on the
// training dataset, the goal of optimization is to reduce the training error.
// However, the goal of statistical inference (and thus of deep learning) is to
// reduce the generalization error.  To accomplish the latter we need to pay
// attention to overfitting in addition to using the optimization algorithm to
// reduce the training error. We begin by importing a few libraries with a function to annotate in a figure.

// +
// %mavenRepo snapshots https://oss.sonatype.org/content/repositories/snapshots/

// %maven ai.djl:api:0.7.0-SNAPSHOT
// %maven org.slf4j:slf4j-api:1.7.26
// %maven org.slf4j:slf4j-simple:1.7.26

// %maven ai.djl.mxnet:mxnet-engine:0.7.0-SNAPSHOT
// %maven ai.djl.mxnet:mxnet-native-auto:1.7.0-b
// -

// %load ../utils/plot-utils

// +
import ai.djl.ndarray.*;

import java.lang.Math;
import org.apache.commons.lang3.ArrayUtils;

// + [markdown] pycharm={"name": "#%% md\n"}
// The graph below illustrates the issue in some more detail. Since we have only a finite amount of data the minimum of the training error may be at a different location than the minimum of the expected error (or of the test error).

// + attributes={"classes": [], "id": "", "n": "2"} pycharm={"name": "#%%\n"}
import java.util.function.Function;

// Saved in Functions class for later use
public float[] callFunc(float[] x, Function<Float, Float> func) {
    float[] y = new float[x.length];
    for (int i = 0; i < x.length; i++) {
        y[i] = func.apply(x[i]);
    }
    return y;
}

Function<Float, Float> f = x -> x * (float)Math.cos(Math.PI * x);

Function<Float, Float> g = x -> f.apply(x) + 0.2f * (float)Math.cos(5 * Math.PI * x);

NDManager manager = NDManager.newBaseManager();

NDArray X = manager.arange(0.5f, 1.5f, 0.01f);
float[] x = X.toFloatArray();
float[] fx = callFunc(x, f);
float[] gx = callFunc(x, g);

String[] grouping = new String[x.length * 2];
for (int i = 0; i < x.length; i++) {
    grouping[i] = "Expected Risk";
    grouping[i + x.length] = "Empirical Risk";
}

Table data = Table.create("Data")
    .addColumns(
        FloatColumn.create("x", ArrayUtils.addAll(x, x)),
        FloatColumn.create("risk", ArrayUtils.addAll(fx, gx)),
        StringColumn.create("grouping", grouping)
    );

LinePlot.create("Risk", data, "x", "risk", "grouping");

// + [markdown] pycharm={"name": "#%% md\n"}
// ## Optimization Challenges in Deep Learning
//
// In this chapter, we are going to focus specifically on the performance of the
// optimization algorithm in minimizing the objective function, rather than a
// model's generalization error.  In :numref:`sec_linear_regression`
// we distinguished between analytical solutions and numerical solutions in
// optimization problems. In deep learning, most objective functions are
// complicated and do not have analytical solutions. Instead, we must use numerical
// optimization algorithms. The optimization algorithms below all fall into this
// category.
//
// There are many challenges in deep learning optimization. Some of the most vexing ones are local minima, saddle points and vanishing gradients. Let us have a look at a few of them.
//
// ### Local Minima
//
// For the objective function $f(x)$, if the value of $f(x)$ at $x$ is smaller than the values of $f(x)$ at any other points in the vicinity of $x$, then $f(x)$ could be a local minimum. If the value of $f(x)$ at $x$ is the minimum of the objective function over the entire domain, then $f(x)$ is the global minimum.
//
// For example, given the function
//
// $$f(x) = x \cdot \text{cos}(\pi x) \text{ for } -1.0 \leq x \leq 2.0,$$
//
// we can approximate the local minimum and global minimum of this function.

// + attributes={"classes": [], "id": "", "n": "3"} pycharm={"name": "#%%\n"}
NDArray X = manager.arange(-1.0f, 2.0f, 0.01f);
float[] x = X.toFloatArray();
float[] fx = callFunc(x, f);

Table data = Table.create("Data")
.addColumns(
    FloatColumn.create("x", x),
    FloatColumn.create("f(x)", fx)
);

LinePlot.create("x * cos(pi * x)", data, "x", "f(x)");

// + [markdown] pycharm={"name": "#%% md\n"}
// The local minimum is at (0.3, -0.25) and the global minimum is at (1.1, -0.95).
// -

// The objective function of deep learning models usually has many local optima. When the numerical solution of an optimization problem is near the local optimum, the numerical solution obtained by the final iteration may only minimize the objective function locally, rather than globally, as the gradient of the objective function's solutions approaches or becomes zero. Only some degree of noise might knock the parameter out of the local minimum. In fact, this is one of the beneficial properties of stochastic gradient descent where the natural variation of gradients over minibatches is able to dislodge the parameters from local minima.
//
// ### Saddle Points
//
// Besides local minima, saddle points are another reason for gradients to vanish. A [saddle point](https://en.wikipedia.org/wiki/Saddle_point) is any location where all gradients of a function vanish but which is neither a global nor a local minimum. Consider the function $f(x) = x^3$. Its first and second derivative vanish for $x=0$. Optimization might stall at the point, even though it is not a minimum.

// + attributes={"classes": [], "id": "", "n": "4"}
Function<Float, Float> cube = x -> x * x * x;

NDArray X = manager.arange(-2.0f, 2.0f, 0.01f);
float[] x = X.toFloatArray();
float[] fx = callFunc(x, cube);

Table data = Table.create("Data")
.addColumns(
    FloatColumn.create("x", x),
    FloatColumn.create("f(x)", fx)
);

LinePlot.create("x^3", data, "x", "f(x)");
// -

// Saddle points in higher dimensions are even more insidious, as the example below shows. Consider the function $f(x, y) = x^2 - y^2$. It has its saddle point at $(0, 0)$. This is a maximum with respect to $y$ and a minimum with respect to $x$. Moreover, it *looks* like a saddle, which is where this mathematical property got its name.

// + [markdown] attributes={"classes": [], "id": "", "n": "5"}
// ![Higher Dimension Saddle.](https://djl-ai.s3.amazonaws.com/resources/d2l-java/higher_dim_saddle.svg)

// + [markdown] pycharm={"name": "#%% md\n"}
// We assume that the input of a function is a $k$-dimensional vector and its
// output is a scalar, so its Hessian matrix will have $k$ eigenvalues
// (refer to :numref:`sec_geometry-linear-algebraic-ops`).
// The solution of the
// function could be a local minimum, a local maximum, or a saddle point at a
// position where the function gradient is zero:
//
// * When the eigenvalues of the function's Hessian matrix at the zero-gradient position are all positive, we have a local minimum for the function.
// * When the eigenvalues of the function's Hessian matrix at the zero-gradient position are all negative, we have a local maximum for the function.
// * When the eigenvalues of the function's Hessian matrix at the zero-gradient position are negative and positive, we have a saddle point for the function.
//
// For high-dimensional problems the likelihood that at least some of the eigenvalues are negative is quite high. This makes saddle points more likely than local minima. We will discuss some exceptions to this situation in the next section when introducing convexity. In short, convex functions are those where the eigenvalues of the Hessian are never negative. Sadly, though, most deep learning problems do not fall into this category. Nonetheless it is a great tool to study optimization algorithms.
//
// ### Vanishing Gradients
//
// Probably the most insidious problem to encounter are vanishing gradients. For instance, assume that we want to minimize the function $f(x) = \tanh(x)$ and we happen to get started at $x = 4$. As we can see, the gradient of $f$ is close to nil. More specifically $f'(x) = 1 - \tanh^2(x)$ and thus $f'(4) = 0.0013$. Consequently optimization will get stuck for a long time before we make progress. This turns out to be one of the reasons that training deep learning models was quite tricky prior to the introduction of the ReLU activation function.
// We can see at the top right of the tanh graph the line becoming parallel to x hence the `vanishing gradient`.

// + attributes={"classes": [], "id": "", "n": "6"} pycharm={"name": "#%%\n"}
Function<Float, Float> tanh = x -> (float)Math.tanh(x);

NDArray X = manager.arange(-2.0f, 5.0f, 0.01f);
float[] x = X.toFloatArray();
float[] fx = callFunc(x, tanh);

Table data = Table.create("Data")
.addColumns(
    FloatColumn.create("x", x),
    FloatColumn.create("f(x)", fx)
);

LinePlot.create("tanh", data, "x", "f(x)");
// -

// As we saw, optimization for deep learning is full of challenges. Fortunately there exists a robust range of algorithms that perform well and that are easy to use even for beginners. Furthermore, it is not really necessary to find *the* best solution. Local optima or even approximate solutions thereof are still very useful.
//
// ## Summary
//
// * Minimizing the training error does *not* guarantee that we find the best set of parameters to minimize the expected error.
// * The optimization problems may have many local minima.
// * The problem may have even more saddle points, as generally the problems are not convex.
// * Vanishing gradients can cause optimization to stall. Often a reparameterization of the problem helps. Good initialization of the parameters can be beneficial, too.
//
//
// ## Exercises
//
// 1. Consider a simple multilayer perceptron with a single hidden layer of, say, $d$ dimensions in the hidden layer and a single output. Show that for any local minimum there are at least $d!$ equivalent solutions that behave identically.
// 1. Assume that we have a symmetric random matrix $\mathbf{M}$ where the entries
//    $M_{ij} = M_{ji}$ are each drawn from some probability distribution
//    $p_{ij}$. Furthermore assume that $p_{ij}(x) = p_{ij}(-x)$, i.e., that the
//    distribution is symmetric (see e.g., :cite:`Wigner.1958` for details).
//     * Prove that the distribution over eigenvalues is also symmetric. That is, for any eigenvector $\mathbf{v}$ the probability that the associated eigenvalue $\lambda$ satisfies $P(\lambda > 0) = P(\lambda < 0)$.
//     * Why does the above *not* imply $P(\lambda > 0) = 0.5$?
// 1. What other challenges involved in deep learning optimization can you think of?
// 1. Assume that you want to balance a (real) ball on a (real) saddle.
//     * Why is this hard?
//     * Can you exploit this effect also for optimization algorithms?

StarcoderdataJupyterScriptsDedupFiltered