arXiv reaDer
ジョイントディープエンベデッドクラスタリングによる統計的表現の学習
Learning Statistical Representation with Joint Deep Embedded Clustering
教師なし学習の最も有望なアプローチの1つは、深い表現学習と深いクラスタリングを組み合わせることです。最近のいくつかの研究では、ディープニューラルネットワークを使用して表現を同時に学習し、埋め込まれた機能に加えてクラスタリング損失を定義することでクラスタリングを実行することを提案しています。ただし、これらのアプローチは、不均衡なデータと分布外のサンプルに敏感です。したがって、これらのメソッドは、ランダムに初期化されたクラスターセンターの近くにデータをプッシュすることにより、クラスタリングを最適化します。これは、インスタンスの数がクラスごとに大きく異なる場合や、サンプルが少ないクラスターに適切な重心が割り当てられる可能性が低い場合に問題になります。これらの制限を克服するために、統計的表現の共同学習とクラスタリングのための新しい教師なしフレームワークであるStatDECを導入します。 StatDECは、2つのディープラーニングモデルを同時にトレーニングします。データ分散をキャプチャするディープ統計ネットワークと、組み込み機能を学習し、クラスタリング損失を明示的に定義することによってクラスタリングを実行するディープクラスタリングネットワークです。具体的には、クラスタリングネットワークと表現ネットワークの両方が、平均、分散、カーディナリティを表す提案された統計プーリングレイヤーを利用して、分布外のサンプルとクラスの不均衡を処理します。私たちの実験は、これらの表現を使用すると、さまざまな画像データセットにわたる不均衡な画像クラスタリングの結果を大幅に改善できることを示しています。さらに、学習された表現は、分布外のデータセットに転送されたときによく一般化されます。
One of the most promising approaches for unsupervised learning is combining deep representation learning and deep clustering. Some recent works propose to simultaneously learn representation using deep neural networks and perform clustering by defining a clustering loss on top of embedded features. However, these approaches are sensitive to imbalanced data and out-of-distribution samples. Hence, these methods optimize clustering by pushing data close to randomly initialized cluster centers. This is problematic when the number of instances varies largely in different classes or a cluster with few samples has less chance to be assigned a good centroid. To overcome these limitations, we introduce StatDEC, a new unsupervised framework for joint statistical representation learning and clustering. StatDEC simultaneously trains two deep learning models, a deep statistics network that captures the data distribution, and a deep clustering network that learns embedded features and performs clustering by explicitly defining a clustering loss. Specifically, the clustering network and representation network both take advantage of our proposed statistics pooling layer that represents mean, variance, and cardinality to handle the out-of-distribution samples as well as a class imbalance. Our experiments show that using these representations, one can considerably improve results on imbalanced image clustering across a variety of image datasets. Moreover, the learned representations generalize well when transferred to the out-of-distribution dataset.
updated: Sat Sep 11 2021 09:26:52 GMT+0000 (UTC)
published: Sat Sep 11 2021 09:26:52 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アソシエイト