arXiv reaDer
脳腫瘍セグメンテーションのための可逆的なモバイル反転ボトルネックを備えたメモリ効率の高い3DU-Net
Memory Efficient 3D U-Net with Reversible Mobile Inverted Bottlenecks for Brain Tumor Segmentation
脳腫瘍セグメンテーション(BraTS)チャレンジのモデルの複雑さを増すために、メモリ節約技術を従来のU-Netアーキテクチャと組み合わせることを提案します。 BraTSチャレンジは、240x240x155x4の入力画像を一連の腫瘍クラスに3Dセグメンテーションすることで構成されます。ボリュームが大きく、3D畳み込み層が必要なため、このタスクは非常にメモリを消費します。これに対処するために、以前のアプローチでは、モデルの深さと幅を制限しながら、より小さなトリミングされた画像を使用します。 3D U-Netは、MobileNetV2、MnasNet、および最新のEfficientNetアーキテクチャで定義されたモバイル反転ボトルネックブロックのリバーシブルバージョンを使用して、トレーニング中のアクティベーションメモリを節約します。リバーシブルレイヤーを使用すると、モデルはそのレイヤーの出力を指定して入力アクティベーションを再計算でき、フォワードパス中にアクティベーションを保存する必要がなくなるため、メモリを節約できます。反転残余ボトルネックブロックは、軽量の深さ方向に分離可能な畳み込みを使用して、畳み込みを点ごとの畳み込みと深さ方向の畳み込みに分解することにより、計算を削減します。さらに、このブロックは、入力と出力の線形1x1畳み込みの間に中間拡張レイヤーを配置することにより、従来のボトルネックブロックを反転させ、チャネルの総数を減らします。固定メモリバジェットが与えられると、これらのメモリ節約技術を使用して、対応する非可逆ネットワークよりも最大3倍大きい画像ボリューム、25%深いモデル、または最大2倍のチャネル数を持つモデルをトレーニングできます。
We propose combining memory saving techniques with traditional U-Net architectures to increase the complexity of the models on the Brain Tumor Segmentation (BraTS) challenge. The BraTS challenge consists of a 3D segmentation of a 240x240x155x4 input image into a set of tumor classes. Because of the large volume and need for 3D convolutional layers, this task is very memory intensive. To address this, prior approaches use smaller cropped images while constraining the model's depth and width. Our 3D U-Net uses a reversible version of the mobile inverted bottleneck block defined in MobileNetV2, MnasNet and the more recent EfficientNet architectures to save activation memory during training. Using reversible layers enables the model to recompute input activations given the outputs of that layer, saving memory by eliminating the need to store activations during the forward pass. The inverted residual bottleneck block uses lightweight depthwise separable convolutions to reduce computation by decomposing convolutions into a pointwise convolution and a depthwise convolution. Further, this block inverts traditional bottleneck blocks by placing an intermediate expansion layer between the input and output linear 1x1 convolution, reducing the total number of channels. Given a fixed memory budget, with these memory saving techniques, we are able to train image volumes up to 3x larger, models with 25% more depth, or models with up to 2x the number of channels than a corresponding non-reversible network.
updated: Wed Apr 21 2021 01:02:05 GMT+0000 (UTC)
published: Mon Apr 19 2021 21:23:55 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アソシエイト