arXiv reaDer
分解不可能なメトリクスを最適化するためのコストに敏感なセルフトレーニング
Cost-Sensitive Self-Training for Optimizing Non-Decomposable Metrics
自己トレーニング ベースの半教師あり学習アルゴリズムにより、ラベル付きデータのほんの一部を使用して、非常に正確なディープ ニューラル ネットワークの学習が可能になりました。ただし、自己トレーニングに関する作業の大部分は、精度を向上させるという目的に焦点を当ててきましたが、実際の機械学習システムには、本質的に分解できない複雑な目標 (たとえば、クラス全体で最小のリコールを最大化するなど) がある場合があります。この作業では、分解不可能なメトリックを最適化するためのセルフ トレーニング ベースの方法を一般化する、Cost-Sensitive Self-Training (CSST) フレームワークを紹介します。セルフトレーニングの分析のために作成された同様のデータ分布の仮定の下で、ラベルのないデータを利用して、フレームワークが目的の分解不可能なメトリックをより適切に最適化できることを証明します。提案された CSST フレームワークを使用して、深層ニューラル ネットワークを使用してさまざまな分解不可能なメトリックを最適化するための実用的なセルフ トレーニング メソッド (ビジョンと NLP タスクの両方) を取得します。私たちの結果は、CSST が、データセットと目的にまたがる大部分のケースで最先端技術を超える改善を達成することを示しています。
Self-training based semi-supervised learning algorithms have enabled the learning of highly accurate deep neural networks, using only a fraction of labeled data. However, the majority of work on self-training has focused on the objective of improving accuracy, whereas practical machine learning systems can have complex goals (e.g. maximizing the minimum of recall across classes, etc.) that are non-decomposable in nature. In this work, we introduce the Cost-Sensitive Self-Training (CSST) framework which generalizes the self-training-based methods for optimizing non-decomposable metrics. We prove that our framework can better optimize the desired non-decomposable metric utilizing unlabeled data, under similar data distribution assumptions made for the analysis of self-training. Using the proposed CSST framework, we obtain practical self-training methods (for both vision and NLP tasks) for optimizing different non-decomposable metrics using deep neural networks. Our results demonstrate that CSST achieves an improvement over the state-of-the-art in majority of the cases across datasets and objectives.
updated: Fri Apr 28 2023 10:31:12 GMT+0000 (UTC)
published: Fri Apr 28 2023 10:31:12 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アソシエイト