arXiv reaDer
カーネル依存性の最大化による自己教師あり学習
Self-Supervised Learning with Kernel Dependence Maximization
統計的依存性の観点から画像表現の自己教師あり学習にアプローチし、ヒルベルト・シュミット独立基準(SSL-HSIC)を使用した自己教師あり学習を提案します。 SSL-HSICは、変換されたバージョンの画像の表現と画像IDの間の依存関係を最大化すると同時に、これらの機能のカーネル化された分散を最小化します。この自己教師あり学習フレームワークは、InfoNCEの新しい理解をもたらします。これは、異なる変換間の相互情報量(MI)の変動的な下限です。 MI自体には、意味のない表現が学習される可能性のある病状があることが知られていますが、その限界ははるかに適切に動作します。暗黙的にSSL-HSICを近似することを示します(わずかに異なる正則化を使用)。 SSL-HSICも同様にサンプルのローカル近傍を学習するため、私たちのアプローチはBYOLへの洞察も提供します。 SSL-HSICを使用すると、制限的なデータの仮定や間接的な相互情報量の推定を行うことなく、バッチサイズの時間線形における統計的依存性を直接最適化できます。ターゲットネットワークの有無にかかわらずトレーニングされたSSL-HSICは、ImageNetでの標準的な線形評価、半教師あり学習、およびセマンティックセグメンテーション、深度推定、オブジェクト認識などの他の分類およびビジョンタスクへの転送のために、現在の最先端技術と一致します。 。
We approach self-supervised learning of image representations from a statistical dependence perspective, proposing Self-Supervised Learning with the Hilbert-Schmidt Independence Criterion (SSL-HSIC). SSL-HSIC maximizes dependence between representations of transformed versions of an image and the image identity, while minimizing the kernelized variance of those features. This self-supervised learning framework yields a new understanding of InfoNCE, a variational lower bound on the mutual information (MI) between different transformations. While the MI itself is known to have pathologies which can result in meaningless representations being learned, its bound is much better behaved: we show that it implicitly approximates SSL-HSIC (with a slightly different regularizer). Our approach also gives us insight into BYOL, since SSL-HSIC similarly learns local neighborhoods of samples. SSL-HSIC allows us to directly optimize statistical dependence in time linear in the batch size, without restrictive data assumptions or indirect mutual information estimators. Trained with or without a target network, SSL-HSIC matches the current state-of-the-art for standard linear evaluation on ImageNet, semi-supervised learning and transfer to other classification and vision tasks such as semantic segmentation, depth estimation and object recognition.
updated: Tue Jun 15 2021 17:51:16 GMT+0000 (UTC)
published: Tue Jun 15 2021 17:51:16 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アソシエイト