arXiv reaDer
座標ごとの最適化による深い対照学習の理解
Understanding Deep Contrastive Learning via Coordinate-wise Optimization
損失関数の幅広いファミリー(InfoNCEを含む)の下での対照学習(CL)は、ネットワークパラメーターθとペアワイズ重要度αの座標ごとの最適化の統一された定式化を持ち、最大プレーヤーθは対照性の表現を学習します。 minplayerαは、類似した表現を共有する別個のサンプルのペアにより多くの重みを置きます。結果として得られるα-CLと呼ばれる定式化は、サンプルペアの重要度αの構築方法によって異なる既存のさまざまな対照損失を統合するだけでなく、外挿して一般的なものを超える新しい対照損失を与えることができ、対照の新しい道を開きます損失設計。これらの新しい損失により、CIFAR10およびSTL-10で従来のInfoNCEと同等(またはそれ以上)のパフォーマンスが得られます。さらに、最大プレーヤーも詳細に分析します。固定αを使用すると、最大プレーヤーは深層線形ネットワークの主成分分析(PCA)と同等であり、ほとんどすべての極小値がグローバルでランク1であり、最適なPCAソリューションを回復することを証明します。 。最後に、max playerの分析を2層のReLUネットワークに拡張し、その固定小数点がより高いランクを持つ可能性があることを示します。
We show that Contrastive Learning (CL) under a broad family of loss functions (including InfoNCE) has a unified formulation of coordinate-wise optimization on the network parameter θ and pairwise importance α, where the max player θ learns representation for contrastiveness, and the min player α puts more weights on pairs of distinct samples that share similar representations. The resulting formulation, called α-CL, unifies not only various existing contrastive losses, which differ by how sample-pair importance α is constructed, but also is able to extrapolate to give novel contrastive losses beyond popular ones, opening a new avenue of contrastive loss design. These novel losses yield comparable (or better) performance on CIFAR10 and STL-10 than classic InfoNCE. Furthermore, we also analyze the max player in detail: we prove that with fixed α, max player is equivalent to Principal Component Analysis (PCA) for deep linear network, and almost all local minima are global and rank-1, recovering optimal PCA solutions. Finally, we extend our analysis on max player to 2-layer ReLU networks, showing that its fixed points can have higher ranks.
updated: Thu Jun 02 2022 23:57:52 GMT+0000 (UTC)
published: Sat Jan 29 2022 23:08:34 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アソシエイト