arXiv reaDer
等角基底ベクトル
Equiangular Basis Vectors
分類タスクの等角基底ベクトル (EBV) を提案します。ディープ ニューラル ネットワークでは、モデルは通常、さまざまな分類タスクを処理するためのソフトマックスを備えた k-way 全結合層で終了します。これらの方法の学習目的は、学習した特徴表現をサンプルのラベル空間にマッピングすることとして要約できます。メトリック学習アプローチでは、主な目的は、トレーニング データ ポイントを元の空間から新しい空間にマッピングする変換関数を学習することです。この空間では、類似点がより近くなり、類似点がより遠くなります。以前の方法とは異なり、当社の EBV は正規化されたベクトル埋め込みを「事前定義された分類子」として生成します。これは、互いに同等の状態であるだけでなく、可能な限り直交する必要があります。トレーニングでカテゴリカル EBV 間の入力の埋め込みの球面距離を最小化することにより、推論中に最小距離でカテゴリカル EBV を識別することによって予測を取得できます。 ImageNet-1K データセットおよびその他のダウンストリーム タスクに関するさまざまな実験は、従来のメトリック学習方法と比較して膨大な追加計算を導入することなく、この方法が一般的な全結合分類器よりも優れていることを示しています。当社の EBV は、2022 DIGIX グローバル AI チャレンジで 1 位を獲得しました。当社のコードはオープンソースであり、https://github.com/NJUST-VIPGroup/Equiangular-Basis-Vectors で入手できます。
We propose Equiangular Basis Vectors (EBVs) for classification tasks. In deep neural networks, models usually end with a k-way fully connected layer with softmax to handle different classification tasks. The learning objective of these methods can be summarized as mapping the learned feature representations to the samples' label space. While in metric learning approaches, the main objective is to learn a transformation function that maps training data points from the original space to a new space where similar points are closer while dissimilar points become farther apart. Different from previous methods, our EBVs generate normalized vector embeddings as "predefined classifiers" which are required to not only be with the equal status between each other, but also be as orthogonal as possible. By minimizing the spherical distance of the embedding of an input between its categorical EBV in training, the predictions can be obtained by identifying the categorical EBV with the smallest distance during inference. Various experiments on the ImageNet-1K dataset and other downstream tasks demonstrate that our method outperforms the general fully connected classifier while it does not introduce huge additional computation compared with classical metric learning methods. Our EBVs won the first place in the 2022 DIGIX Global AI Challenge, and our code is open-source and available at https://github.com/NJUST-VIPGroup/Equiangular-Basis-Vectors.
updated: Tue Mar 21 2023 07:08:51 GMT+0000 (UTC)
published: Tue Mar 21 2023 07:08:51 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アソシエイト