arXiv reaDer
一人でクラスタリングすることはありません
You Never Cluster Alone
インスタンス レベルの対照的な目的を使用した自己教師あり学習の最近の進歩により、教師なしクラスタリングが容易になります。ただし、スタンドアロンのデータは全体的なクラスターのコンテキストを認識していないため、最適ではない割り当てを受ける可能性があります。この論文では、主流の対照的学習パラダイムをクラスターレベルのスキームに拡張します.クラスターレベルのスキームでは、同じクラスターの対象となるすべてのデータが、各データグループのコンテキストをエンコードする統一された表現に貢献します。この表現を使用した対照学習は、各データの割り当てに報酬を与えます。このビジョンを実装するために、ツイン コントラスト クラスタリング (TCC) を提案します。インスタンス レベルの学習トラックとクラスター レベルの学習トラックをリンクするクラスタリング割り当て信頼度としてカテゴリ変数のセットを定義します。一方では、対応する割り当て変数が重みである場合、データ ポイントに沿った重み付き集計は、クラスターの集合表現を実装します。さらに、クラスター レベルの対照的な学習を可能にするヒューリスティックなクラスター拡張に相当するものを提案します。一方、インスタンス レベルの対照的な目的の証拠の下限を割り当てを導出します。割り当て変数を再パラメータ化することにより、TCC はエンド ツー エンドでトレーニングされ、交互の手順は必要ありません。大規模な実験により、TCC は困難なベンチマークで最先端のパフォーマンスを上回ることが示されています。
Recent advances in self-supervised learning with instance-level contrastive objectives facilitate unsupervised clustering. However, a standalone datum is not perceiving the context of the holistic cluster, and may undergo sub-optimal assignment. In this paper, we extend the mainstream contrastive learning paradigm to a cluster-level scheme, where all the data subjected to the same cluster contribute to a unified representation that encodes the context of each data group. Contrastive learning with this representation then rewards the assignment of each datum. To implement this vision, we propose twin-contrast clustering (TCC). We define a set of categorical variables as clustering assignment confidence, which links the instance-level learning track with the cluster-level one. On one hand, with the corresponding assignment variables being the weight, a weighted aggregation along the data points implements the set representation of a cluster. We further propose heuristic cluster augmentation equivalents to enable cluster-level contrastive learning. On the other hand, we derive the evidence lower-bound of the instance-level contrastive objective with the assignments. By reparametrizing the assignment variables, TCC is trained end-to-end, requiring no alternating steps. Extensive experiments show that TCC outperforms the state-of-the-art on challenging benchmarks.
updated: Thu Jun 03 2021 14:59:59 GMT+0000 (UTC)
published: Thu Jun 03 2021 14:59:59 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アソシエイト