現代の複雑なニューラルネットワークが記録を破り、より困難な問題を解決し続けるにつれて、それらの予測もますます理解しにくくなります。現在の解釈可能性の欠如は、機密性の高い設定での正確な機械学習ツールの展開を損なうことがよくあります。この作業では、現在のアプローチの制限のいくつかを解決する、シャープレイ係数の階層的拡張(Hierarchical Shap(h-Shap))に基づく画像分類のモデルにとらわれない説明方法を提示します。他のShapleyベースの説明方法とは異なり、h-Shapはスケーラブルであり、近似を必要とせずに計算できます。複数インスタンスの学習で一般的なものなど、特定の分布の仮定の下で、h-Shapは、計算の複雑さを指数関数的に改善して、正確なシャープレイ係数を取得します。階層的アプローチを、合成データセット、医用画像シナリオ、および一般的なコンピュータービジョンの問題について、一般的なShapleyベースおよび非Shapleyベースの方法と比較し、h-Shapが精度と実行時間の両方で最先端を上回っていることを示します。コードと実験は公開されています。
As modern complex neural networks keep breaking records and solving harder problems, their predictions also become less and less intelligible. The current lack of interpretability often undermines the deployment of accurate machine learning tools in sensitive settings. In this work, we present a model-agnostic explanation method for image classification based on a hierarchical extension of Shapley coefficients--Hierarchical Shap (h-Shap)--that resolves some of the limitations of current approaches. Unlike other Shapley-based explanation methods, h-Shap is scalable and can be computed without the need of approximation. Under certain distributional assumptions, such as those common in multiple instance learning, h-Shap retrieves the exact Shapley coefficients with an exponential improvement in computational complexity. We compare our hierarchical approach with popular Shapley-based and non-Shapley-based methods on a synthetic dataset, a medical imaging scenario, and a general computer vision problem, showing that h-Shap outperforms the state of the art in both accuracy and runtime. Code and experiments are made publicly available.