arXiv reaDer
解釈可能性のための神経基盤モデル
Neural Basis Models for Interpretability
実世界のアプリケーションで複雑な機械学習モデルが広く使用されているため、モデルの予測を説明することが重要になっています。ただし、これらのモデルは通常、ブラックボックスのディープニューラルネットワークであり、既知の忠実度の制限があるメソッドを介して事後的に説明されます。一般化加法モデル(GAM)は、本質的に解釈可能なクラスのモデルであり、各フィーチャの非線形形状関数を個別に学習し、その後に線形モデルを追加することで、この制限に対処します。ただし、これらのモデルは通常、トレーニングが難しく、多数のパラメーターが必要であり、スケーリングが困難です。形状関数の基底分解を利用するGAMのまったく新しいサブファミリーを提案します。少数の基底関数がすべての機能間で共有され、特定のタスクに対して共同で学習されるため、特に機能がスパースである場合に、モデルのスケールが高次元の機能を備えた大規模データにはるかに向上します。単一のニューラルネットワークを使用してこれらのベースを学習するニューラルベースモデル(NBM)と呼ばれるアーキテクチャを提案します。さまざまな表形式および画像データセットで、解釈可能な機械学習の場合、NBMは精度、モデルサイズ、スループットの点で最先端であり、すべての高次機能の相互作用を簡単にモデル化できることを示しています。ソースコードはhttps://github.com/facebookresearch/nbm-spamで入手できます。
Due to the widespread use of complex machine learning models in real-world applications, it is becoming critical to explain model predictions. However, these models are typically black-box deep neural networks, explained post-hoc via methods with known faithfulness limitations. Generalized Additive Models (GAMs) are an inherently interpretable class of models that address this limitation by learning a non-linear shape function for each feature separately, followed by a linear model on top. However, these models are typically difficult to train, require numerous parameters, and are difficult to scale. We propose an entirely new subfamily of GAMs that utilizes basis decomposition of shape functions. A small number of basis functions are shared among all features, and are learned jointly for a given task, thus making our model scale much better to large-scale data with high-dimensional features, especially when features are sparse. We propose an architecture denoted as the Neural Basis Model (NBM) which uses a single neural network to learn these bases. On a variety of tabular and image datasets, we demonstrate that for interpretable machine learning, NBMs are the state-of-the-art in accuracy, model size, and, throughput and can easily model all higher-order feature interactions. Source code is available at https://github.com/facebookresearch/nbm-spam.
updated: Tue Oct 18 2022 18:45:19 GMT+0000 (UTC)
published: Fri May 27 2022 17:31:19 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アソシエイト