「畳み込みでさらに深くなる」のDepthConcat操作はどのように機能しますか?


12

畳み込みでさらに深く読む私は、さまざまなサイズの複数のテンソルの出力を組み合わせる提案された開始モジュールのビルディングブロックであるDepthConcatレイヤーに出くわしました。著者はこれを「フィルター連結」と呼んでいます。Torchの実装があるようですが、それが何をするのか私にはよくわかりません。誰かが簡単な言葉で説明できますか?

回答:


9

inceptionモジュールの出力のサイズが異なるとは思いません。

ここに画像の説明を入力してください

畳み込み層の場合、人々はしばしば空間解像度を維持するためにパディングを使用します。

他の畳み込みレイヤーの中で右下のプーリングレイヤー(青いフレーム)は扱いにくいように見える場合があります。ただし、従来のプーリングサブサンプリングレイヤー(赤いフレーム、ストライド> 1)とは異なり、そのプーリングレイヤーではストライド1を使用していました。Stride-1プーリングレイヤーは、実際には畳み込みレイヤーと同じように機能しますが、畳み込み演算がmax演算に置き換えられています。

そのため、プーリングレイヤーの後の解像度も変更されず、プーリングレイヤーとたたみ込みレイヤーを「深さ」の次元で連結できます。

ここに画像の説明を入力してください

論文の上の図に示すように、開始モジュールは実際には空間分解能を維持します。


6

ホワイトペーパーと参照したリソースを読んで実装を手助けしたのと同じ質問を心に留めていました。

あなたが参照しトーチコードでは、それは言う:

--[[ DepthConcat ]]--
-- Concatenates the output of Convolutions along the depth dimension
-- (nOutputFrame). This is used to implement the DepthConcat layer
-- of the Going deeper with convolutions paper :

ディープラーニングの「深さ」という言葉は少し曖昧です。幸いにも、このSO Answerはいくつかの明確さを提供します:

ディープニューラルネットワークでは、深さはネットワークの深さを指しますが、このコンテキストでは、深さは視覚認識に使用され、画像の3次元に変換されます。

この場合、画像があり、この入力のサイズは(幅、高さ、奥行き)である32x32x3です。深さがトレーニング画像の異なるチャネルに変換されるため、ニューラルネットワークはこのパラメーターに基づいて学習できる必要があります。

そのため、DepthConcatは、テンソルの最後の次元である深さ次元(この場合は3Dテンソルの3番目の次元)に沿ってテンソルを連結します。

Top コードが言うように、DepthConcatは深さ次元以外のすべての次元でテンソルを同じにする必要があります:

-- The normal Concat Module can't be used since the spatial dimensions
-- of tensors to be concatenated may have different values. To deal with
-- this, we select the largest spatial dimensions and add zero-padding
-- around the smaller dimensions.

例えば

A = tensor of size (14, 14, 2)
B = tensor of size (16, 16, 3)
result = DepthConcat([A, B])
where result with have a height of 16, a width of 16 and a depth of 5 (2 + 3). 

ここに画像の説明を入力してください

上の図では、DepthConcat結果テンソルの画像が表示されています。ここで、白い領域はゼロパディングで、赤はAテンソル、緑はBテンソルです。

この例のDepthConcatの疑似コードは次のとおりです。

  1. テンソルAとテンソルBを見て、最大の空間次元を見つけます。この場合、テンソルBの幅16と高さ16になります。テンソルAは小さすぎてテンソルBの空間次元と一致しないため、パディングする必要があります。
  2. 1番目と2番目の次元にゼロを追加してテンソルAのサイズを作ることにより、テンソルAの空間次元をゼロでパディングします(16、16、2)。
  3. 奥行き(3番目)の次元に沿って、パディングされたテンソルAをテンソルBと連結します。

これが、同じ質問を同じホワイトペーパーを読んでいる他の誰かに役立つことを願っています。


ええ。完璧な導入。これは、深さ方向に連結されます。空間方向ではありません。
Shamane Siriwardhana 2017
弊社のサイトを使用することにより、あなたは弊社のクッキーポリシーおよびプライバシーポリシーを読み、理解したものとみなされます。
Licensed under cc by-sa 3.0 with attribution required.