arXiv reaDer
共同分類器学習によるバイナリネットワークの教師なし表現学習
Unsupervised Representation Learning for Binary Networks by Joint Classifier Learning
自己監視学習は、大規模な浮動小数点ネットワークで成功を収めた有望な教師なし学習フレームワークです。ただし、このようなネットワークは、エッジデバイスに簡単に展開することはできません。さまざまなダウンストリームタスクのためのそのようなリソースが制限されたデバイスへの教師なし表現学習の利点を備えたモデルの展開を加速するために、移動ターゲットネットワークを使用するバイナリネットワークの自己監視学習方法を提案します。特に、事前にトレーニングされた浮動小数点特徴抽出器に接続されたランダムに初期化された分類器を、バイナリネットワークで共同でトレーニングすることを提案します。さらに、特徴類似性損失、動的損失バランシング、および精度をさらに向上させるための修正された多段階トレーニングを提案し、メソッドBURNを呼び出します。 7つのデータセットを使用した5つのダウンストリームタスクでの経験的検証は、BURNがバイナリネットワークの自己監視ベースラインよりも優れており、教師あり事前トレーニングよりも優れている場合があることを示しています。
Self-supervised learning is a promising unsupervised learning framework that has achieved success with large floating point networks. But such networks are not readily deployable to edge devices. To accelerate deployment of models with the benefit of unsupervised representation learning to such resource limited devices for various downstream tasks, we propose a self-supervised learning method for binary networks that uses a moving target network. In particular, we propose to jointly train a randomly initialized classifier, attached to a pretrained floating point feature extractor, with a binary network. Additionally, we propose a feature similarity loss, a dynamic loss balancing and modified multi-stage training to further improve the accuracy, and call our method BURN. Our empirical validations over five downstream tasks using seven datasets show that BURN outperforms self-supervised baselines for binary networks and sometimes outperforms supervised pretraining.
updated: Wed Dec 01 2021 08:00:17 GMT+0000 (UTC)
published: Sun Oct 17 2021 15:38:39 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アソシエイト