DeepCluster

ディープラーニング




Deep Cluster

論文:
Mathilde Caron, Piotr Bojanowski, Armand Joulin, Matthijs Douze(2018). Deep Clustering for Unsupervised Learning of Visual Features
[github]

Deep Clusterの構造

実装するときの処理の流れは、下のようになると思います。

Deep Clusterは、既存のクラスタリング手法であるKmeansを使って、Pseudo Labelingを行い、分類をする手法です。

結局既存の手法を使うのかと少しがっかりしました(特徴量抽出に近いのかな?)

DBSCANなどのアルゴリズムを使った応用が考えられます。

実装

論文のチームがgithubにコードを公開しているので、すごく助かります。
[github]facebookresearch/deepcluster
Pytorchで書かれているので、馴染みがあるひとも多いでしょう(という私もPyTorch派です)

公式のコードを参考に、自分で書き直した。
MNISTを試してみたのですが、あまり上手くいかなかった。。。

参考

タイトルとURLをコピーしました