arXiv reaDer
MIO : 自己教師あり二値対比学習を用いた相互情報最適化
MIO : Mutual Information Optimization using Self-Supervised Binary Contrastive Learning
自己管理型の対照学習フレームワークは、ここ数年で急速に進歩しました。この論文では、対照学習のための新しい相互情報最適化ベースの損失関数を提案します。事前トレーニング タスクをバイナリ分類問題としてモデル化して、暗黙の対比効果を誘発し、ペアが正か負かを予測します。 Majorize-Minimizer の原則を使用して単純な損失関数をさらに改善し、そのような改善は問題を数学的に追跡するのに役立ちます。既存の方法とは異なり、提案された損失関数は、正と負の両方のペアで相互情報を最適化します。また、パラメーター勾配フローの閉じた形式の式を提示し、そのヘッセ固有スペクトルを使用して提案された損失関数の動作を比較して、SSL フレームワークの収束を分析的に研究します。提案された方法は、CIFAR-10、CIFAR-100、STL-10、および Tiny-ImageNet などのベンチマーク データセットで、SOTA の対照的な自己教師ありフレームワークよりも優れています。 ResNet-18 をバックボーンとして 200 エポックの事前トレーニングを行った後、提案されたモデルは、CIFAR-10、CIFAR-100、STL-10、Tiny- で 86.2%、58.18%、77.49%、30.87% の精度を達成します。それぞれ ImageNet データセットであり、SOTA 対照ベースラインをそれぞれ 1.23%、3.57%、2.00%、0.33% 上回っています。
Self-supervised contrastive learning frameworks have progressed rapidly over the last few years. In this paper, we propose a novel mutual information optimization-based loss function for contrastive learning. We model our pre-training task as a binary classification problem to induce an implicit contrastive effect and predict whether a pair is positive or negative. We further improve the näive loss function using the Majorize-Minimizer principle and such improvement helps us to track the problem mathematically. Unlike the existing methods, the proposed loss function optimizes the mutual information in both positive and negative pairs. We also present a closed-form expression for the parameter gradient flow and compare the behavior of the proposed loss function using its Hessian eigen-spectrum to analytically study the convergence of SSL frameworks. The proposed method outperforms the SOTA contrastive self-supervised frameworks on benchmark datasets like CIFAR-10, CIFAR-100, STL-10, and Tiny-ImageNet. After 200 epochs of pre-training with ResNet-18 as the backbone, the proposed model achieves an accuracy of 86.2%, 58.18%, 77.49%, and 30.87% on CIFAR-10, CIFAR-100, STL-10, and Tiny-ImageNet datasets, respectively, and surpasses the SOTA contrastive baseline by 1.23%, 3.57%, 2.00%, and 0.33%, respectively.
updated: Fri Mar 10 2023 04:12:36 GMT+0000 (UTC)
published: Wed Nov 24 2021 17:51:29 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アソシエイト