arXiv reaDer
微分可能なTop-k分類学習
Differentiable Top-k Classification Learning
top-k分類の精度は、機械学習の主要な指標の1つです。ここで、kは通常、1または5などの正の整数であり、上位1または上位5のトレーニング目標につながります。この作業では、この仮定を緩和し、単一のkを使用する代わりに、複数のkに対してモデルを同時に最適化します。微分可能な並べ替えとランク付けの最近の進歩を利用して、微分可能なトップkクロスエントロピー分類損失を提案します。これにより、トップ1の予測だけでなく、たとえばトップ2とトップ5の予測も考慮しながらネットワークをトレーニングできます。提案された損失関数を評価して、最先端のアーキテクチャの微調整と、ゼロからのトレーニングを行います。 kを緩和すると、上位5の精度が向上するだけでなく、上位1の精度も向上することがわかります。公開されているImageNetモデルを微調整することで、これらのモデルの新しい最先端を実現します。
The top-k classification accuracy is one of the core metrics in machine learning. Here, k is conventionally a positive integer, such as 1 or 5, leading to top-1 or top-5 training objectives. In this work, we relax this assumption and optimize the model for multiple k simultaneously instead of using a single k. Leveraging recent advances in differentiable sorting and ranking, we propose a differentiable top-k cross-entropy classification loss. This allows training the network while not only considering the top-1 prediction, but also, e.g., the top-2 and top-5 predictions. We evaluate the proposed loss function for fine-tuning on state-of-the-art architectures, as well as for training from scratch. We find that relaxing k does not only produce better top-5 accuracies, but also leads to top-1 accuracy improvements. When fine-tuning publicly available ImageNet models, we achieve a new state-of-the-art for these models.
updated: Wed Jun 15 2022 04:13:59 GMT+0000 (UTC)
published: Wed Jun 15 2022 04:13:59 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アソシエイト