バランスヒューリスティックの導出
この記事はNEEとMISの実装 (MIS編) - MochiMochi3DのおまけでMISのバランスヒューリスティックの導出はどうやっているのかの流れを書いたものです。
MISの理論的な話
MISによるレンダリング結果を見れば大幅に分散(ノイズ)が減っていることが見てわかると思います。実用的にはバランスヒューリスティックはとても良い手法であることはわかりましたが、どうやってバランスヒューリスティックなんてものが出てきたのかということが疑問に浮かびます。また、これより最適な方法はないのかということも頭に浮かびます。
MISの製作者であるVeachという方の論文[Multiple Importance Sampling]を見るとちゃんとその辺のことが書いてありました。この記事ではこの論文にあった証明を書いていこうと思います。ただし自分の解釈で書いていくため、間違いがある可能性があることに留意してください。
multi-sapmle Model
今、ある関数についてある積分範囲で積分値を調べるとする。その積分は
と表します。これに対してモンテカルロ積分をして数値積分することを考えます。
に対してn個のサンプリング戦略があるとします。通常のモンテカルロ積分であれば、これらから1つだけ選んでサンプリングを行いますが一般的には1つのサンプリング戦略がすべての場合で効率的であるとは限りません。場合によっては他のサンプリング戦略がいいこともありますし、同等の値を出すような戦略も存在します。
なので、1つのサンプリング方法を使い続けるのではなく、うまいこと効率のいいサンプリング戦略をn個の中から必ずとれるようにしたいことになります。そこで、「n個のサンプリング戦略を別々に行って、その結果をうまく合わせれば弱点を補えるのではないか?」というアイディアによって生まれたのがMultiple Importance Samplingとなります。
Veachの論文ではいくつかサンプリングした結果をウェイトをかけて総和を取るという形でサンプリング戦略の合成考えています。このモデルはmulti-sample Modelと呼ばれています。
に対してサンプリング戦略が個存在しているものとして、multi-sample Modelは
と表されます。各文字について
: i番目サンプリング戦略でのサンプル数,
: i番目サンプリング戦略でj番目のサンプル(乱数)
: i番目サンプリング戦略におけるウェイト関数
: i番目サンプリング戦略におけるPDF
これが意味する所はそれぞれのサンプリング戦略で別々に回サンプリングを行い、その平均を合わせるという形で求めるということである。この際、この合成がうまいこと行くように調整項としてウェイト関数をサンプリングごとに入ってきます。
ウェイト関数は当然何でもいいというわけではなく、条件として
を満たす必要があります。こうなる理由は次に説明します。
こうしてアイディアであった複数のサンプリングを使ったモンテカルロ積分を考えることができました。これでうまいこと行くようなウェイト関数を作れば複数のサンプリング戦略を混ぜたMultiple Importance Samplingができるというわけです。
multi-sample Modelの収束値
こうした複数サンプリングのモデルとしてmulti-sample Modelが作られましたが、これが実際に通常のモンテカルロ積分のように積分値にちゃんと収束しなければ意味がありません。このモデルの期待値について求めると
となります。この値は(1)式に等しくならなくてはならないため、ウェイトの条件として
得られるわけです。なのでこの条件下のウェイトであれば、multi-sample Modelはちゃんと積分値へと収束することが求められます。
バランスヒューリスティックの導出
MISのウェイト関数としてバランスヒューリスティックというものがあります。
なんとなく納得できる形ではありますがこれがどこから来たかというと、分散が最も小さくなるようなウェイトの取り方を導出するとこれが得られます(厳密にはさらに分散を小さくするウェイトの取り方がある可能性はあります)。
multi-sample Modelの分散を求める
まず、簡単のためにiサンプリング戦略におけるj番目のサンプリングをと表すことにします。
この時、iサンプリング戦略の収束値は
となります。
次にmulti-sample Modelの分散について求めます。multi-sample Modelは
となっています。これの分散は
となります。定数であるを外に出し(2乗になる)、分散の線形性(サンプル[tex:F_{i,j}がそれぞれ独立なので)から
分散と期待値の関係性から、期待値に書き換えると
を展開して表すと
第一項目について総和を取って、積分の中に入れると
と求めることができます。これがmulti-sample Modelの分散になり、これをいかに小さくするかが問題となっていきます。この時、色々と変数はありますが結局のところ私たちが無理やり値を変えていいのはウェイト関数だけです。なので、分散を最小化するようなウェイト関数について考えていけばいいのではないかということになります。
第一項目の最小化
まず、第一項目について最小化を考えていきます。
この値は積分値で一見最小値をどうすればいいかがわからないと思います。この式は積分であるため、積分値は結局のところ各点で
という値を求めて、その総和を取っていくという話になります。仕方がって、各点において上の値が各々最小になることができれば積分値(7)というのも最小になるはずです。 ということで、任意の点において被積分関数が最小の値を取るようなウェイト関数を求めていく必要があります。これについてはを固定して、その時のウェイト関数がどのようになるかを調べていけばいいというわけです。を固定したものとして考えると、は定数となり、を除くと、
という式になります。これについて最小化していけば(7)式が最小化されることになります。
これを最小にするウェイト関数について求めることになるのですが、これには式(3)という条件が付きます。こうした束縛条件が存在する時の最小化はラグランジュの未定乗数法を用いることで行えます。 ja.wikipedia.org
束縛条件(3)と変数を用いて、
という式を考えます。ラグランジュの未定乗数法によって、これらは各偏微分が0になります。そのため、偏微分すると
と各ウェイト関数との関係式が求められます。よって束縛条件を含めるとn+1個の方程式
が求められます。この時、ウェイト関数の方程式を束縛条件に代入すると
が得られます。これを代入すると
バランスーヒューリスティックが求められます。すなわち、バランスーヒューリスティックとは式(7)を最小化するウェイト関数なのであります。
第二項目の最小化
第二項目について
こちらも最小化を考えてみます。こちらの場合、直接ウェイト関数が出てこないためについて求めていきます(ウェイト関数はすでに求めているじゃんと思われるかもしれませんが一旦それは置いておきます)。 については束縛条件として収束値との関係式
があります。なので同様にラグランジュの未定乗数で最適化することができます。
より、
が求められます。従って最小値は
となります。
バランスヒューリスティックは最適か?
バランスヒューリスティックはあくまで第一項目の最適化を目的としており、第二項目に関しては最小化することは考慮されていません。そのため、バランスヒューリスティックでは第二項目が最小化されておらず、完全に最適ではない可能性が存在しています。
実際に最小化されていないことを数式として求めることはとの関係は積分の関係にあることから非常に難しいです。第二項目の最小化は現実的には特殊な状態が必要なため、バランスヒューリスティックでは第二項目が最小化されないものとして考えても問題はないと考えられます。
そうなると第二項目の非最小化による分散の影響を考えなくてはなりません。もしかしたら、第一項目より第二項目のほうが分散の影響が強く、こっちを最適化しなくてはならないのではないかということもあり得ます。
第二項目の範囲
こうしたことを考えるため、第二項目の上限を調べます。の内、最も小さい値だけでn個だけ総和を取った値を
と表すこととします。この時、
という不等式が成り立ちます。これを(13)式に代入すると
この時、さらに
と求められるので、
つまり、第二項目については
という範囲にあることが求められます。
理想との分散の差
バランスヒューリスティックは完璧な最適化をすることができません。もし、仮に完璧な最適化ができるEstimater が存在するとして、パワーヒューリスティックによるEstimater と分散で差を取るとします。
この値はどれだけバランスヒューリスティックが最適化から離れているかを示すことになります。
バランスヒューリスティックは第一項目を最適化していることは求めた通りであり、これ以上の最適化は存在しません。従って、この二つの分散での第一項目は等しい値となります。なので、この差は第二項目の差となります。
当然、完璧な最適化をしているは第二項目も最適化をしています。すなわち第二項目は最小値(15)を取っていることになります。一方で、は第二項目はどの値を取っているかはわかりません。しかしながら、第二項目の範囲は式(17)に書かれている通りであり、最大でも(16)の上限が存在しています。
なのでこの分散の差について不等式として
という形で得ることができます。は各サンプルの回数であり、実用上ではそこまでサンプリングしなくても各分母が100とか1000といったぐらいの大きさになります。また、それぞれ極端に違うということはなかなかありません。そうなるとカッコ内の値というのは結構小さいものとなります(大体0.001とかそういうレベル)。従って実用上ではこの値というのはそこまで問題になるほどの値ではないと思われます。
こうしたことから「バランスヒューリスティックはある程度最適解から離れているかもしれないけど、その差は精々(18)程度であり、かなり最適解に近い分散を持っている」ことが式(18)の意味となります。
one-sample Modelでの分散
Veachの論文では確率的にサンプリング戦略を選択して1サンプルを行うone-sample Modelについても記述しています。こちらについても同様にバランスヒューリスティックを考えることができます。
実はone-sample Modelにおいてバランスヒューリスティックはmulti-sample Modelとは異なり、最も分散が小さくなるウェイト関数であることが証明されているらしいです。(証明についてはちょっと理解できてないです)
MISはどんなシーンでも早くなるのか?
(ちょっとここら辺は自信がないです 飛ばしても大丈夫です)
MISの画像とPathTracer単体、NEE単体のものと比べるとノイズは全然なく、結構強力な手法なんだとわかりました。なので何でもかんでもMIS使えばいいんじゃないみたいなことを考えたのですが、ほんとにMISってどんなシーンでも分散を減らしてくれるような万能なんだろうかっていう疑問がありました。今回の式で折角分散がわかったので1つのサンプリング戦略しか使わないサンプリング(PathtracerとかNEEとか)との分散の比較ができるんじゃないかってことで一応考えてみたことを書いてみます。
ある1つのサンプリング戦略iのみを使うEstimaterをとします。サンプル数をとして、の分散は
という形になります。(PathTracerとかの分散でも見られるやつですね)
multi-sample Modelについて、としてほかのウェイト関数というウェイト関数を考えた際、この時のmulti-sample Modelはになるはずです。なのでは一つのmulti-sample Modelとしても考えられるはずです。従って式(19)も今まで考えていた第一項目、第二項目として分離できるはずです。
バランスヒューリスティックとの比較
まず、第二項目についてですがこれは(15)の分母部分がサンプル数の総和、つまり全サンプル数を意味しているため、(19)式の第二項目は(15)式と一致しており最適化されていることがわかります。従って第二項目を注目すればバランスヒューリスティックに勝っていることになりますが、その差というのは式(18)で表したように微小な値であるのであまり問題はありません。
結局のところ、大きな比較となる部分が第一項目となります。しかしながら、バランスヒューリスティックはこの第一項目を最適化したもののため、Estimaterは第一項目ではバランスヒューリスティックに同等あるいはそれより大きくなってしまうほかありません。その差についてはが未知の関数である以上求められませんが、サンプリング戦略が変わると分散も大きく変わるため、おそらく大きな差となるはずです。
従って、1つのみのサンプリング戦略を使ったEstimaterは数量的に求めることはできませんがバランスヒューリスティックに打ち勝つことはほとんどないのだと考えられます。
しかしながら、すごい特殊な状況下ではバランスヒューリスティックのウェイト関数が丁度Estimaterと等しくなる場合があります。
例えば、理想鏡面しか存在しないシーンがあった場合、PathTracerは完璧なサンプリング戦略となります。一方で、NEEなどのほかのサンプリング戦略をしたとしてもそもそもPDFがになるため、ウェイト関数がPathTracerのやつだけが1になり他の戦略ではすべて0となります。これはすなわちPathTracerのみを取ったEstimaterに相当することとなります。なので、結果としてEstimaterはバランスヒューリスティックを超えた、このシーンにおける最適解であることになります。
結論
結論としては、「パワーヒューリスティックEstimaterは現実的にはより分散が少ない。(ただし、コーナーケースとしてのほうが良い場合もありうる)」と考えられます。長々と話しましたが結局MISって全部のサンプリング戦略やるんだから絶対ノイズ減るじゃんっていう話です。
ただし、これはあくまでサンプル数が同じ場合における話です。現実的にはMISはサンプリング回数が多いため、実行時間が長く、同じ実行時間で見ればほかの方法がサンプル数を回せてノイズを少なく出せるということもあり得ます。でもまあ、同じ時間でもMISのほうが明らかにノイズが小さくなり、また色々とサンプリング戦略の弱点をカバーできるので不自然な点もなくなりやすいです。事実上はMISを使うことが最も良いことになるのだと思います。
終わりに
以上のようにバランスヒューリスティックはこのような感じで求められることがわかりました(多分これで合ってるはずです)。どっからこんな式が出てきたのかと思いちゃんと調べてみるとちゃんと理論的に数式をがりがりといじることで出てくるようなものであったことが知れたのは結構面白かったです。
参考文献
https://graphics.stanford.edu/courses/cs348b-03/papers/veach-chapter9.pdf