arXiv reaDer
ComBiNet: Compact Convolutional Bayesian Neural Network for Image Segmentation
完全畳み込みU字型ニューラルネットワークは、主にピクセル単位の画像セグメンテーションの主要なアプローチです。この作業では、実際のアプリケーションでの展開を妨げる2つの欠陥に取り組みます。1)予測には、多くの意思決定システムにとって重要な不確実性の定量化が欠けています。 2)大容量のメモリストレージと大量のハードウェアリソースを必要とする計算消費。これらの問題に対処し、その実用性を向上させるために、いくつかのパラメーターのコンパクトなベイジアン畳み込みアーキテクチャを示します。これにより、大幅に少ないパラメーターと計算操作を使用する関連作業と比較して、精度がわずかに向上します。このアーキテクチャは、分離可能な畳み込み、双一次内挿、マルチスケールフィーチャの伝播、モンテカルロドロップアウトによるピクセルごとの不確実性の定量化のためのベイズ推定などのパラメータ効率の高い操作を組み合わせたものです。最高のパフォーマンスを発揮する構成では、さまざまな困難なデータセットで250万未満のパラメーターが必要であり、観測はほとんどありませんでした。
Fully convolutional U-shaped neural networks have largely been the dominant approach for pixel-wise image segmentation. In this work, we tackle two defects that hinder their deployment in real-world applications: 1) Predictions lack uncertainty quantification that may be crucial to many decision making systems; 2) Large memory storage and computational consumption demanding extensive hardware resources. To address these issues and improve their practicality we demonstrate a few-parameter compact Bayesian convolutional architecture, that achieves a marginal improvement in accuracy in comparison to related work using significantly fewer parameters and compute operations. The architecture combines parameter-efficient operations such as separable convolutions, bi-linear interpolation, multi-scale feature propagation and Bayesian inference for per-pixel uncertainty quantification through Monte Carlo Dropout. The best performing configurations required fewer than 2.5 million parameters on diverse challenging datasets with few observations.
updated: Wed Apr 14 2021 16:33:48 GMT+0000 (UTC)
published: Wed Apr 14 2021 16:33:48 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト