Newton's method

Revision as of 16:08, 1 July 2022 by Arr0w (talk | contribs) (I added some applications into some more advanced mathematics, specifically how Newton's Method could be applied to concepts in analytic number theory.)

Newton's method uses the derivative of a differentiable function to approximate its real or complex roots. For a function $f(x)$, the approximations are defined recursively by \[x_{i+1} = x_i - \frac{f(x_i)}{f'(x_i)}.\] To begin the recursion, an initial guess $x_0$ must be chosen. Often the choice of $x_0$ determines which of several possible roots is found, and in some cases the initial guess can cause the recursion to cycle or diverge instead of converging to a root.

Derivation

At each step of the recursion, we have $x_i$ and seek a root of $f(x)$. Since all nonconstant linear functions have exactly one root, as long as $f'(x_i) \neq 0$ we can construct a tangent-line approximation to $f(x)$ and find its root as an approximation. The tangent-line approximation of $f(x)$ about $x_i$ is \[f(x_i) + (x - x_i)f'(x_i).\] We seek the value $x = x_{i+1}$ such that the above expression equals $0$; hence, \[f(x_i) + (x_{i+1} - x_i)f'(x_i) = 0.\] Dividing by $f'(x_i)$ (as long as $f'(x_i) \neq 0$), \[\frac{f(x_i)}{f'(x_i)} + x_{i+1} - x_i = 0.\] Therefore, the desired value of $x_{i+1}$ is \[x_{i+1} = x_i - \frac{f(x_i)}{f'(x_{i})}.\]

Worked example

Problem: Find the values of the roots of $f(x) = x^2 - x - 1$.

Solution: We could use the quadratic formula to find that the roots are $\frac{1 \pm \sqrt 5}{2}$, but this approach does not immediately yield a decimal value.

To form a guess, we note that $\sqrt 5$ is slightly larger than $2$, so a suitable guess for the greater root is $\frac{1 + 2}{2} = 1.5$.

The derivative of $x^2 - x - 1$ is $2x - 1$ by the power rule for derivatives. Therefore, the recursive formula is

\[x_{i+1} = x_i - \frac{x_i^2 - x_i - 1}{2x_i - 1} = \frac{x_i^2 + 1}{2x_i - 1}.\]

Starting at $x_0 = 1.5$, we apply this formula repeatedly: \begin{align*} x_1 &= \frac{1.5^2 + 1}{2 * 1.5 - 1} &= 1.625, \\  x_2 &= \frac{1.625^2 + 1}{2 * 1.625 - 1}  &= 1.6180556, \\ x_3 &= \frac{1.6180556^2 + 1}{2 * 1.6180556 - 1} &= 1.6180340, \\ x_4 &= \frac{1.6180340^2 + 1}{2 * 1.6180340 - 1} &= 1.6180340. \\ \end{align*}

Because $x_4 = x_3$ as calculated, the ratio $\frac{f(x_3)}{f'(x_3)}$ must be very close to $0$. Since $f'(x_3) = 2.2360680$ is not too large, $f(x_3)$ must be quite close to $0$, so $x_3 = 1.6180340$ is a very good estimate of the greater root.

The sum of the roots is $1$ by Vieta's formulas, so the lesser root is simply $1 - 1.6180340 = -0.6180340$.

Failure cases

Although powerful, Newton's method is delicate and can be very sensitive to the initial guess and type of function, even when the function is differentiable everywhere.

Zero derivative

Suppose in the above example of finding a root of $x^2 - x - 1$, we had started with $x_0 = 0.5$. In the process of calculating $x_1$, then, we would have to divide by $f'(x_0) = 2x_0 - 1 = 0$, creating an undefined result.

Wrong root

Suppose we try to estimate the greater root as before, but start with $x_0 = -0.5$. Applying Newton's method to $x^2 - x - 1$ yields the following sequence of estimates:

\begin{align*} x_1 &= \frac{-0.5^2 + 1}{2 * (-0.5) - 1} &= -0.625, \\  x_2 &= \frac{(-0.625)^2 + 1}{2 * (-0.625) - 1}  &= -0.6180556, \\ x_3 &= \frac{(-0.6180556)^2 + 1}{2 * (-0.6180556) - 1} &= -0.6180340, \\ x_4 &= \frac{(-0.6180340)^2 + 1}{2 * (-0.6180340) - 1} &= -0.61803395, \\ x_5 &= \frac{(-0.61803395)^2 + 1}{2 * (-0.61803395) - 1} &= -0.61803406, \\ x_6 &= \frac{(-0.61803406)^2 + 1}{2 * (-0.61803406) - 1} &= -0.61803395. \\ \end{align*}

These estimates converge to the lesser root, in this case because the initial estimate was closer to the lesser root than the greater root. In general, predicting which root Newton's method will finally converge to is difficult, and Newton's method may converge to a farther root even if a closer root is available within the interval between the initial estimate and the root.

Periodic behavior without finding a root

When finding roots of $x^2 - x - 1$ as above, the estimates eventually stabilize into a predictable pattern: in the $x_0 = 1.5$ case, $x_4 = x_3$, so every subsequent estimate will equal $x_3$; in the $x_0 = -0.5$ case, $x_6 = x_4$, so the subsequent estimates will oscillate, with every even estimate equal to $x_4$ and every odd estimate equal to $x_5$. These behaviors are due to the limited precision of the calculations; with indefinite precision, the estimates would continue to approach the root without becoming periodic.

However, for some functions, Newton's method can cycle without coming close to a root.

Divergence

The function $f(x) = e^x$ is always positive, so it has no roots. Since $f(x) = e^x$ satisfies the differential equation $f(x) = f'(x)$ for all $x$, the ratio $\frac{f(x)}{f'(x)}$ is always $1$, so the sequence of estimates produced by Newton's method becomes $x_{i+1} = x_i - 1$, diverging in the negative direction.

Applications in Analytic Number Theory (Advanced)

The applications of Newton's Method actually apply to Hensel's Lemma, specifically in the $p$ adic integers $\mathbb{Z}_p$. In particular, take some polynomial $p\in\mathbb{Z}[x]$ and an integer $x$ such that $p(x)\equiv 0\pmod{p}$. Our goal is to approximate some $\hat{x}$ such that $p(\hat{x})\equiv 0\pmod{p^2}$. However, in the $p$ adics, we can't just ``lift" arbitrarily. We actually need to approximate such a solution in the next $p$ power level, and then refine it iteratively.

Here's what that looks like realistically. WLOG we assume that $x$ is some $a_0$ between $0$ and $p-1$. We search for $\hat{x}=a_0+a_1p$ such that $p(\hat{x})\equiv 0\pmod{p^2}$. Note the following critical step:

\[p(\hat{x})=p(a_0+a_1p)=p(a_0)+p'(a_0)a_1p+(a_1p)^2b\]

for some $b\in\mathbb{Z}$. By assumption we know that $p(a_0)\equiv 0\pmod{p}$ so let $p(a_0)=pt$ for some $t$. The desired congruence holds modulo $p^2$ when $t+p'(a_0)a_1\equiv 0\pmod{p}$. When $p'(a_0)\not\equiv 0\pmod{p}$, we see that $a_1\equiv -t/p'(a_0)\pmod{p}$ which means that

\[\hat{x}=a_0+a_1p=a_0-\frac{pt}{p'(a_0)}=x-\frac{p(a_0)}{p'(a_0)}.\]

This is exactly the statement of the Newtonian Algorithm! Our root is approximated, so we found some $\hat{x}$ such that $p(\hat{x})\equiv 0\pmod{p^2}$. But this isn't the whole story.


Theorem: Let $p(x)\in\mathbb{Z}_p[x]$ and $x\in\mathbb{Z}_p$ such that $p(x)\equiv 0\pmod{p^n}$. If $k=v(P'(x))<n/2$ then there exists an $\hat{x}=x-p(x)/p'(x)$ such that (1) $p(\hat{x})\equiv 0\pmod{p^{n+1}}$, (2) $\hat{x}\equiv x\pmod{p^{n-k}}$ and (3) $k=v(p'(x))=v(p'(\hat{x}))$.

Proof: Let $p(x)=p^ny$ for $y\in\mathbb{Z}_p$ and $p'(x)=p^ku$ where $u\in\mathbb{Z}_p^{\times}$. By definition we see that

\[\hat{x}-x=-\frac{p(x)}{p'(x)}=-\frac{p^ny}{p^ku}=-p^{n-k}yu^{-1}\in p^{n-k}\mathbb{Z}_p.\]

This proves (2). Notice, however, that in the Taylor expansion of $p$ about $x$ we see that

\[p(\hat{x})=p(x)-\frac{p(x)p'(x)}{p'(x)}+(\hat{x}-x)^2t=(\hat{x}-x)^2t\]

where $t\in\mathbb{Z}_p$. Also notice that

\[p(\hat{x})=(\hat{x}-x)^2t\in p^{2n-2k}\mathbb{Z}_p\]

but we finish (1) since $p^{2n-2k}\mathbb{Z}_p=p^n\cdot p^{n-2k}\mathbb{Z}_p\subset p^{n-1}\mathbb{Z}_p$. To prove (3) we just compute the order of $p'(\hat{x})$. Taking the Taylor expansion of $p'$ at point $x$ gives

\[p'(\hat{x})=p'(x+(\hat{x}-x))=p'(x)+(\hat{x}-x)s=p^ku+p^{n-k}zs=p^k(u+p^{n-2k}zs)=p^kv.\]

But since $k<n/2\implies 2k<n\implies 2k-n<0$ and $u\in\mathbb{Z}_p^{\times}$ we have

\[v=u+p^{n-2k}zs\in u+p\mathbb{Z}_p\subset \mathbb{Z}_p^{\times}\]

which proves that $k=v(p'(x))=v(p'(\hat{x}))$ $\square$

This sets up the entire motivation behind Hensel's Lemma in a more formal way. In addition, the proof of the lemma is just an extension of what we previously noted, which actually makes all of our previous work worth it in the long run.


Hensel's Lemma: Assume $P\in\mathbb{Z}_p[x]$ and $x\in\mathbb{Z}_p$ satisfies $P(x)\equiv 0\pmod{p^n}$. If $k=v(P'(x))<n/2$, then there exists a unique root $\xi$ of $P\in\mathbb{Z}_p[x]$ such that $\xi\equiv x\pmod{p^{n-k}}$ and $k=v(P'(x))=v(P'(\xi))$.

Proof: We first prove the existence of $\xi$. Let $x_0=x$ be given. We can construct $x_1\in\mathbb{Z}$ such that \[x_1\equiv x\pmod{p^{n-k}}~\text{and}~p(x_1)\equiv 0\pmod{p^{n+1}},~v(p'(x_0))=v(p'(x_1))=k.\] We can do this for $x_2$ and $x_n$ such that we have a sequence $(x_n)_{n\ge 0}$ where $x_n\to \xi$ as $n\to \infty$ where $p(\xi)=0$ and $\xi\equiv x\pmod{p^{n-k}}$. Now we prove that $\xi$ is unique. For the sake of contradiction, suppose that there are two roots $\xi$ and $\omega$ satisfying

\[\xi\equiv \omega\pmod{p^{n-k}}.\]

Since it was given that $k<n/2$ we have $2k<n$. This means that $n-k\ge k+1$ so we have at the very least

\[\xi\equiv \omega\pmod{p^{k+1}}.\]

Now we have,

\[p(\omega)=p(\xi)+p'(\xi)(\omega-\xi)+(\omega-\xi)^2a\]

for some $a\in\mathbb{Z}_p$. Since $\xi$ and $\omega$ were assumed to be roots, we factor to get

\[(\omega-\xi)(p'(\xi)+(\omega-\xi)a)=0\]

but since $v(p'(\xi))=k$ and $v((\omega-\xi)a)\ge k+1$

\[p'(\xi)+(\omega-\xi)a\ne 0\]

which means that $\omega-\xi=0\implies \omega=\xi$, so uniqueness holds $\square$

See also