arXiv reaDer
座標ごとの最適化による深層対照学習の理解
Understanding Deep Contrastive Learning via Coordinate-wise Optimization
損失関数の広範なファミリ (InfoNCE を含む) の下での Contrastive Learning (CL) には、ネットワーク パラメーター θ とペアごとの重要度 α に関する座標ごとの最適化の統一された定式化があることを示します。最小プレーヤー α は、同様の表現を共有する異なるサンプルのペアにより多くの重みを付けます。得られた定式化は α-CL と呼ばれ、サンプル ペアの重要度 α がどのように構築されるかによって異なるさまざまな既存のコントラスト損失を統合するだけでなく、一般的なものを超えた新しいコントラスト損失を推定して、コントラストの新しい道を開くこともできます。ロスデザイン。これらの新しい損失により、CIFAR10、STL-10、および CIFAR-100 で従来の InfoNCE と同等 (またはそれ以上) のパフォーマンスが得られます。さらに、最大プレーヤーも詳細に分析します。α が固定されている場合、最大プレーヤーは深い線形ネットワークの主成分分析 (PCA) と同等であり、ほとんどすべての極小値はグローバルでランク 1 であり、最適な PCA ソリューションを回復することを証明します。 .最後に、最大プレーヤーの分析を 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, STL-10 and CIFAR-100 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: Sun Nov 20 2022 18:56:00 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アソシエイト