ディープニューラルネットワーク(DNN)から抽出された機能は、コンテンツベースの画像検索(CBIR)のコンテキストで非常に効果的であることが証明されています。最近の研究では、生物学的に着想を得たヘッブの学習アルゴリズムがDNNトレーニングの可能性を示しています。この寄稿では、CBIRタスク用の特徴抽出器の開発におけるそのようなアルゴリズムのパフォーマンスを研究します。具体的には、2つのステップで半教師あり学習戦略を検討します。最初に、画像データセットでヘッブ学習を使用して、教師なし事前トレーニング段階を実行します。次に、監視された確率的勾配降下法(SGD)トレーニングを使用してネットワークを微調整します。教師なし事前トレーニング段階では、非線形ヘッブ主成分分析(HPCA)学習ルールを検討します。監視対象の微調整段階では、ラベル付けされたサンプルの量がデータセット全体のごく一部であるサンプル効率シナリオを想定しています。 CIFAR10およびCIFAR100データセットで実施された実験分析は、ラベル付けされたサンプルがほとんどない場合、ヘッブのアプローチがさまざまな代替方法と比較して関連する改善を提供することを示しています。
Features extracted from Deep Neural Networks (DNNs) have proven to be very effective in the context of Content Based Image Retrieval (CBIR). In recent work, biologically inspired Hebbian learning algorithms have shown promises for DNN training. In this contribution, we study the performance of such algorithms in the development of feature extractors for CBIR tasks. Specifically, we consider a semi-supervised learning strategy in two steps: first, an unsupervised pre-training stage is performed using Hebbian learning on the image dataset; second, the network is fine-tuned using supervised Stochastic Gradient Descent (SGD) training. For the unsupervised pre-training stage, we explore the nonlinear Hebbian Principal Component Analysis (HPCA) learning rule. For the supervised fine-tuning stage, we assume sample efficiency scenarios, in which the amount of labeled samples is just a small fraction of the whole dataset. Our experimental analysis, conducted on the CIFAR10 and CIFAR100 datasets shows that, when few labeled samples are available, our Hebbian approach provides relevant improvements compared to various alternative methods.