最小二乗法で始めるカーブフィッティング

この記事は, 旧ブログから移植された記事です. よって, その内容として, 旧ブログに依存した文脈が含まれている可能性があります. 予めご了承下さい.

要旨

本エントリー(WIP)はカーブフィッティング全般に関して記述したものであり, それぞれの原理, 性質について学んだ際のメモとして, より単純なものから広く浅く挙げています. 極力ないようにはしていますが, 本内容は独学で得た知見より書いておりますので, 一部正確さが欠けている可能性があることは否めません. 何かありましたら, コメント等で指摘していただけるとありがたいです. また, 本エントリ内における近似およびプロット等に関する実装は次のリポジトリ

にまとまっています.

線形回帰

まずは, 回帰解析のうち最も基本的な手法である最小二乗法について. 次のような散布図1を考える.

データセット

このデータセットは, マイルランという中距離マラソンにおける男子世界記録の遷移を表しており, 横軸が世界記録を更新した年, 縦軸がその記録の秒数となっている. この散布図は負の相関関係があるといえる. まあ, 記録の更新というのは, ゴールするまでのタイムが縮んだということをいうので, これは当たり前の相関関係である.

このような直線的な関係があるようにみえるような散布について, それなりにそれらの点に相応しいような直線, つまり各点からの距離が最も小さくなるような直線を引きたいとしよう. これが本エントリにおける主題である.

線形最小二乗法

最小二乗法は, 上記のようなデータの組 (xi,yi)(x_i,y_i)nn 組与えられたとき(xix_i の全てが等しい場合を除いて)に, それらの点に相応しい関数 yi=ai+aix11++aixin(i)y_i=a_i+a_ix^1_1+\cdots+a_ix^n_i\tag{i} の係数(傾きと切片)を決定する方法である(定義式は (14)(14)). なお, このときの線形性とは, 係数 aka_k の線形性を意味しており, すなわち応答変数 yy は係数 aka_k の線形関数を表している. まずは最もシンプルな例として, 回帰モデルを y=ax+by=ax+b として考えると, これに対する線形最小二乗法は, 簡単に導出できる:

ある点 (xi,yi)(x_i, y_i) とモデルとなる直線の誤差, すなわち偏差は yi(axi+b)y_i-(ax_i+b) とかける. このとき, 各点からの距離が最小であってほしいのだから, まず総和を ϵ(a,b):=i=1n(yiaxib)2(1)\epsilon(a,b):=\displaystyle\sum^{n}_{i=1}(y_i-ax_i-b)^2\tag{1} とおくこととする. いまここで二乗したのは, モデルとなる直線よりも下に点があった場合, 符号が負となり, これが誤差を相殺してしまったり, 値を負にしてしまうからである. 絶対値を用いないのは, 後の微分計算を可能にするためである.

さて, いま (1)(1) の最小値を求めたいわけだが, 簡単のためにここでまず bb を固定したと考える. すると, (1)(1) は単なる aa の二次関数と捉えることができる. その係数 i=1nxi2\displaystyle\sum^{n}_{i=1}x_i^2 は正であるので, この二次関数は, 下に凸の放物線を描くことがわかる. よってこの二次関数の最小値は, 接線の傾きを 00 とした値をとることがいえるので, それを

ϵ(a,b)a=0\frac{\partial\epsilon(a,b)}{\partial a}=0

とかける. 同様に, aa を固定したと考えれば, 係数は nn でこれは正であるから, これも下に凸の放物線を描くことがわかる. つまりこの場合の最小値も,

ϵ(a,b)b=0\frac{\partial\epsilon(a,b)}{\partial b}=0

とかける. いま求めたかったのはこのどちらをも満たす a,ba,b であるので, これらの連立方程式を解けば良いこととなる. 従って (1)(1) より

{ϵ(a,b)a=i=1n2(yi(axi+b))(xi)=0ϵ(a,b)b=i=1n2(yi(axi+b))=0 \begin{aligned} &&\begin{cases} &\displaystyle\frac{\partial\epsilon(a,b)}{\partial a}&=&\displaystyle\sum_{i=1}^{n}2(y_i-(ax_i+b))\cdot(-x_i)&=&0 \\ &\displaystyle\frac{\partial\epsilon(a,b)}{\partial b}&=&\displaystyle\sum_{i=1}^{n}2(y_i-(ax_i+b))&=&0 \end{cases} \end{aligned}

このような, 線形方程式におけるすべての定数項が 00 であるものを同次線形系(英:homogeneous linear system)という. この両辺を 22 で割って,

{i=1n(yi(axi+b))xi=0i=1n(yi(axi+b))=0{ai=1nxi2+bi=1nxi=i=1nxiyiai=1nxi+bn=i=1nyi(2) \begin{aligned} &\leftrightarrow &\begin{cases} &\displaystyle\sum_{i=1}^{n}(y_i-(ax_i+b))\cdot x_i&=&0 \\ &\displaystyle\sum_{i=1}^{n}(y_i-(ax_i+b))&=&0 \end{cases} \\ &\leftrightarrow &\begin{cases} &\displaystyle a\sum_{i=1}^{n}x^2_i&+&\displaystyle b\sum_{i=1}^{n}x_i&=&\displaystyle\sum_{i=1}^{n}x_i y_i \\ &\displaystyle a\sum_{i=1}^{n}x_i&+&bn&=&\displaystyle\sum_{i=1}^{n}y_i \end{cases} \tag{2} \end{aligned}

両辺を nn で割って,

{ai=1nxi2n+bi=1nxin=i=1nxiyinai=1nxin+b=i=1nyin \begin{aligned} &\leftrightarrow &\begin{cases} &\displaystyle a\frac{\sum_{i=1}^{n}x^2_i}{n}&+&\displaystyle b\frac{\sum_{i=1}^{n}x_i}{n}&=&\displaystyle\frac{\sum_{i=1}^{n}x_i y_i}{n} \\ &\displaystyle a\frac{\sum_{i=1}^n x_i}{n}&+&b&=&\frac{\sum_{i=1}^ny_i}{n} \end{cases} \end{aligned} ここで, i=1nxin\frac{\sum_{i=1}^n x_i}{n}xx の総和をその個数で割っているので xx の平均, i=1nyin\frac{\sum_{i=1}^ny_i}{n}yy の総和をその個数で割っているので yy の平均であるから, x=i=1nxin,y=i=1nyin,x2=i=1nxi2n,xy=i=1nxiyin\overline{x}=\frac{\sum_{i=1}^{n}x_i}{n}, \overline{y}=\frac{\sum_{i=1}^ny_i}{n}, \overline{x^2}=\frac{\sum_{i=1}^{n}x_i^2}{n}, \overline{xy}=\frac{\sum_{i=1}^{n}x_i y_i}{n} と平均の記号を用いて書くことができる. よって, b=ax+yb=-a\overline{x}+\overline{y} を代入すれば aa も求まるわけだが, 一旦これを行列で表現すると,

(x2xx1)(ab)=(xyy) \begin{aligned} \left(\begin{array}{cc} \overline{x^2} & \overline{x} \\ \overline{x} & 1 \end{array}\right) \left(\begin{array}{c} a \\ b \end{array}\right) &=& \left(\begin{array}{c} \overline{xy} \\ \overline{y} \end{array}\right) \end{aligned}

左辺の行列の行列式

det(x2xx1) \begin{aligned} {\rm det}\left(\begin{array}{cc} \overline{x^2} & \overline{x} \\ \overline{x} & 1 \end{array}\right) \end{aligned}

は, xix_i がすべて等しくない限り 00 とはならない. いまはそのような場合を除いているから, 同行列は正則で

(ab)=(x2xx1)1(xyy) \begin{aligned} \left(\begin{array}{c} a \\ b \end{array}\right) &=&\left(\begin{array}{cc} \overline{x^2} & \overline{x} \\ \overline{x} & 1 \end{array}\right)^{-1} \left(\begin{array}{c} \overline{xy} \\ \overline{y} \end{array}\right) \end{aligned}

より (a b)T(a\ b)^T

=(xyxyx2x2x2yxyxx2x2)(3) \begin{aligned} &=&\left(\begin{array}{c} \frac{\overline{xy}-\overline{x}\cdot\overline{y}}{\overline{x^2}-\overline{x}^2} \\ \frac{\overline{x^2}\cdot\overline{y}-\overline{xy}\cdot\overline{x}}{\overline{x^2}-\overline{x}^2} \end{array}\right)\tag{3} \end{aligned}

と求まる. ここで, xyxy\overline{xy}-\overline{x}\cdot\overline{y} は共分散, x2x2\overline{x^2}-\overline{x}^2 は分散の形になっているので, aaCov(x,y)σx2\frac{\mathrm{Cov}(x,y)}{\sigma_x^2} とまとめることができ, よくみる最小二乗法の定義式の形となった. 実際にプログラムにすることを考えるときは, 平均などはどうでもよくて, 単に (3)(3) の各項に nn を乗じた形で計算すればよい. つまり,

(2)(i=1nxi2i=1nxii=1nxin)(ab)=(i=1nxiyii=1nyi)(ab)=(i=1nxi2i=1nxii=1nxin)1(i=1nxiyii=1nyi)=((i=1nxiyi)n(i=1nxi)(i=1nyi)(i=1nxi2)n(i=1nxi)2(i=1nxi2)(i=1nyi)(i=1nxiyi)(i=1nxi)(i=1nxi2)n(i=1nxi)2) \begin{aligned} (2)&\leftrightarrow& \left(\begin{array}{cc} \displaystyle \sum_{i=1}^{n}x^2_i &\displaystyle \sum_{i=1}^{n}x_i \\ \displaystyle \sum_{i=1}^{n}x_i &n \end{array}\right) \left(\begin{array}{c} a \\ b \end{array}\right)&=& \left(\begin{array}{c} \displaystyle\sum_{i=1}^{n}x_i y_i \\ \displaystyle\sum_{i=1}^{n}y_i \end{array}\right) \\ &\leftrightarrow& \left(\begin{array}{c} a \\ b \end{array}\right)&=& \left(\begin{array}{cc} \displaystyle \sum_{i=1}^{n}x^2_i &\displaystyle \sum_{i=1}^{n}x_i \\ \displaystyle \sum_{i=1}^{n}x_i &n \end{array}\right)^{-1} \left(\begin{array}{c} \displaystyle\sum_{i=1}^{n}x_i y_i \\ \displaystyle\sum_{i=1}^{n}y_i \end{array}\right) \\ &&&=&\left(\begin{array}{c} \frac{(\sum^n_{i=1}x_iy_i) n-(\sum^n_{i=1}x_i)(\sum^n_{i=1}y_i)}{(\sum^n_{i=1}x^2_i) n-(\sum^n_{i=1}x_i)^2} \\ \frac{(\sum^n_{i=1}x^2_i)(\sum^n_{i=1}y_i)-(\sum^n_{i=1}x_iy_i)(\sum^n_{i=1}x_i)}{(\sum^n_{i=1}x^2_i) n-(\sum^n_{i=1}x_i)^2} \end{array}\right) \end{aligned}

である. これを用いて, 次のように近似できる.

lenear equations

というのが, 最も素朴な最小二乗法の例である2. より一般に, yymm 次の多項式 f(x)=b+j=1majxj\displaystyle f(x)=b+\sum_{j=1}^{m}a_j x^{j} として表されるような場合についても, 同様にしていうことができる. この場合, 偏差の二乗和は,

ϵ=i=1n(yibj=1majxj)2(4)\displaystyle\epsilon=\sum^{n}_{i=1}(y_i-b-\sum_{j=1}^{m}a_j x^{j})^2\tag{4}

先と同様, 各変数ごとの偏微分が 00 となる連立方程式を解けば良いから,

{ϵb=i=1n2(yibj=1majxj)=0ϵa1=i=1n2xi(yibj=1majxj)=0ϵam=i=1n2xim(yibj=1majxj)=0 \begin{aligned} \begin{cases} &\displaystyle\frac{\partial\epsilon}{\partial b}&=&\displaystyle-\sum_{i=1}^{n}2(y_i-b-\sum_{j=1}^{m}a_j x^{j})&=&0 \\ &\displaystyle\frac{\partial\epsilon}{\partial a_1}&=&\displaystyle-\sum_{i=1}^{n}2x_i(y_i-b-\sum_{j=1}^{m}a_j x^{j})&=&0 \\ &&&\vdots& \\ &\displaystyle\frac{\partial\epsilon}{\partial a_m}&=&\displaystyle-\sum_{i=1}^{n}2x^m_i(y_i-b-\sum_{j=1}^{m}a_j x^{j})&=&0 \end{cases} \end{aligned}

先の例に合わせて, 両辺を 2n2n で割った行列とすると, 平均の記号を用いて

(1xxmxx2xm+1xmxm+1x2m)(ba1am)=(yxyxmy) \begin{aligned} \left(\begin{array}{cccc} 1 & \overline{x} & \cdots & \overline{x^m} \\ \overline{x} & \overline{x^2} & \cdots & \overline{x^{m+1}} \\ \vdots & \vdots & \ddots & \vdots \\ \overline{x^m} & \overline{x^{m+1}} & \cdots & \overline{x^{2m}} \end{array}\right) \left(\begin{array}{c} b \\ a_1 \\ \vdots \\ a_m \end{array}\right)= \left(\begin{array}{c} \overline{y} \\ \overline{xy} \\ \vdots \\ \overline{x^my} \end{array}\right) \end{aligned}

なので,

(ba1am)=(1xxmxx2xm+1xmxm+1x2m)1(yxyxmy)=(ni=1nxii=1nximi=1nxii=1nxi2i=1nxim+1i=1nximi=1nxim+1i=1nxi2m)1(i=1nyii=1nxiyii=1nximyi) \begin{aligned} \left(\begin{array}{c} b \\ a_1 \\ \vdots \\ a_m \end{array}\right)&=& \left(\begin{array}{cccc} 1 & \overline{x} & \cdots & \overline{x^m} \\ \overline{x} & \overline{x^2} & \cdots & \overline{x^{m+1}} \\ \vdots & \vdots & \ddots & \vdots \\ \overline{x^m} & \overline{x^{m+1}} & \cdots & \overline{x^{2m}} \end{array}\right)^{-1} \left(\begin{array}{c} \overline{y} \\ \overline{xy} \\ \vdots \\ \overline{x^my} \end{array}\right) \\ &=&\left(\begin{array}{cccc} n & \displaystyle\sum^{n}_{i=1}x_i & \cdots & \displaystyle\sum^{n}_{i=1}x^m_i \\ \displaystyle\sum^{n}_{i=1}x_i & \displaystyle\sum^{n}_{i=1}x_i^2 & \cdots & \displaystyle\sum^{n}_{i=1}x_i^{m+1} \\ \vdots & \vdots & \ddots & \vdots \\ \displaystyle\sum^{n}_{i=1}x_i^m & \displaystyle\sum^{n}_{i=1}x_i^{m+1} & \cdots & \displaystyle\sum^{n}_{i=1}x_i^{2m} \end{array}\right)^{-1} \left(\begin{array}{c} \displaystyle\sum^n_{i=1}y_i \\ \displaystyle\sum^n_{i=1}x_iy_i \\ \vdots \\ \displaystyle\sum^n_{i=1}x_i^my_i \end{array}\right) \end{aligned}

で求まる. この正確な解を機械的に求める場合には, この逆行列を求めなくとも, ガウスの消去法などを基本とした解法(直接解法)で解ける.

先のデータは直線的であったので, 今度は曲線が引けそうなデータセットとして, xi=i1,yi=sin(xi)+ϵ,i=1,2,,11x_i=i-1, y_i=\sin(x_i)+\epsilon, i=1,2,\cdots,11 に対し, フィッティングを試行してみる事とする. ここで ϵ\epsilonN(0,0.2)\mathrm{N}(0, 0.2) の正規分布に従う確率変数である. 次のアニメーションでは, 次数 1m91\leq m\leq 9 に応じた近似の遷移を観察できる(LU 分解による計算 \to 解説).

lenear equations

ところで, 冒頭で示した関数 (i)(i) の線形回帰モデルは, 次のように表すことができる.

yi=a0+j=1majfj(xi1,xi2,,xin)+ui, i=1,,m(ii)y_i=a_0+\sum^m_{j=1}a_jf_j(x_i^1,x_i^2,\cdots,x_i^n)+u_i,\ i=1,\cdots,m\tag{ii}

ここで, fjf_j は独立変数 xikx_{i}^k のスカラー関数, uiu_iii 番目のノイズ項(確率変数)である. 線形最小二乗法は, 単にすべてのデータ値に対する偏差の二乗和を最小化する. すなわち, データに関わらず全ての値が同じように扱われる. これは, すべてのノイズ項 uiu_i の確率分布が同一であると仮定することと同値であり, 従って, すべての uiu_i は無相関かつ i.i.dN(0,σ2)\mathrm{N}(0,\sigma^2) (標準正規分布) に従うことを前提としているといえる3.

一般逆行列

ここまでは, 回帰直線の考え方に沿って近似曲線/直線を得た訳であるが, そもそも, (xi,yi)(x_i,y_i) の組があって線形方程式 yyxx に関する関数における”適当な”係数が”直接”求まるような行列があれば良いのではないだろうか. つまり mm を方程式の個数, n=n+1n'=n+1 を未知数の個数とし, Xa=y  where XRm×n,aRn×1,yRm×1X{\boldsymbol a}={\boldsymbol y}\ \ {\rm where}\ X\in\mathbb{R}^{m\times n'}, {\boldsymbol a}\in\mathbb{R}^{n'\times 1}, {\boldsymbol y}\in\mathbb{R}^{m\times 1}4 としたとき

(y1y2ym)=(x10x11x1nx20x21x2nxm0xm1xmn)(a0a1an)(x10x11x1nx20x21x2nxm0xm1xmn)1(y1y2ym)=(a0a1an)(5) \begin{aligned} \left(\begin{array}{c} y_1 \\ y_2 \\ \vdots \\ y_m \end{array}\right)= \left(\begin{array}{cccc} x_1^0 & x_1^1 & \cdots & x_1^n \\ x_2^0 & x_2^1 & \cdots & x_2^n \\ \vdots & \vdots & \ddots & \vdots \\ x_m^0 & x_m^1 &\cdots & x_m^n \end{array}\right) \left(\begin{array}{c} a_0 \\ a_1 \\ \vdots \\ a_n \end{array}\right) \\ \left(\begin{array}{cccc} x_1^0 & x_1^1 & \cdots & x_1^n \\ x_2^0 & x_2^1 & \cdots & x_2^n \\ \vdots & \vdots & \ddots & \vdots \\ x_m^0 & x_m^1 &\cdots & x_m^n \end{array}\right)^{-1} \left(\begin{array}{c} y_1 \\ y_2 \\ \vdots \\ y_m \end{array}\right) = \left(\begin{array}{c} a_0 \\ a_1 \\ \vdots \\ a_n \end{array}\right)\tag{5} \end{aligned}

を解いて, それが求まれば良いのではないか, ということである(n+1n+1mm は必ずしも等しくないことに注意5). このときに考えられるパターンは, 次のとおりである:

  1. n=mn'=m かつそのランク rank(X){\rm rank}(X)n=mn'=m(フルランク)ならば, XX は正則である. 従って, X1Xa=X1yX^{-1}X{\boldsymbol a}=X^{-1}{\boldsymbol y} として解が求まる.
  2. n<mn'\lt m かつそのランクが nn'(列フルランク)ならば, すべての方程式を満たすような解が存在しないことがいえる. これは, すべての方程式をそれぞれ直線と捉えたときに, それらすべての交点となる一点が存在しないことをイメージするとわかりやすい. 要するに, a{\boldsymbol a} に対する XXy{\boldsymbol y} の制約が相互的に成立しないのである. ここで rank(X)<n{\rm rank}(X)\lt n'(ランク落ち)ならば, 方程式のどこかに重複がある.
  3. m<nm\lt n' かつそのランクが mm(行フルランク)ならば, 解が一意とならないことがいえる. これも, 方程式を直線に捉えると, a{\boldsymbol a} に対する制約が足りないことで, 方程式で構成される直線上のすべてが解となりうることから納得できる. ここで rank(X)<m{\rm rank}(X)\lt m(ランク落ち)ならば, 方程式のどこかに重複がある.

いま 2 つの重複がある場合を考えることができたが, 重複を除けば, いま述べたうちのどれかに帰着させることができる. 重複の場合を直線で捉えると, それぞれの方程式が x,yx, y の関係に関して全く異なる解を示しているということなので, それぞれの直線は平行の関係にあることとなる.

まとめると, つまり Xa=yX{\boldsymbol a}={\boldsymbol y} というように表される線形方程式には, 以上の 3 つのパターン(重複について考えれば 4 パターン)があることがわかる. これらのすべてのパターンに対して”適当であるような”解を与える逆行列を考えれば, どのような方程式にも”適当であるような”解を与えることができる. このように, 正則でない行列に対する擬似的な逆行列の定義を一般逆行列という.

一般逆行列

次の式を満たす行列 XRm×nX^-\in\mathbb{R}^{m\times n'} を一般逆行列といい, XX が特異行列ならば XX^- は一意ではないが常に存在する. XXX=XX X^-X = X

“適当であるような”解は様々に考えられるように, 一般逆行列の定義も様々である. 以下やや天下り的ではあるが便宜上の理由より示してしまうと, いくらかの一般逆行列は次で定めるムーア・ベンローズ一般逆行列(以下 MP 逆行列)に従っており, 暗に一般逆行列と言ってこの MP 逆行列のことを示すような場合が巷ではある6.

Moore-Penrose 一般逆行列

次のすべての条件を満たす一般逆行列 XX^{\dagger} は Moore-Penrose 一般逆行列(MP 逆行列)といい, その存在は一意7である. XXX=X(6)X X^\dagger X=X\tag{6} XXX=X(7)X^{\dagger}X X^{\dagger}=X^{\dagger}\tag{7} (XX)T=XX(8)(X^{\dagger}X)^T=X^{\dagger}X\tag{8} (XX)T=XX(9)(X X^{\dagger})^T=X X^{\dagger}\tag{9}

最小二乗形一般逆行列

まず, ケース 2 の場合について考える. これは, 最小二乗形一般逆行列といわれる一般逆行列を用いる. これが定める”適当であるような”解とは, その名の通り, すべての方程式の二乗誤差が最小である値であり, まさしく上で述べた最小二乗法の値である.

最小二乗形一般逆行列

正規方程式 a=Xy{\boldsymbol a}=X^-{\boldsymbol y} の解 a{\boldsymbol a} を二乗誤差最小の値で定める 一般逆行列 XRm×n s.t. m>n(XX)T=XX^\exists X^-\in\mathbb{R}^{m\times n'}\ {\rm s.t.}\ m\gt n'\land(X X^-)^T = X X^-XX の最小二乗形一般逆行列である.

以下, 最小二乗形一般逆行列の定式を求めるが, 上で既に述べた内容と本質的には全く変わらない. ここで, 少し扱いやすくするために, nn 次多項式を fn(x)=a0xi0+a1xi1++anxin=j=0najxijf_n(x)=a_0x^0_i+a_1x^1_i+\cdots+a_nx^n_i=\sum^{n}_{j=0}a_jx^j_i, (5)(5)xijx_i^j についての行列を XX とする. そしてその ii 行目を 1 つの縦ベクトルとしたものを xi{\boldsymbol x_i} (e.g.  x1=(x10,x11,,x1n)T{\rm e.g.}\ \ {\boldsymbol x_1}=(x_1^0, x_1^1, \cdots, x_1^n )^T) とし, (4)(4) の式を

ϵ=i=1m(yifn(xi))2(10)\epsilon=\sum_{i=1}^{m}(y_i-f_n({\boldsymbol x_i}))^2\tag{10}

というように表す(これは (4)(4) と全く同じことを書いただけである)とする. f(xi)=xiTaf({\boldsymbol x_i})={\boldsymbol x_i}^T{\boldsymbol a} だから

=i=1m(yixiTa)2=\sum_{i=1}^{m}(y_i-{\boldsymbol x_i}^T{\boldsymbol a})^2

(x1T,x2T,,xmT)T=X({\boldsymbol x_1}^T,{\boldsymbol x_2}^T,\cdots,{\boldsymbol x_m}^T)^T=X なので

=(yXa)T(yXa)=({\boldsymbol y}-X{\boldsymbol a})^T({\boldsymbol y}-X{\boldsymbol a})

ここで, 先にやった, 偏微分を考えることで下に凸な二次関数となることを利用し, その値を 00 とした上でそれらすべての連立方程式を求め, 最小値を得たことを思いだし, この式を a{\boldsymbol a} で微分する(すべての ai{\boldsymbol a_i} で偏微分する, すなわち勾配を求める)と

ϵ(a)=2XTXa2XTy=2XT(yXa)\nabla\epsilon({\boldsymbol a})=2X^T X{\boldsymbol a}-2X^T{\boldsymbol y}=-2X^T({\boldsymbol y}-X{\boldsymbol a})

ϵ(a)=0\nabla\epsilon({\boldsymbol a})=0 とおくと,

XTXa=XTyX^T X{\boldsymbol a}=X^T{\boldsymbol y}

と正規方程式が求まった. ここで, n=m1n'=m-1 のとき XX はヴァンデルモンド行列8となり, x1,,xm{\boldsymbol x_1}, \cdots, {\boldsymbol x_m} が相異なるとき XX は正則となる. 従って, 正規方程式の解は

a=(XTX)1XTy=X1y{\boldsymbol a}=(X^T X)^{-1}X^T {\boldsymbol y}={X}^{-1}{\boldsymbol y}

とかける. n<m1n'\lt m-1 ならば行列 XTXX^T X が正則なので, 正規方程式の解は

a=(XTX)1XTy(11){\boldsymbol a}=(X^T X)^{-1}X^T{\boldsymbol y}\tag{11}

とかける. このとき m<nm\lt n ならば, XTXX^T X が非正則となってしまうから, 最小二乗形一般逆行列は構成できない. この結果から, 時間計算量は多項式時間 O(n3)\mathrm{O}(n^3) であることがわかる. また, 最小二乗形一般逆行列は, MP 逆行列であることが導出できる9.

最小ノルム形一般逆行列

次に, ケース 3 の場合を考える. この場合, 最小ノルム形一般逆行列を用いる. ケース 3 は様々な値が解になりうるということであったが, 最小ノルム形一般逆行列は, いまそれを XX^- としたとき, a=Xy{\boldsymbol a}=X^-{\boldsymbol y} の解 a{\boldsymbol a} を その L2L^2 ノルム a2\mid\mid {\boldsymbol a}\mid\mid_2 が最小となるように定める.

最小ノルム形一般逆行列

正規方程式 a=Xy{\boldsymbol a}=X^-{\boldsymbol y} の解 a{\boldsymbol a} をその L2L^2 ノルム a2\mid\mid {\boldsymbol a}\mid\mid_2 が最小となる値で定める一般逆行列 XRm×n s.t. m<n(XX)T=XX^\exists X^-\in\mathbb{R}^{m\times n'}\ {\rm s.t.}\ m\lt n'\land(X^- X)^T = X^- XXX の最小ノルム形一般逆行列である.

つまり, 解くべきは次に示す制約付き最適化問題/条件付き極小値問題である.

minaa22 s.t. y=Xa\min_{{\boldsymbol a}}\mid\mid{\boldsymbol a}\mid\mid^2_2\ {\rm s.t.}\ {\boldsymbol y}=X{\boldsymbol a}

条件付き極値の問題はラグランジュの未定乗数法で解ける. この証明は中々大変なので, 本エントリでは公理として認めた上で用いることとする(TODO). 従って, ラグランジアンを次のように定義する.

L(a,λ):=a22+λT(yXa)\mathcal{L}({\boldsymbol a}, {\boldsymbol \lambda}):=\mid\mid{\boldsymbol a}\mid\mid^2_2+{\boldsymbol \lambda}^T({\boldsymbol y}-X{\boldsymbol a})

ラグランジュの未定乗数法に従い, それぞれの偏導関数から求めて

{aL(a)=2aXTλ=0λL(a)=yXa=0 \begin{aligned} \begin{cases} \frac{\partial}{\partial{\boldsymbol a}}\mathcal{L}({\boldsymbol a})&=&2{\boldsymbol a}-X^T{\boldsymbol \lambda}&=&0 \\ \frac{\partial}{\partial{\boldsymbol \lambda}}\mathcal{L}({\boldsymbol a})&=&{\boldsymbol y}-X{\boldsymbol a}&=&0 \end{cases} \end{aligned}

よって

{a=12XTλy=Xay=12XXTλ \begin{aligned} \begin{cases} {\boldsymbol a}&=&\frac{1}{2}X^T\lambda \\ {\boldsymbol y}&=&X{\boldsymbol a} \end{cases}\leftrightarrow{\boldsymbol y}=\frac{1}{2}X X^T\lambda \end{aligned}

m<nm\lt n' ならば XXTX X^T は正則なので

λ=2(XXT)1ya=XT(XXT)1y\lambda=2(X X^T)^{-1}{\boldsymbol y}\leftrightarrow {\boldsymbol a}=X^T(X X^T)^{-1}{\boldsymbol y}

Xa=X{XT(XXT)1y}=(XXT)(XXT)1y=yX{\boldsymbol a}=X\left\{X^T(X X^T)^{-1}{\boldsymbol y}\right\}=(X X^T)(X X^T)^{-1}{\boldsymbol y}={\boldsymbol y} よりこの正規方程式の解が一般逆行列として成立していることが確認できる.

制限つき最小二乗法

最後に, 重複がある(ランク落ちである)ケースを考える. この場合は, XTX,XXTX^T X, X X^T がともに特異行列となってしまうため, 対象の行列に対してまず特異値分解(以下 SVD)を行う.

特異値分解

XRm×n^\forall X\in\mathbb{R}^{m\times n'} に対して URm×m,VRn×n,ΣRn×m s.t. X=UΣVTwhere Σ=(λ100λr00),λ1λr0,r=rank(X)=min(m,n) ^\exists U\in\mathbb{R}^{m\times m}, ^\exists V\in\mathbb{R}^{n'\times n'}, ^\exists \Sigma\in\mathbb{R}^{n'\times m}\ {\rm s.t.}\ X = U\Sigma V^T \\ {\rm where}\ \Sigma=\left(\begin{array}{ccccc}\lambda_1&\cdots&0 \\ \vdots&\ddots&\vdots \\ 0&\cdots&\lambda_{r} \\ &&&0 \\ &&&&0 \end{array}\right), \lambda_1\geq\cdots\geq\lambda_{r}\geq 0,r=\mathrm{rank}(X)=\min(m,n') このとき UΣVTU\Sigma V^TXX の特異値分解 (英: Singular value decomposition) という.

これは

i=1rλiuiviT where (u1,,um)T=(u11u1mum1umm)=U(v1,,vn)T=(v11v1nvn1vnn)=V \begin{aligned} \displaystyle\sum^{r}_{i=1}\lambda_i{\boldsymbol u_i}{\boldsymbol v_i}^T\ {\rm where} \ \begin{array}{cc} ({\boldsymbol u_1},\cdots,{\boldsymbol u_m})^T &=&\left(\begin{array}{ccc} u_{11}&\cdots&u_{1m} \\ \vdots&\ddots&\vdots\\ u_{m1}&\cdots&u_{mm} \end{array}\right)&=&U \\ ({\boldsymbol v_1},\cdots,{\boldsymbol v_{n'}})^T &=&\left(\begin{array}{ccc} v_{11}&\cdots&v_{1n'} \\ \vdots&\ddots&\vdots\\ v_{n'1}&\cdots&v_{n'n'} \end{array}\right)&=&V \end{array} \end{aligned}

と同値であり, 一般に λi\lambda_i を特異値, ui{\boldsymbol u_i} を左特異ベクトル vi{\boldsymbol v_i} を右特異ベクトルという.

TODO: 詳解を追記

オーバーフィッティングと正則化およびその評価

先に, 次数に応じた近似の遷移が観察できるアニメーションを示したが, あまり次数を大きくすると, データ点の間で誤差が大きくなってしまうことがある. これをオーバーフィッティングという. 先と同様, xi=i1,yi=sin(xi)+ϵ,i=1,2,,11x_i=i-1, y_i=\sin(x_i)+\epsilon, i=1,2,\cdots,11 に対する各次元での係数を見てみると((5)(5) では, 係数のベクトルを (a0,a1,,an)T(a_0,a_1,\cdots,a_n)^T と並べているが, 下記は (an,an1,,a0)T(a_n,a_{n-1},\cdots,a_0)^T の順である),

λ> :m +Data.Tuple.Extra Data.Maybe System.Random Random.Normal Control.Monad ML.Approx.OLS.ByPinv Utils
λ> let uni = normalIO' (0, 0.2)
λ> let d = zipWith (\x y -> (x, sin x+y)) [0..10] <$> replicateM 11 uni
λ> dd <- d
λ> dd
[(0.0,-5.563765361160251e-4),(1.0,0.9418472638241775),(2.0,1.1378051539092622),(3.0,0.30341406458452413),(4.0,-0.8411970236084821),(5.0,-0.8558604338359868),(6.0,-0.2586281201459223),(7.0,0.8031257237891795),(8.0,1.1562504257723663),(9.0,0.39633872602316167),(10.0,-0.8085898217611907)]
λ> let outCoes i = putStrLn $ maybe "failed" (((++) $ show i ++ " -----\n") . foldr1 (++) . (map ((++ "\n") . show))) $ resolve i dd
λ> mapM_ outCoes [1..10]
1 -----
[-4.168066672607503e-2]
[0.38785329563173637]

2 -----
[6.311957210780993e-3]
[-0.10480023883388496]
[0.4825326537934513]

(略)

9 -----
[6.2497127164502315e-6]
[-2.783924911591259e-4]
[5.143290075166487e-3]
[-5.026523522855991e-2]
[0.2724261548894942]
[-0.7770268952402895]
[1.0147233691594897]
[-0.7860963186273294]
[1.2602569068891822]
[-2.3687551597216985e-4]

10 -----
[-1.626701126479776e-5]
[8.196002759563384e-4]
[-1.7705498506683254e-2]
[0.21421719275219878]
[-1.5895590768542918]
[7.43609912277545]
[-21.670288785218535]
[37.428871273332284]
[-34.68074111840572]
[13.820707197220901]
[-5.563765361160251e-4]

m=10m=10 で, 21.670,37.428,34.680,13.820-21.670\cdots,37.428\cdots,-34.680\cdots,13.820\cdots といった, 絶対値の大きな値が見られる. 実際に m=9,10m=9, 10 でプロットしてみると,

λ> mapM_ (\i -> plot $ PP ("./image" ++ show i ++ ".png") ("m = " ++ show i) "points" "line" dd $ fromJust $ implicitFn $ fromJust $ resolve i dd) [9, 10]

m=10m=10 のモデルによる近似は m=9m=9 の場合と比べて激しく振れていることが見てとれる.

non regulared 1 non regulared 2

天下り的になってしまうが, このような現象は推定する係数に対して標本数が少ないようなときによく遭遇する. その特徴として, いま示したように, 係数の絶対値が大きくなることが挙げられる. 従って, 次数を適当に固定した上で(この場合 n=n=データ数1-1, すなわちデータ数から機械的に次数を決定する), 係数を絶対値を制限することができれば, これを防ぐことができるだろう. 具体的な手法として, 式 (4)(4) に対してノルムを加え, その最小化を求めるといったような10手法が広く知られている. この手法は, ノルムに対して, 次のように平滑化パラメータ λ0\lambda \geq 0 を作用させることで正則化の強度を設定することができる.

ϵ(a)λ=i=1m(yifn(xi))2+λR(a)正則化項\epsilon({\boldsymbol a})_\lambda=\sum^m_{i=1}(y_i-f_n({\boldsymbol x}_i))^2+\underbrace{\lambda R({\boldsymbol a})}_{\rm 正則化項}

このような最適化を正則化法という. こうすると, モデルの変動が大きくなるにつれて正則化項も大きくなり, それが最小化問題へのペナルティとなって, 結果的に滑らかな曲線の推定に繋がる. ただし, 過剰に大きいパラメータをとると, 高次の項へのペナルティが強くなってしまうことで, 結局, 高次の項を無視するのと同等となってしまい, 低次の関数でモデルを作るのと同等になってしまう. 従って, 依然として適切なパラメータの設定が要されるわけだが, モデルの次数を決定するよりかは楽である.

またパラメータ λ\lambda を標本数で割った形式が取られることもある.

ϵ(a)λ=i=1m(yifn(xi))2+λmR(a)正則化項\epsilon({\boldsymbol a})_\lambda=\sum^m_{i=1}(y_i-f_n({\boldsymbol x}_i))^2+\underbrace{\frac{\lambda}{m} R({\boldsymbol a})}_{\rm 正則化項}

両者の違いは正則化項の影響度である. 先に, 推定する係数に対して標本数が少ないようなときにオーバーフィッティングはよく起こると述べたが, ならば当然, 標本数が十分である場合には正則化項は必要ない. パラメータ λ\lambda を標本数で割ってやれば, 標本数の増加に従って正則化項の影響度を抑制できる. どちらを用いるかはその時々で選択の余地があるだろう.

例えば, λ\lambda を標本数に依らず直接作用させる形式で R(v)R({\boldsymbol v})L2L^2 ノルムとする11と,

ϵ(a)λ=i=1m(yifn(xi))2+λj=0naj2=(yXa)T(yXa)+λaTa \begin{aligned} \epsilon({\boldsymbol a})_\lambda&=&\sum^m_{i=1}(y_i-f_n({\boldsymbol x}_i))^2+\lambda\sum_{j=0}^{n}a^2_j \\ &=&({\boldsymbol y}-X{\boldsymbol a})^T({\boldsymbol y}-X{\boldsymbol a})+\lambda{\boldsymbol a}^T{\boldsymbol a} \end{aligned}

先と同様に ϵ(a)λ=0\nabla\epsilon({\boldsymbol a})_\lambda=0 とおいて,

ϵ(a)λ=2XT(yXa)+2λa=2(λI+XTX)a2XTy=0 \begin{aligned} \nabla\epsilon({\boldsymbol a})_\lambda&=&-2X^T({\boldsymbol y}-X{\boldsymbol a})+2\lambda{\boldsymbol a} \\ &=&2(\lambda I+X^T X){\boldsymbol a}-2X^T{\boldsymbol y}\\ &=&0 \end{aligned}

従ってこの正規方程式の解は,

a=(λI+XTX)1XTy{\boldsymbol a}=(\lambda I+X^T X)^{-1}X^T{\boldsymbol y}

となる. a{\boldsymbol a} を求めるに際する時間計算量について加味すると, 逆行列を計算するよりも LU 分解を行った方が良いので,

(λI+XTX)a=XTy(\lambda I+X^T X){\boldsymbol a}=X^T{\boldsymbol y}

としておく. この正規方程式を用いて, 平滑化パラメータ λ=0.1,1,10\lambda=0.1,1,10 を適用しプロットすると,

λ> mapM_ (\i -> plot $ PP ("./image" ++ show i ++ ".png") ("parameter = " ++ show i) "points" "line" dd $ fromJust $ implicitFn $ fromJust $ resolveRegular i dd) [0.1, 1, 10]

次のようになる.

regulared 0.1 regulared 1.0 regulared 10.0

問題は, どのようにしてオーバーフィッティングを評価するかである. データセット x\bf x に対し, 真の値 tit_iD={(x1,t1),(x2,t2),,(xm,tm)}where ti=g+ui (  (ii)より)D=\left\{({\bf x_1},t_1), ({\bf x_2},t_2),\cdots,({\bf x_m},t_m)\right\} {\rm where}\ t_i=g+u_i\ (\because\ \ {\text (ii) より}) とし, 回帰分析によって得られるモデル f^n(yi)=fn(xi) where xi=(yi0,yi1,,yin)T( (10)より)\hat{f}_n({\bf y_i})=f_n({\boldsymbol x'_i})\ {\rm where}\ {\boldsymbol x'_i}=({\bf y_i^0},{\bf y_i^1},\cdots,{\bf y_i^n})^T (\because \ {\text (10) より}) との差を次のように定義する.

L(ti,f^n(xi)):=(tif^n(xi))2L(t_i, \hat{f}_n({\boldsymbol {\bf x_i}})):=(t_i-\hat{f}_n({\boldsymbol {\bf x_i}}))^2

この LL は損失関数といわれる. ここで, xi\bf x_itit_i が得られる同時確率を考慮すると, 損失の期待値は

E[L(ti,f^n(xi))]=(tif^n(xi))2P(tixi)dtidxi={(tif^n(xi))2P(tixi)dti}P(xi)dxi (条件付き確率の乗法定理) \begin{aligned} E\left[L(t_i, \hat{f}_n({\bf x_i}))\right]&=& \int\int(t_i-\hat{f}_n({\bf x_i}))^2P(t_i\cap{\bf x_i})dt_id{\bf x_i} \\ &=&\int\left\{\int(t_i-\hat{f}_n({\bf x_i}))^2P(t_i\mid{\bf x_i})dt_i\right\}P({\bf x_i})d{\bf x_i}\ (\because \href{/roki.log/2018/10/28/probabilityTerms/#MulTheoremConditionalProbability}{\text 条件付き確率の乗法定理}) \end{aligned}

(tif^n(xi))2P(tixi)\int(t_i-\hat{f}_n({\bf x_i}))^2P(t_i|{\bf x_i}) を最小化したいので, これを g(y)=(tif^n(yi))2P(tiyi)g({\bf y})=\int(t_i-\hat{f}_n({\bf y_i}))^2P(t_i|{\bf y_i}) とおいて

f^n(xi)g(xi)=2(tif^n(xi))P(txi)dti=2{f^n(xi)P(tixi)}dti2tiP(tixi)dti=2f^n(xi)P(tixi)dti2tiP(tixi)dti=2f^n(xi)2tiP(tixi)dti (規格化条件) \begin{aligned} \frac{\partial}{\partial \hat{f}_n({\bf x_i})} g({\bf x_i})&=&2\int(t_i-\hat{f}_n({\bf x_i}))P(t\mid{\bf x_i})dt_i \\ &=&2\int\left\{\hat{f}_n({\bf x_i})P(t_i\mid{\bf x_i})\right\}dt_i-2\int t_i P(t_i\mid{\bf x_i})dt_i \\ &=&2\hat{f}_n({\bf x_i})\int P(t_i\mid{\bf x_i})dt_i-2\int t_i P(t_i\mid{\bf x_i})dt_i \\ &=&2\hat{f}_n({\bf x_i})-2\int t_i P(t_i\mid {\bf x_i})dt_i\ (\because \href{/roki.log/2018/10/28/probabilityTerms/#normalizationLaw}{\text 規格化条件}) \end{aligned}

\therefore

g(x)=0f^n(xi)=tiP(tixi)=E[tixi] \nabla g({\bf x})=0\leftrightarrow \hat{f}_n({\bf x_i})=\int t_i P(t_i\mid{\bf x_i})=E\left[t_i\mid{\bf x_i}\right]

よって, f^n\hat{f}_n は条件付き期待値 E[tixi]E\left[t_i\mid{\bf x_i}\right] で決めると最小化されることがわかった. 先に示した損失関数 LL は, 一つの点における差なので, すべての点における差を次のように定義する(英: mean-square error から).

MSE:=i=1mL(ti,f^n(xi)){\rm MSE}:=\sum^m_{i=1}L(t_i,\hat{f}_n({\bf x_i}))

この期待値をできる限り小さくしたい.

E[MSE]=E[i=1mL(ti,f^n(x))]=i=1mE[L(ti,f^n(x))] (期待値の線形性)E\left[{\rm MSE}\right]=E\left[\sum^m_{i=1}L(t_i,\hat{f}_n({\bf x}))\right]=\sum_{i=1}^mE\left[L(t_i,\hat{f}_n({\bf x}))\right]\ (\because \href{/roki.log/2018/10/28/probabilityTerms/#fn-2}{\text 期待値の線形性})

E[L(ti,f^n(xi))]E\left[L(t_i,\hat{f}_n({\bf x_i}))\right] について展開すると,

E[L(ti,f^n(xi))]=E[(tif^n(xi))2]=E[(tiE[tixi]+E[tixi]f^n(xi))2](augmentation trick)=E[{(tiE[tixi])a+(E[tixi]f^n(xi))b}2]=E[(tiE[tixi])2+(E[tixi]f^n(xi))2+2(tiE[tixi])(E[tixi]f^n(xi)) ]((a+b)2=a2+b2+2ab)=E[(tiE[tixi])2]+E[(E[tixi]f^n(xi))2]+E[2(tiE[tixi])(E[tixi]f^n(xi))](期待値の線形性)(12) \begin{aligned} E\left[L(t_i,\hat{f}_n({\bf x_i}))\right]&=& E\left[(t_i-\hat{f}_n({\bf x_i}))^2\right] \\ &=&E\left[(t_i-E\left[t_i\mid{\bf x_i}\right]+E\left[t_i\mid{\bf x_i}\right]-\hat{f}_n({\bf x_i}))^2\right] (\because {\rm augmentation\ trick})\\ &=&E\left[\left\{\underbrace{(t_i-E\left[t_i\mid{\bf x_i}\right])}_{a}+\underbrace{(E\left[t_i\mid{\bf x_i}\right]-\hat{f}_n({\bf x_i}))}_{b}\right\}^2\right] \\ &=&E\left[ \begin{array}{c} (t_i-E\left[t_i\mid{\bf x_i}\right])^2+ \\ (E\left[t_i\mid{\bf x_i}\right]-\hat{f}_n({\bf x_i}))^2+ \\ 2(t_i-E\left[t_i\mid{\bf x_i}\right])(E\left[t_i\mid{\bf x_i}\right]-\hat{f}_n({\bf x_i})) \end{array}\ \right] (\because (a+b)^2=a^2+b^2+2ab\tag{12}) \\ &=& \begin{array}{c} E\left[(t_i-E\left[t_i\mid{\bf x_i}\right])^2\right] \\ +E\left[(E\left[t_i\mid{\bf x_i}\right]-\hat{f}_n({\bf x_i}))^2\right] \\ +E\left[2(t_i-E\left[t_i\mid{\bf x_i}\right])(E\left[t_i\mid{\bf x_i}\right]-\hat{f}_n({\bf x_i}))\right] \end{array} (\because \href{/roki.log/2018/10/28/probabilityTerms/#fn-2}{\text 期待値の線形性}) \end{aligned}

第三項について

E[2(tiE[tixi])(E[tixi]f^n(x))]=2(E[tixi]f^n(x))E[(tig)] (E[aA]=aE[A])=2(E[tixi]f^n(x))(E[ti]E[E[tixi]]) (期待値の線形性)=2(E[tixi]f^n(x))(E[ti]E[ti]) (E[B]=E[E[BA]])=0 \begin{aligned} E\left[2(t_i-E\left[t_i\mid{\bf x_i}\right])(E\left[t_i\mid{\bf x_i}\right]-\hat{f}_n({\bf x}))\right]&=&2(E\left[t_i\mid{\bf x_i}\right]-\hat{f}_n({\bf x}))E\left[(t_i-g)\right]\ (\because \href{/roki.log/2018/10/28/probabilityTerms/#mjx-eqn-eq%3Aexaxiom3}{E\left[a A\right]=a E\left[A\right]}) \\ &=&2(E\left[t_i\mid{\bf x_i}\right]-\hat{f}_n({\bf x}))(E\left[t_i\right]-E\left[E\left[t_i\mid{\bf x_i}\right]\right])\ (\because \href{/roki.log/2018/10/28/probabilityTerms/#fn-2} {\text 期待値の線形性}) \\ &=&2(E\left[t_i\mid{\bf x_i}\right]-\hat{f}_n({\bf x}))(E\left[t_i\right]-E\left[t_i\right]) \ (\because \href{/roki.log/2018/10/28/probabilityTerms/#fn-4}{E\left[B\right]=E\left[E\left[B\mid A\right]\right]}) \\ &=&0 \end{aligned}