《高速微分法》

提供: ORWiki
ナビゲーションに移動 検索に移動

【こうそくびぶんほう (fast differentiation)】

 非線形関数の勾配, ヤコビ行列, ヘッセ行列等の値を数値的に計算する方法のひとつ. 高速自動微分法(fast automatic differentiation), 計算微分法(computational differentiation), 単純に自動微分(automatic differentiation; 以下 AD)ともいう. 主なアルゴリズムは2種あり, ボトムアップ(前進)自動微分(bottom-up AD, forward AD; 以下 BUAD) と, トップダウン(逆行)自動微分(top-down AD, reverse AD, backward AD; 以下 TDAD) という [1, 2]. 高速微分法は狭義には, TDADを指す. AD は「関数の値を計算するプログラム」から「偏導関数の値を計算するプログラム」を生成する手順を与え, 生成物を(コンパイルし)実行すれば, 差分商近似のような打ち切り誤差無しで, 正確な偏導関数の値を計算できる. 大規模システムの数学モデル等の大規模プログラム(数千行以上)により表現される関数の偏導関数を計算できるのが特長. 変数関数の勾配の個の値を関数計算の手間の定数倍で計算できる点が「高速」微分の由来である.

 以下,BUAD と TDAD による計算法を説明する.例として,2変数関数 について, の値を計算する代入文の列(プログラム), を考えよう. ただし, 各代入文の右辺には, 演算(基本演算とよぶ)が高々1回だけ現れるとする. , , に対応し, の値が計算される. 一般には, 変数関数 について, 回目の代入文には, 回目までに計算される変数が現れうるから, 延べ 回の演算を行なう代入文の列はと表される. これを計算過程といい, を中間変数という. のとき という入力定数の代入演算に相当する.

 BUADは, 補助変数 を導入し, 任意に を固定して, 合成関数の に関する偏微分則 に基づき, を計算する式を導出する. 基本演算 を四則演算や初等関数などの2項・単項の演算に限れば, 表1により, (これを要素的偏導関数という)を導出できる. , と初期設定すれば, について を計算済みとみなすことができ, の値を計算できる. 最終的に となる.

表1:基本演算と要素的偏導関数





 先の例では, に注意して, , , , , , という代入文の列を生成する. これを実行すると には の値が計算される( の計算の直後に を計算してもよい). 高々2項までの基本演算だけ使用するという条件の下では, BUADの手間は である. , と一部変更し, もう一度計算すれば, には, の値が計算される. 変数関数の勾配を計算するには, 同様の計算を 回繰り返す必要がある.

 TDADはこれとは異なり, 先の計算過程を と書き直し, これらを に関する制約式とみなす. この制約の下で, (の値) の停留点を考える. ラグランジュ関数 の停留点( かつ が成立する点)では, ラグランジュ乗数 は, 番目の制約式の摂動に対する関数値 の感度を与えるが, については に等しい. 入力 を定めると は一意に定まるが, は連立一次方程式を満たす. これを解くには, が実質的に単項・2項演算であることを考慮すると, と初期化しておき, の順に を計算する. 各 について高々2個の についてだけ計算すればよい.

 先の例では, を計算し, と初期化した後, , となる. 最終的に の値が計算される. 同じ条件の下で, TDADの手間は である. 1回の計算で勾配の値は全て計算できることに注意.

  変数 値関数 について, 全成分の値を計算するのに延べ 回の基本演算を実行したとする. ヤコビ行列 の列の線形結合はBUADで, 行についてはTDADで の手間で計算できる. 全成分については BUADでは , TDAD では である.

 実際には, 基本演算は表1に限らず, 代入文(やその列)を一つの基本演算とみなしてよい. また, プログラム中に条件分岐があっても, 与えられた入力値に関する関数の合成は上記の形で書けるから, ADを適用できる. ただし, 分岐の境目では, ADの結果は, 真の偏導関数値と異なることがある. たとえば, の様なプログラムを自動微分すると, の値が1.0 のときには不具合が起こりうるので注意が必要である.



参考文献

[1] M.Bucker, G.Corliss, P.Hovland, U.Naumann, and B.Norris (eds.), Automatic Differentiation: Applications, Theory, and Implementations, Lecture Notes in Computational Science and Engineering, Springer, Vol.50, 2006.

[2]久保田光一, 伊理正夫, 『アルゴリズムの自動微分と応用』, コロナ社, 1998.