arXiv reaDer
画像のノイズ除去と分類のための構造制限付きボルツマン マシン
Structural Restricted Boltzmann Machine for image denoising and classification
制限付きボルツマン マシンは、可視ユニットの別の層に接続された隠れ変数の層で構成される生成モデルであり、可視変数の分布をモデル化するために使用されます。より高い表現力を得るために、多くの隠れユニットが一般的に使用され、多数の可視ユニットと組み合わせることで、多数のトレーニング可能なパラメータが得られます。この研究では、構造制限ボルツマン マシン モデルを導入します。これは、手持ちのデータの構造を利用して、パフォーマンスを損なうことなく、トレーニング可能なパラメータの数を大幅に減らすために、隠れたユニットの接続を可視ユニットのサブセットに制限します。応用可能な分野として、画像モデリングに焦点を当てます。画像の性質に基づいて、接続の構造は、モデルの可視変数を構成する画像のピクセル上の空間的近傍の観点から与えられます。私たちはさまざまな画像領域について広範な実験を行っています。画像のノイズ除去は、MNIST データセットからの破損した画像を使用して評価されます。私たちのモデルの生成能力は、バニラ RBM と比較され、さらに 5 つの異なる画像ドメインで評価される分類パフォーマンスも比較されます。結果は、私たちが提案したモデルは、可視ユニットと非表示ユニットの間に制約された接続がない RBM と比較して、より高速で安定したトレーニングを行うと同時に、より良い結果が得られることを示しています。
Restricted Boltzmann Machines are generative models that consist of a layer of hidden variables connected to another layer of visible units, and they are used to model the distribution over visible variables. In order to gain a higher representability power, many hidden units are commonly used, which, in combination with a large number of visible units, leads to a high number of trainable parameters. In this work we introduce the Structural Restricted Boltzmann Machine model, which taking advantage of the structure of the data in hand, constrains connections of hidden units to subsets of visible units in order to reduce significantly the number of trainable parameters, without compromising performance. As a possible area of application, we focus on image modelling. Based on the nature of the images, the structure of the connections is given in terms of spatial neighbourhoods over the pixels of the image that constitute the visible variables of the model. We conduct extensive experiments on various image domains. Image denoising is evaluated with corrupted images from the MNIST dataset. The generative power of our models is compared to vanilla RBMs, as well as their classification performance, which is assessed with five different image domains. Results show that our proposed model has a faster and more stable training, while also obtaining better results compared to an RBM with no constrained connections between its visible and hidden units.
updated: Fri Jun 16 2023 05:18:26 GMT+0000 (UTC)
published: Fri Jun 16 2023 05:18:26 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アソシエイト