自動微分

提供: miniwiki
移動先:案内、検索

自動微分（じどうびぶん、アルゴリズム的微分とも）とは、プログラムで定義された関数を解析し、偏導関数の値を計算するプログラムを導出する技術である。自動微分は複雑なプログラムであっても加減乗除などの基本的な算術演算や基本的な関数（指数関数・対数関数・三角関数など）のような基本的な演算の組み合わせで構成されていることを利用し、これらの演算に対して連鎖律を繰り返し適用することによって実現される。自動微分を用いることで偏導関数値を少ない計算量で自動的に求めることができる。

自動微分は

ファイル:AutomaticDifferentiationNutshell.png
図1: 自動微分と数式微分の関係

のどちらとも異なる。

数式微分は効率が悪くなりやすく、プログラムで定義された関数から微分表現を導くのは困難であるという問題がある。一方、数値微分では離散化の際の丸め誤差や桁落ちによる精度の低下が問題である。さらに、どちらの手法も計算量や誤差の関係で高次の微分係数を求めることが難しい。また、勾配を用いた最適化で必要となる、多くの入力変数を持つ関数に対する偏微分値の計算を行うには速度が遅い。自動微分はこれらの古典的手法の問題を解決する。

英語では Automatic differentiation (あるいは単に AD) と呼ばれる。

自動微分の導出

自動微分の基本原理は連鎖律を用いた微分の分解である。たとえば、簡単な合成関数である y = g(h(x)) = g(w) に対して連鎖律を適用すると

[math]\frac{dy}{dx} = \frac{dy}{dw} \frac{dw}{dx}[/math]

を得る。

自動微分は大きく2種類に分けられ、それぞれボトムアップ型自動微分（フォーワードモード、狭義の自動微分）とトップダウン型自動微分（リバースモード、高速自動微分）と呼ばれる。 ボトムアップ型自動微分では連鎖律を内側から外側に計算し（dw/dxを計算した後で dy/dw を計算する）、トップダウン型自動微分では外側から内側に計算する。

ボトムアップ型自動微分

ファイル:ForwardAccumulationAutomaticDifferentiation.png
図2: ボトムアップ型自動微分の計算グラフの例

ボトムアップ型自動微分では最初に微分を行う入力変数を固定し、それぞれの部分式を再帰的に計算する。手計算においては連鎖律において内側の関数を繰り返し置き換えていくことに相当する。

[math]\frac{\partial y}{\partial x} = \frac{\partial y}{\partial w_1} \frac{\partial w_1}{\partial x} = \frac{\partial y}{\partial w_1} \left(\frac{\partial w_1}{\partial w_2} \frac{\partial w_2}{\partial x}\right) = \frac{\partial y}{\partial w_1} \left(\frac{\partial w_1}{\partial w_2} \left(\frac{\partial w_2}{\partial w_3} \frac{\partial w_3}{\partial x}\right)\right) = \cdots[/math]

多変数の場合はヤコビ行列の積として一般化できる。

トップダウン型自動微分と比較すると、ボトムアップ型自動微分は自然であり、微分に関する情報の流れが計算の順序と一致するため簡単に実行できる。それぞれの変数にその偏導関数値ẇ

[math]\dot w = \frac{\partial w}{\partial x}[/math]

の計算値を保存する領域を付け加えるだけで、変数値の計算と同時に偏導関数値を計算することができる。

例として次の関数を考える。

[math]\begin{align} z &= f(x_1, x_2) \\ &= x_1 x_2 + \sin x_1 \\ &= w_1 w_2 + \sin w_1 \\ &= w_3 + w_4 \\ &= w_5 \end{align}[/math]

簡単のためそれぞれの部分式を wi としてラベル付けした。

どの入力変数で微分するかによってẇ1 や ẇ2の初期値が異なる。x1 に関する微分をそのつど計算する場合の初期値は、

[math]\begin{align} \dot w_1 = \frac{\partial x_1}{\partial x_1} = 1 \\ \dot w_2 = \frac{\partial x_2}{\partial x_1} = 0 \end{align}[/math]

となる。初期値が決まったら下の表に示す連鎖律に従って各偏導関数値を順に計算していく。図2はこの処理を計算グラフとして表している。

[math]\begin{array}{l|l} \text{Operations to compute value} & \text{Operations to compute derivative} \\ \hline w_1 = x_1 & \dot w_1 = 1 \text{ (seed)} \\ w_2 = x_2 & \dot w_2 = 0 \text{ (seed)} \\ w_3 = w_1 \cdot w_2 & \dot w_3 = w_2 \cdot \dot w_1 + w_1 \cdot \dot w_2 \\ w_4 = \sin w_1 & \dot w_4 = \cos w_1 \cdot \dot w_1 \\ w_5 = w_3 + w_4 & \dot w_5 = \dot w_3 + \dot w_4 \end{array}[/math]

この関数に対する勾配を求めるためにはx1だけではなくx2に関する偏導関数値も求める必要がある。そのため、初期値を[math]\dot w_1 = 0; \dot w_2 = 1[/math] とした同様の計算を追加で行わなければならない。

ボトムアップ型自動微分を1回行うのに必要な計算量は、元のプログラムの計算量に比例する。

勾配を求める場合に必要なボトムアップ型自動微分の実行回数は入力変数の個数と等しく、トップダウン型自動微分では出力変数の個数に等しい。そのため、微分する関数f : ℝn → ℝm が m ≫ n を満たす場合、ボトムアップ型自動微分はトップダウン型自動微分よりも効率的である。

トップダウン型自動微分

ファイル:ReverseaccumulationAD.png
図3: トップダウン型自動微分の計算グラフの例

トップダウン型自動微分では、はじめに微分される出力変数を固定して、それぞれの部分式に関する偏導関数値を再帰的に計算する。手計算においては部分式を連鎖律における外側の関数の微分で繰り返し置き換えていくことに相当する。

[math]\frac{\partial y}{\partial x} = \frac{\partial y}{\partial w_1} \frac{\partial w_1}{\partial x} = \left(\frac{\partial y}{\partial w_2} \frac{\partial w_2}{\partial w_1}\right) \frac{\partial w_1}{\partial x} = \left(\left(\frac{\partial y}{\partial w_3} \frac{\partial w_3}{\partial w_2}\right) \frac{\partial w_2}{\partial w_1}\right) \frac{\partial w_1}{\partial x} = \cdots[/math]

トップダウン型自動微分において、求める値は随伴でありテンプレート:訳語疑問点、上線(w̄)で表す。これは選択された出力変数のwに関する微分

[math]\bar w = \frac{\partial y}{\partial w}[/math]

である。

トップダウン型自動微分は連鎖律を外側から内側（図3の計算グラフでは上から下）にたどっていく。例示した関数は1変数関数なので、勾配を計算するには計算グラフを一度たどるだけでよい。これは2回計算グラフをたどる必要があったボトムアップ型自動微分の計算量の半分である。しかし、トップダウン型自動微分は中間変数wi を保存するための領域と、1964年に R.E. Wengert が発表した Wengert リスト（テープとも呼ばれる）[1][2] というデータ構造に中間変数を保存するための命令が必要となるためメモリ使用量の点で不利であり、計算グラフが巨大になるとメモリ使用量が問題となる可能性がある。この問題は保存する中間変数を限定し、必要な中間変数を再計算することによって軽減される。

トップダウン型自動微分を用いて偏導関数値を計算するための演算は以下の通りである（関数値を求める時と順番が逆であることに注意）

[math]\begin{array}{l} \text{Operations to compute derivative} \\ \hline \bar w_5 = 1 \text{ (seed)} \\ \bar w_4 = \bar w_5 \\ \bar w_3 = \bar w_5 \\ \bar w_2 = \bar w_3 \cdot w_1 \\ \bar w_1 = \bar w_3 \cdot w_2 + \bar w_4 \cdot \cos w_1 \end{array} [/math]

微分する関数f : ℝn → ℝm が m ≪ nを満たすとき、トップダウン型自動微分はボトムアップ型自動微分よりも効率的である。

機械学習で用いられる多層パーセプトロンのバックプロパゲーション はトップダウン型自動微分の特殊なケースである。

二重数を用いた自動微分

フォワードモード自動微分は実数の代数に（元を）添加して新しい算術を導入することによって可能である。全ての数（通常の実数）に対して、その数における関数の微分を表現する追加の成分が足され、全ての算術演算がこの添加代数に拡張される。すなわち二重数の代数である。このアプローチはoperational calculus on programming spaces（English版）の理論（つまり双対空間のテンソル代数）によって一般化される（Analytic programming space（English版）を見よ)。

各数 [math]x[/math] を数 [math]x + x'\varepsilon[/math] に置き換える。ここで [math]x'[/math] は実数だが、[math]\varepsilon[/math] は [math]\varepsilon^2=0[/math] を満たす抽象的数（English版）である（無限小；滑らかな無限小解析（English版）を参照）。ちょうどこれだけを用いて通常の演算が得られる：

[math]\begin{align} (x + x'\varepsilon) + (y + y'\varepsilon) &= x + y + (x' + y')\varepsilon \\ (x + x'\varepsilon) \cdot (y + y'\varepsilon) &= xy + xy'\varepsilon + yx'\varepsilon + x'y'\varepsilon^2 = xy + (x y' + yx')\varepsilon \end{align}[/math]

引き算と割り算についても同様である。

いまやこの拡張算術のもとで多項式を計算できる。もし [math]P(x) = p_0 + p_1 x + p_2x^2 + \cdots + p_n x^n[/math] ならば、

[math]\begin{align} P(x + x'\varepsilon) &= p_0 + p_1(x + x'\varepsilon) + \cdots + p_n (x + x'\varepsilon)^n \\ &= p_0 + p_1 x + \cdots + p_n x^n + p_1x'\varepsilon + 2p_2xx'\varepsilon + \cdots + np_n x^{n-1} x'\varepsilon \\ &= P(x) + P^{(1)}(x)x'\varepsilon \end{align}[/math]

ここで [math]P^{(1)}[/math] は [math]P[/math] の最初の引数に対する微分であり、[math]x'[/math]（種と呼ばれる）は任意に取れる。

上に述べたように、この新しい算術は、順序対（[math]\langle x, x' \rangle[/math] と書かれる元）と、最初の成分に対しては通常の算術を、第二の成分に対しては一階微分の算術を、それぞれ与えたものからなる。多項式に関する上の結果を解析関数に広げれば、新しい算術に対する、基本的な算術と幾つかの標準的な関数のリストが得られる：

[math]\begin{align} \left\langle u,u'\right\rangle + \left\langle v,v'\right\rangle &= \left\langle u + v, u' + v' \right\rangle \\ \left\langle u,u'\right\rangle - \left\langle v,v'\right\rangle &= \left\langle u - v, u' - v' \right\rangle \\ \left\langle u,u'\right\rangle * \left\langle v,v'\right\rangle &= \left\langle u v, u'v + uv' \right\rangle \\ \left\langle u,u'\right\rangle / \left\langle v,v'\right\rangle &= \left\langle \frac{u}{v}, \frac{u'v - uv'}{v^2} \right\rangle \quad ( v\ne 0) \\ \sin\left\langle u,u'\right\rangle &= \left\langle \sin(u) , u' \cos(u) \right\rangle \\ \cos\left\langle u,u'\right\rangle &= \left\langle \cos(u) , -u' \sin(u) \right\rangle \\ \exp\left\langle u,u'\right\rangle &= \left\langle \exp u , u' \exp u \right\rangle \\ \log\left\langle u,u'\right\rangle &= \left\langle \log(u) , u'/u \right\rangle \quad (u\gt 0) \\ \left\langle u,u'\right\rangle^k &= \left\langle u^k , k u^{k - 1} u' \right\rangle \quad (u \ne 0) \\ \left| \left\langle u,u'\right\rangle \right| &= \left\langle \left| u \right| , u' \mbox{sign} u \right\rangle \quad (u \ne 0) \end{align}[/math]

一般に、プリミティヴの関数 [math]g[/math] に対して、

[math]g(\langle u,u' \rangle , \langle v,v' \rangle ) = \langle g(u,v) , g_u(u,v) u' + g_v(u,v) v' \rangle[/math]

ここで [math]g_u[/math] と [math]g_v[/math] はそれぞれ [math]g[/math] の最初と2番目の引数に対する微分である。

基本的な二項算術演算を（実数と二重数の）混在した引数に対して、つまり順序対 [math]\langle u, u' \rangle[/math] と実数 [math]c[/math] に対して適用するとき、まずこの実数を [math]\langle c, 0 \rangle[/math] に引き上げる（lifting）。関数 [math]f : \mathbb{R}\rightarrow\mathbb{R}[/math] の [math]x_0[/math] に於ける微分はいまや [math]f(\langle x_0, 1 \rangle)[/math] に上の算術を使って計算することによって得られる。これは [math]\langle f ( x_0 ) , f' ( x_0 ) \rangle [/math] を結果として与える。

ベクトル引数と関数

多変数関数は、方向微分作用素を用いることで、一変数関数の場合と同様の効率と仕組みで取り扱える。つまり、[math]f:\mathbb{R}^n\rightarrow\mathbb{R}^m[/math] の [math]x \in \mathbb{R}^n[/math] における [math]x' \in \mathbb{R}^n[/math] 方向微分 [math]y' = \nabla f(x)\cdot x'[/math] を計算したい場合、それは [math](\langle y_1,y'_1\rangle, \ldots, \langle y_m,y'_m\rangle) = f(\langle x_1,x'_1\rangle, \ldots, \langle x_n,x'_n\rangle)[/math] を上と同様の算術を使って計算すればよい。もし [math]\nabla f[/math] の全ての要素が望みならば、[math]n[/math] 個の関数評価が要求される。ここで、多くの最適化アプリケーションでは、実際には方向微分があれば十分である。

高階・多変数

上の算術は多変数関数の二階やもっと高階の微分の計算の為に一般化出来る。しかし、その算術規則は直ちに極めて複雑なものとなる：複雑性は最高次の微分の次数に対して二次関数的となる。その代わりに、途中で打ち切った（truncated）テイラー多項式の代数を使用できる。結果得られる算術（一般化された二重数の上で定義された）は、関数を新しいデータ型であるかのように使って、効率的に計算することを可能にする。ひとたび関数のテイラー多項式が分かれば、その導関数たちは容易に抽出できる。 厳密で一般的な定式化はテンソル級数展開（English版）を通してプログラミング空間上の演算子法（English版）を用いることにより達成される。

実装

自動微分の実装方法には大きく分けて、ソースコードの変換とオペレータオーバーローディングによる方法の2つがある。

ソースコード変換

ファイル:SourceTransformationAutomaticDifferentiation.png
図4: ソースコード変換の動作例

関数値を求める関数を記述した元のソースコードから、偏導関数値を計算する処理を含んだプログラムを自動的に生成する手法である。 ソースコード変換はあらゆるプログラミング言語で実装でき、コンパイル時の最適化を行いやすいが、自動微分ツールの作成は難しい。

オペレータオーバーローディング

ファイル:OperatorOverloadingAutomaticDifferentiation.png
図5: オペレータオーバーローディングの動作例

この手法は演算子のオーバーロードがサポートされているプログラミング言語で記述されたソースコードに対してのみ適用可能である。元のソースコードの流れを大きく変更することなく実現できるが、基本データ型の変更などの小さな変更は必要である。

ボトムアップ型自動微分をオペレータオーバーロードで実現するのは容易である。トップダウン型自動微分についても可能であるが、現状のコンパイラではボトムアップ型自動微分と比べると最適化の面で不利である。

ソフトウェア

自動微分を実装したライブラリなどのソフトウェアが存在する。詳細は英語版を参照のこと。

脚注

  1. ↑ R.E. Wengert (1964). “A simple automatic derivative evaluation program”. Comm. ACM 7: 463–464. doi:10.1145/355586.364791. 
  2. ↑ Bartholomew-Biggs, Michael; Brown, Steven; Christianson, Bruce; Dixon, Laurence (2000). “Automatic differentiation of algorithms”. Journal of Computational and Applied Mathematics 124 (1-2): 171-190. doi:10.1016/S0377-0427(00)00422-2. http://ac.els-cdn.com/S0377042700004222/1-s2.0-S0377042700004222-main.pdf?_tid=61070b3c-31f1-11e5-a550-00000aacb35f&acdnat=1437735029_c639352923bb07e65307eb75c420d42f. 

参考文献

  • (1998) アルゴリズムの自動微分と応用, 現代非線形科学シリーズ. コロナ社. ISBN 978-4339026023. 
  • (1991), 伊理 正夫, 久保田,光一, "高速自動微分法(I)", 応用数理 1(1), 17-35, 19910315, 日本応用数理学会.

外部リンク



「http:///http:///mymemo.xyz/mymemo.xyz/wiki/index.php?title=自動微分&oldid=111493」から取得