web-dev-qa-db-ja.com

ケラスのFlatten()とGlobalAveragePooling2D()の違いは何ですか

ConvLSTMとConv2Dの出力をKerasの高密度レイヤーに渡したいのですが、私の場合、グローバル平均プーリングとフラット化の両方を使用することの違いは何ですか?.

model.add(ConvLSTM2D(filters=256,kernel_size=(3,3)))
model.add(Flatten())
# or model.add(GlobalAveragePooling2D())
model.add(Dense(256,activation='relu'))
10
user239457

どちらも機能しているように見えても、同じように機能するわけではありません。

Flattenは、任意の形状のテンソルを取り、それを1次元テンソル(およびサンプル次元)に変換しますが、すべての値をテンソルに保持します。たとえば、テンソル(サンプル、10、20、1)は(サンプル、10 * 20 * 1)に平坦化されます。

GlobalAveragePooling2Dは別のことを行います。各空間次元が1になるまで空間次元に平均プーリングが適用され、他の次元は変更されません。この場合、値は平均化されるため保持されません。たとえば、テンソル(サンプル、10、20、1)は、(サンプル、1、1、1)として出力されます。

20

平坦化は簡単です。要素を再配置することで、多次元オブジェクトを1次元に変換するだけです。

GlobalAveragePoolingは、ベクトルをより適切に表現するために使用される方法です。 1D/2D/3Dにすることができます。オブジェクト全体を移動するパーサーウィンドウを使用し、データを平均化(GlobalAveragePooling)または最大値(GlobalMaxPooling)を選択してプールします。コーナーケースを考慮に入れるには、基本的にパディングが必要です。

どちらも、より簡単な方法でシーケンスの効果を考慮に入れるために使用されます。

0
The Free Soul