arXiv reaDer
堅牢な微分可能SVD
Robust Differentiable SVD
対称行列の固有分解は、多くのコンピュータービジョンアルゴリズムの中心です。ただし、固有ベクトルの導関数は、SVDを使用して解析的に計算する場合でも、べき乗法(PI)を使用して近似する場合でも、数値的に不安定になる傾向があります。この不安定性は、互いに近い固有値が存在する場合に発生します。これにより、固有分解を深いネットワークに統合することが困難になり、特に大きな行列を処理する場合に、収束が不十分になることがよくあります。これは、データを小さな任意のグループに分割することで軽減できますが、理論的な根拠がなく、固有分解の全能力を活用することは不可能です。以前の作業では、フォワードパス中にSVDを使用し、バックワードパス中に勾配を計算するためにPIを使用して、これを軽減しました。ただし、PIを使用して複数の固有ベクトルを計算するために必要な反復収縮手順は、エラーを蓄積し、不正確な勾配を生成する傾向があります。ここでは、SVD勾配のテイラー展開が、実際には反復プロセスに依存せずにPIを使用して得られた勾配と理論的に同等であり、したがってより正確な勾配が得られることを示します。画像分類とスタイル転送の精度が向上することの利点を示します。
Eigendecomposition of symmetric matrices is at the heart of many computer vision algorithms. However, the derivatives of the eigenvectors tend to be numerically unstable, whether using the SVD to compute them analytically or using the Power Iteration (PI) method to approximate them. This instability arises in the presence of eigenvalues that are close to each other. This makes integrating eigendecomposition into deep networks difficult and often results in poor convergence, particularly when dealing with large matrices. While this can be mitigated by partitioning the data into small arbitrary groups, doing so has no theoretical basis and makes it impossible to exploit the full power of eigendecomposition. In previous work, we mitigated this using SVD during the forward pass and PI to compute the gradients during the backward pass. However, the iterative deflation procedure required to compute multiple eigenvectors using PI tends to accumulate errors and yield inaccurate gradients. Here, we show that the Taylor expansion of the SVD gradient is theoretically equivalent to the gradient obtained using PI without relying in practice on an iterative process and thus yields more accurate gradients. We demonstrate the benefits of this increased accuracy for image classification and style transfer.
updated: Thu Apr 08 2021 15:04:15 GMT+0000 (UTC)
published: Thu Apr 08 2021 15:04:15 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト