arXiv reaDer
アクティブな微調整: 事前トレーニングと微調整のパラダイムにおけるアノテーション バジェットの活用
Active Finetuning: Exploiting Annotation Budget in the Pretraining-Finetuning Paradigm
大規模なデータと高い注釈コストを考えると、事前トレーニングと微調整は、複数のコンピューター ビジョン タスクで一般的なパラダイムになります。以前の研究では、このパラダイムでの教師なし事前トレーニングと教師あり微調整の両方がカバーされていましたが、微調整のためのアノテーション バジェットの活用にはほとんど注意が払われていませんでした。このギャップを埋めるために、事前トレーニングと微調整のパラダイムでの注釈用のサンプルの選択に焦点を当てた、この新しいアクティブな微調整タスクを正式に定義します。ラベル付けされていないプール全体と同様に分布し、連続空間でパラメトリックモデルを最適化することにより十分な多様性を維持するデータのサブセットを選択するアクティブ微調整タスク用の ActiveFT と呼ばれる新しい方法を提案します。選択したサブセットの分布とデータ プール全体との間の Earth Mover の距離も、このプロセスで減少することを証明します。広範な実験により、画像分類とセマンティック セグメンテーションの両方でベースラインよりも優れた ActiveFT の優れたパフォーマンスと高効率が示されています。私たちのコードは https://github.com/yichen928/ActiveFT でリリースされています。
Given the large-scale data and the high annotation cost, pretraining-finetuning becomes a popular paradigm in multiple computer vision tasks. Previous research has covered both the unsupervised pretraining and supervised finetuning in this paradigm, while little attention is paid to exploiting the annotation budget for finetuning. To fill in this gap, we formally define this new active finetuning task focusing on the selection of samples for annotation in the pretraining-finetuning paradigm. We propose a novel method called ActiveFT for active finetuning task to select a subset of data distributing similarly with the entire unlabeled pool and maintaining enough diversity by optimizing a parametric model in the continuous space. We prove that the Earth Mover's distance between the distributions of the selected subset and the entire data pool is also reduced in this process. Extensive experiments show the leading performance and high efficiency of ActiveFT superior to baselines on both image classification and semantic segmentation. Our code is released at https://github.com/yichen928/ActiveFT.
updated: Sat Mar 25 2023 07:17:03 GMT+0000 (UTC)
published: Sat Mar 25 2023 07:17:03 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アソシエイト