arXiv reaDer
柔軟に正則化された混合モデルと画像セグメンテーションへの応用
Flexibly Regularized Mixture Models and Application to Image Segmentation
確率的有限混合モデルは、教師なしクラスタリングに広く使用されています。これらのモデルは、多くの場合、データのトポロジに適合させることで改善できます。たとえば、空間的に隣接するデータポイントを同様に分類するために、各データポイントがクラスに属する事後確率にラプラシアン制約を導入するのが一般的です。あるいは、混合確率を正規化するためにガウス-マルコフまたはより複雑な事前確率を仮定しながら、混合確率を自由パラメーターとして扱うことができます。ただし、これらのアプローチは、事前の形状によって制約され、複雑または手に負えない推論につながることがよくあります。ここでは、過剰パラメータ化された混合分布の混合確率を柔軟に正規化するために、ディリクレ分布の新しいパラメータ化を提案します。期待値最大化アルゴリズムを使用して、私たちのアプローチにより、特別な場合としての空間平滑化正則化を含む、混合確率の線形更新ルールを定義できることを示します。次に、この柔軟な設計を拡張して、複数の混合モデル間でクラス情報を共有できることを示します。アルゴリズムを人工および自然の画像セグメンテーションタスクに適用し、バークレーセグメンテーションデータセットでのガウスおよびスチューデント-t混合のパフォーマンスの定量的および定性的な比較を提供します。また、確率的に最適な方法で深い畳み込みニューラルネットワークのレイヤー全体にクラス情報を伝播する方法を示し、生物学的視覚システムにおけるフィードバック信号の新しい解釈を提案します。私たちの柔軟なアプローチは、確率的混合モデルを任意のデータトポロジに適応させるために簡単に一般化できます。
Probabilistic finite mixture models are widely used for unsupervised clustering. These models can often be improved by adapting them to the topology of the data. For instance, in order to classify spatially adjacent data points similarly, it is common to introduce a Laplacian constraint on the posterior probability that each data point belongs to a class. Alternatively, the mixing probabilities can be treated as free parameters, while assuming Gauss-Markov or more complex priors to regularize those mixing probabilities. However, these approaches are constrained by the shape of the prior and often lead to complicated or intractable inference. Here, we propose a new parametrization of the Dirichlet distribution to flexibly regularize the mixing probabilities of over-parametrized mixture distributions. Using the Expectation-Maximization algorithm, we show that our approach allows us to define any linear update rule for the mixing probabilities, including spatial smoothing regularization as a special case. We then show that this flexible design can be extended to share class information between multiple mixture models. We apply our algorithm to artificial and natural image segmentation tasks, and we provide quantitative and qualitative comparison of the performance of Gaussian and Student-t mixtures on the Berkeley Segmentation Dataset. We also demonstrate how to propagate class information across the layers of deep convolutional neural networks in a probabilistically optimal way, suggesting a new interpretation for feedback signals in biological visual systems. Our flexible approach can be easily generalized to adapt probabilistic mixture models to arbitrary data topologies.
updated: Tue Feb 08 2022 16:45:18 GMT+0000 (UTC)
published: Sat May 25 2019 16:55:19 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アソシエイト