arXiv reaDer
分布マッチングによるデータセットの凝縮
Dataset Condensation with Distribution Matching
多くの学習問題で最先端のディープモデルをトレーニングするための計算コストは、より洗練されたモデルとより大きなデータセットのために急速に増加しています。トレーニング時間を短縮するための最近の有望な方向性は、情報を保持しながら、元の大きなトレーニングセットを大幅に小さい学習済み合成セットに置き換えることを目的としたデータセットの凝縮です。凝縮された画像の小さなセットで深いモデルをトレーニングすることは非常に高速ですが、複雑な2レベルの最適化と二次微分計算のため、それらの合成は計算コストが高くなります。この作業では、サンプリングされた埋め込みスペース内の合成および元のトレーニング画像の特徴分布を一致させることにより、同等のパフォーマンスで大幅に低いトレーニングコストを必要とするシンプルで効果的なデータセット凝縮手法を提案します。その効率のおかげで、洗練されたニューラルアーキテクチャを備えたより現実的でより大きなデータセットに私たちの方法を適用し、より大きな合成トレーニングセットを使用しながら大幅なパフォーマンスの向上を実現します。また、継続的な学習とニューラルアーキテクチャの検索における私たちの方法のさまざまな実用的な利点を示します。
Computational cost to train 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 to reduce training time is dataset condensation that aims to replace the original large training set with a significantly smaller learned synthetic set while preserving its 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 dataset condensation technique that requires significantly lower training cost with comparable performance by matching feature distributions of the synthetic and original training images in sampled embedding spaces. Thanks to its efficiency, we apply our method to more realistic and larger datasets with sophisticated neural architectures and achieve a significant performance boost while using larger synthetic training set. We also show various practical benefits of our method in continual learning and neural architecture search.
updated: Fri Oct 08 2021 15:02:30 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アソシエイト