arXiv reaDer
PocketNet:医療画像分析のためのより小さなニューラルネットワーク
PocketNet: A Smaller Neural Network for Medical Image Analysis
医用画像の深層学習モデルは、多くの場合、大きく複雑であり、これらのモデルをトレーニングおよび評価するための専用ハードウェアが必要です。このような問題に対処するために、畳み込みニューラルネットワークのチャネル数の増加を抑えることで深層学習モデルのサイズを縮小するPocketNetパラダイムを提案します。さまざまなセグメンテーションおよび分類タスクについて、PocketNetアーキテクチャは、パラメータの数を数桁削減し、GPUメモリを最大90%削減し、トレーニング時間を短縮しながら、従来のニューラルネットワークと同等の結果を生成することを示します。最大40%増加するため、このようなモデルをリソースに制約のある設定でトレーニングおよびデプロイできます。
Medical imaging deep learning models are often large and complex, requiring specialized hardware to train and evaluate these models. To address such issues, we propose the PocketNet paradigm to reduce the size of deep learning models by throttling the growth of the number of channels in convolutional neural networks. We demonstrate that, for a range of segmentation and classification tasks, PocketNet architectures produce results comparable to that of conventional neural networks while reducing the number of parameters by multiple orders of magnitude, using up to 90% less GPU memory, and speeding up training times by up to 40%, thereby allowing such models to be trained and deployed in resource-constrained settings.
updated: Wed May 18 2022 14:34:55 GMT+0000 (UTC)
published: Wed Apr 21 2021 20:10:30 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アソシエイト