arXiv reaDer
説明可能なセグメンテーションのための微分可能なガウス プロトタイプ レイヤー
A differentiable Gaussian Prototype Layer for explainable Segmentation
勾配ベースのプロトタイプ学習のためのガウス プロトタイプ層を​​導入し、説明可能なセグメンテーションのための 2 つの新しいネットワーク アーキテクチャを示します。そのうちの 1 つは領域提案に依存します。どちらのモデルも農業データセットで評価されます。これまでニューラル ネットワークの潜在分布をモデル化するためにガウス混合モデル (GMM) が使用されてきましたが、通常は EM アルゴリズムを使用してフィッティングされます。代わりに、提案されたプロトタイプ層は勾配ベースの最適化に依存しているため、エンドツーエンドのトレーニングが可能になります。これにより開発が容易になり、トレーニング可能な深部特徴抽出器の可能性を最大限に活用できるようになります。私たちは、これが説明可能なニューラル ネットワークの新しい構成要素として使用できることを示します。ガウス プロトタイプ レイヤーは、(1) 潜在グリッドでプロトタイプが検出されるモデル、および (2) 領域提案として SLIC スーパーピクセルを使用する Fast-RCNN からインスピレーションを得たモデルで使用されます。前者は最先端技術と比較して同様のパフォーマンスを達成しますが、後者は精度がわずかに低下する代わりに、より正確なプロトタイプの位置特定という利点があります。勾配ベースの GMM レイヤーを導入することで、エンドツーエンドのトレーニングの利点と GMM のシンプルさと理論的基盤を組み合わせ、将来的には既存の半教師あり学習戦略をプロトタイプの部品モデルに適応させることができます。
We introduce a Gaussian Prototype Layer for gradient-based prototype learning and demonstrate two novel network architectures for explainable segmentation one of which relies on region proposals. Both models are evaluated on agricultural datasets. While Gaussian Mixture Models (GMMs) have been used to model latent distributions of neural networks before, they are typically fitted using the EM algorithm. Instead, the proposed prototype layer relies on gradient-based optimization and hence allows for end-to-end training. This facilitates development and allows to use the full potential of a trainable deep feature extractor. We show that it can be used as a novel building block for explainable neural networks. We employ our Gaussian Prototype Layer in (1) a model where prototypes are detected in the latent grid and (2) a model inspired by Fast-RCNN with SLIC superpixels as region proposals. The earlier achieves a similar performance as compared to the state-of-the art while the latter has the benefit of a more precise prototype localization that comes at the cost of slightly lower accuracies. By introducing a gradient-based GMM layer we combine the benefits of end-to-end training with the simplicity and theoretical foundation of GMMs which will allow to adapt existing semi-supervised learning strategies for prototypical part models in future.
updated: Sun Jun 25 2023 22:33:21 GMT+0000 (UTC)
published: Sun Jun 25 2023 22:33:21 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アソシエイト