arXiv reaDer
画像クラスタリングのための階層グラフニューラルネットワークの学習
Learning Hierarchical Graph Neural Networks for Image Clustering
互いに素なアイデンティティのセットに属するラベルで注釈が付けられた画像のトレーニングセットを使用して、画像のセットを未知の数のアイデンティティにクラスター化する方法を学習する階層グラフニューラルネットワーク(GNN)モデルを提案します。階層GNNは、新しいアプローチを使用して、階層の各レベルで予測された連結成分をマージし、次のレベルで新しいグラフを形成します。完全に教師なしの階層的クラスタリングとは異なり、グループ化と複雑さの基準の選択は、当然、トレーニングセットの監督から生じます。結果として得られる方法であるHi-LANDERは、現在のGNNベースのクラスタリングアルゴリズムと比較して、Fスコアが平均54%向上し、正規化相互情報量(NMI)が8%向上します。さらに、最先端のGNNベースの方法は、クラスタリングプロセスの中間ステップとして、リンケージ確率とノード密度を予測するために個別のモデルに依存しています。対照的に、私たちの統一されたフレームワークは、計算コストを7分の1に削減します。トレーニングと推論のコードはhttps://github.com/dmlc/dgl/tree/master/examples/pytorch/hilanderでリリースされています。
We propose a hierarchical graph neural network (GNN) model that learns how to cluster a set of images into an unknown number of identities using a training set of images annotated with labels belonging to a disjoint set of identities. Our hierarchical GNN uses a novel approach to merge connected components predicted at each level of the hierarchy to form a new graph at the next level. Unlike fully unsupervised hierarchical clustering, the choice of grouping and complexity criteria stems naturally from supervision in the training set. The resulting method, Hi-LANDER, achieves an average of 54% improvement in F-score and 8% increase in Normalized Mutual Information (NMI) relative to current GNN-based clustering algorithms. Additionally, state-of-the-art GNN-based methods rely on separate models to predict linkage probabilities and node densities as intermediate steps of the clustering process. In contrast, our unified framework achieves a seven-fold decrease in computational cost. We release our training and inference code at https://github.com/dmlc/dgl/tree/master/examples/pytorch/hilander.
updated: Sat Jul 17 2021 14:50:19 GMT+0000 (UTC)
published: Sat Jul 03 2021 01:28:42 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アソシエイト