# Can a neural network compute $y = x^2$?

In spirit of the famous Tensorflow Fizz Buzz joke and XOr problem I started to think, if it's possible to design a neural network that implements $$y = x^2$$ function?

Given some representation of a number (e.g. as a vector in binary form, so that number 5 is represented as [1,0,1,0,0,0,0,...]), the neural network should learn to return its square - 25 in this case.

If I could implement $$y=x^2$$, I could probably implement $$y=x^3$$ and generally any polynomial of x, and then with Taylor series I could approximate $$y=\sin(x)$$, which would solve the Fizz Buzz problem - a neural network that can find remainder of the division.

Clearly, just the linear part of NNs won't be able to perform this task, so if we could do the multiplication, it would be happening thanks to activation function.

Can you suggest any ideas or reading on subject?

## 4 Answers

Neural networks are also called as the universal function approximation which is based in the universal function approximation theorem. It states that:

In the mathematical theory of artificial neural networks, the universal approximation theorem states that a feed-forward network with a single hidden layer containing a finite number of neurons can approximate continuous functions on compact subsets of Rn, under mild assumptions on the activation function

Meaning a ANN with a non linear activation function could map the function which relates the input with the output. The function $$y = x^2$$ could be easily approximated using regression ANN.

You can find an excellent lesson here with a notebook example.

Also, because of such ability ANN could map complex relationships for example between an image and its labels.

• Thank you very much, this is exactly what I was asking for! Commented Mar 22, 2019 at 13:23
• Although true, it a very bad idea to learn that. I fail to see where any generalization power would arise from. NN shine when there's something to generalize. Like CNN for vision that capture patterns, or RNN that can capture trends. Commented Mar 22, 2019 at 15:21

I think the answer of @ShubhamPanchal is a little bit misleading. Yes, it is true that by Cybenko's universal approximation theorem we can approximate $$f(x)=x^2$$ with a single hidden layer containing a finite number of neurons can approximate continuous functions on compact subsets of $$\mathbb{R}^n$$, under mild assumptions on the activation function.

But the main problem is that the theorem has a very important limitation. The function needs to be defined on a compact subsets of $$\mathbb{R}^n$$ (compact subset = bounded + closed subset). But why is this problematic? When training the function approximator you will always have a finite data set. Hence, you will approximate the function inside a compact subset of $$\mathbb{R}^n$$. But we can always find a point $$x$$ for which the approximation will probably fail. That being said. If you only want to approximate $$f(x)=x^2$$ on a compact subset of $$\mathbb{R}$$ then we can answer your question with yes. But if you want to approximate $$f(x)=x^2$$ for all $$x\in \mathbb{R}$$ then the answer is no (I exclude the trivial case in which you use a quadratic activation function).

Side remark on Taylor approximation: You always have to keep in mind that a Taylor approximation is only a local approximation. If you only want to approximate a function in a predefined region then you should be able to use Taylor series. But approximating $$\sin(x)$$ by the Taylor series evaluated at $$x=0$$ will give you horrible results for $$x\to 10000$$ if you don't use enough terms in your Taylor expansion.

• Nice catch! "compact set". Commented Mar 22, 2019 at 17:14
• Many thanks, mate! Eye-opener! Commented Mar 22, 2019 at 17:23
• In $\mathbb{R}^n$, compact subsets are exactly the sets that are closed and bounded (such as $[0,1]\subset \mathbb{R}$). This is called the Heine-Borel theorem.
– Dave
Commented May 25, 2020 at 12:54

Yes, theoretically speaking they can approximate any function.

Other answers have been very detailed and thorough. However, let me add one interesting aspect: when you try to approximate very simple functions, do not despair if you find that most of models you'll train will fail. That is because simple functions require simple models, and simple models (i.e. models with a very low number of parameters) are extremely non-robust to random initialization of weights. Therefore it's perfectly possible that 9 out of 10 networks you train will fail, initially, at comparatively simple tasks.

Fortunately y = x**2 is a convex optimization problem, which means that, for gradient descent, initialization does not matter.

• Please consider $y=x^2$ is what you want to find. The cost function should be convex. If you make the convex problem, you'll find out it is convex, but the function is approximated using a limited range of numbers. It will have poor outcomes for data out of training range which is far from train set. Commented May 25, 2020 at 11:32
• Yeah... I wrote just above that y = x**2 is a convex problem... it's right there in my comment... Commented May 25, 2020 at 12:34

You can solve this problem using a neural network of just one layer which contains one neuron.

The only requirement for an activation function is that it should be non-linear for at least one neuron in the network (because if all neurons are linear, the network can be replaced by a simple linear function because a linear combination of linear functions is a linear function).

As $$x^2$$ is non-linear, it can be used as an activation function of a neuron.

A neural network with a single neuron having $$x^2$$ as an activation function will solve your problem with a 100% accuracy.

• "Look, Watson, this person is a mathematician. First, he spent a long time, thinking on a trivial question. Second, he gave a perfectly correct answer. Third, it is absolutely useless" :) We don't expect the activation function to be specifically tailored to match the inputs of the neural network. The question silently assumes that we are not lucky and the activation function of the neural network is different from the function we aim to model. Commented Jul 7, 2022 at 1:48
• @BorisBurkov 😂 Just making fun on universal obsession with sigmoid and relu Commented Jul 7, 2022 at 2:01