arXiv reaDer
分布マッチングによるデータセット凝縮
Dataset Condensation with Distribution Matching
多くの学習問題で最先端のディープモデルをトレーニングするための計算コストは、より洗練されたモデルとより大きなデータセットのために急速に増加しています。トレーニングコストを削減するための最近の有望な方向性は、元の情報を保持しながら、元の大きなトレーニングセットを大幅に小さい学習済み合成セットに置き換えることを目的としたデータセットの凝縮です。凝縮された画像の小さなセットで深いモデルをトレーニングすることは非常に高速ですが、複雑な2レベルの最適化と二階導関数の計算により、それらの合成は計算コストが高くなります。この作業では、多くのサンプリングされた埋め込みスペースで合成と元のトレーニング画像の特徴分布を一致させることにより、凝縮された画像を合成するシンプルで効果的な方法を提案します。私たちの方法は、同等以上のパフォーマンスを達成しながら、合成コストを大幅に削減します。その効率のおかげで、洗練されたニューラルアーキテクチャを備えたより現実的でより大きなデータセットにこの方法を適用し、パフォーマンスを大幅に向上させます。また、継続的な学習とニューラルアーキテクチャの検索における私たちの方法の有望な実用的な利点を示しています。
Computational cost of training state-of-the-art deep models in many learning problems is rapidly increasing due to more sophisticated models and larger datasets. A recent promising direction for reducing training cost is dataset condensation that aims to replace the original large training set with a significantly smaller learned synthetic set while preserving the original information. While training deep models on the small set of condensed images can be extremely fast, their synthesis remains computationally expensive due to the complex bi-level optimization and second-order derivative computation. In this work, we propose a simple yet effective method that synthesizes condensed images by matching feature distributions of the synthetic and original training images in many sampled embedding spaces. Our method significantly reduces the synthesis cost while achieving comparable or better performance. Thanks to its efficiency, we apply our method to more realistic and larger datasets with sophisticated neural architectures and obtain a significant performance boost. We also show promising practical benefits of our method in continual learning and neural architecture search.
updated: Thu Dec 22 2022 02:50:07 GMT+0000 (UTC)
published: Fri Oct 08 2021 15:02:30 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アソシエイト