arXiv reaDer
NASiam: シャム ネットワークのニューラル アーキテクチャ検索を使用した効率的な表現学習
NASiam: Efficient Representation Learning using Neural Architecture Search for Siamese Networks
シャム ネットワークは、自己教師あり視覚表現学習 (SSL) を実現するための最もトレンドのある方法の 1 つです。手作業によるラベル付けはコストがかかるため、ラベル付けされていない大規模なデータセットでディープ ラーニングをトレーニングできるようにすることで、SSL は重要な役割を果たすことができます。一方、ニューラル アーキテクチャ サーチ (NAS) は、新しいディープ ラーニング アーキテクチャを発見する手法としてますます重要になっています。しかし、強化学習や進化的アルゴリズムに基づく初期の NAS 手法は、途方もない計算コストとメモリ コストに悩まされていました。対照的に、勾配ベースのアプローチである微分可能 NAS には、はるかに効率的であるという利点があり、過去数年間、ほとんどの注目を集め続けています。この記事では、NASiam を紹介します。NASiam は、微分可能な NAS を初めて使用して、シャム ネットワーク ベースの対照的な学習フレームワーク (SimCLR、SimSiam、および MoCo) 以前のベースラインのシンプルさを維持しながら。多層パーセプトロン用に明示的に設計された検索空間を作成し、その中で標準の ReLU 活性化関数に代わるいくつかの方法を検討しました。これらの新しいアーキテクチャにより、ResNet バックボーンの畳み込みモデルが強力な表現を効率的に学習できることを示します。 NASiam は、小規模 (つまり、CIFAR-10/CIFAR-100) と大規模 (つまり、ImageNet) の画像分類データセットの両方で、わずか数 GPU 時間のコストで競争力のあるパフォーマンスを達成します。 NAS が発見したアーキテクチャの構成について議論し、それらが崩壊動作を防ぐことができた理由について仮説を立てます。コードは https://github.com/aheuillet/NASiam で入手できます。
Siamese networks are one of the most trending methods to achieve self-supervised visual representation learning (SSL). Since hand labeling is costly, SSL can play a crucial part by allowing deep learning to train on large unlabeled datasets. Meanwhile, Neural Architecture Search (NAS) is becoming increasingly important as a technique to discover novel deep learning architectures. However, early NAS methods based on reinforcement learning or evolutionary algorithms suffered from ludicrous computational and memory costs. In contrast, differentiable NAS, a gradient-based approach, has the advantage of being much more efficient and has thus retained most of the attention in the past few years. In this article, we present NASiam, a novel approach that uses for the first time differentiable NAS to improve the multilayer perceptron projector and predictor (encoder/predictor pair) architectures inside siamese-networks-based contrastive learning frameworks (e.g., SimCLR, SimSiam, and MoCo) while preserving the simplicity of previous baselines. We crafted a search space designed explicitly for multilayer perceptrons, inside which we explored several alternatives to the standard ReLU activation function. We show that these new architectures allow ResNet backbone convolutional models to learn strong representations efficiently. NASiam reaches competitive performance in both small-scale (i.e., CIFAR-10/CIFAR-100) and large-scale (i.e., ImageNet) image classification datasets while costing only a few GPU hours. We discuss the composition of the NAS-discovered architectures and emit hypotheses on why they manage to prevent collapsing behavior. Our code is available at https://github.com/aheuillet/NASiam.
updated: Tue Jan 31 2023 19:48:37 GMT+0000 (UTC)
published: Tue Jan 31 2023 19:48:37 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アソシエイト