arXiv reaDer
DiSparse:マルチタスクモデル圧縮のための解きほぐされたスパース化
DiSparse: Disentangled Sparsification for Multitask Model Compression
モデル圧縮とマルチタスク学習の人気にもかかわらず、マルチタスクモデルを効果的に圧縮する方法は、パラメーター空間でのタスクの絡み合いが難しいため、十分に分析されていません。この論文では、DiSparseを提案します。これは、シンプルで効果的で、他に類を見ないマルチタスクプルーニングおよびスパーストレーニングスキームです。重要度の測定を解きほぐすことにより、各タスクを個別に検討し、パラメーターのプルーニングと選択を実行するときに、すべてのタスク間で全会一致の決定を行います。私たちの実験結果は、一般的なスパーストレーニングおよび剪定方法と比較して、さまざまな構成および設定で優れたパフォーマンスを示しています。圧縮の効果に加えて、DiSparseはマルチタスク学習コミュニティに強力なツールも提供します。驚いたことに、DiSparseによってモデルのスパース性が高いにもかかわらず、いくつかのケースでは、いくつかの専用マルチタスク学習方法よりも優れたパフォーマンスが観察されました。 DiSparseで生成された剪定マスクを分析し、トレーニングが開始される前でも、各タスクによって識別された非常に類似したスパースネットワークアーキテクチャを観察しました。また、タスクの関連性が急激に低下する「流域」レイヤーの存在も観察されます。これは、パラメーターの共有を継続してもメリットがないことを意味します。コードとモデルは、https://github.com/SHI-Labs/DiSparse-Multitask-Model-Compressionで入手できます。
Despite the popularity of Model Compression and Multitask Learning, how to effectively compress a multitask model has been less thoroughly analyzed due to the challenging entanglement of tasks in the parameter space. In this paper, we propose DiSparse, a simple, effective, and first-of-its-kind multitask pruning and sparse training scheme. We consider each task independently by disentangling the importance measurement and take the unanimous decisions among all tasks when performing parameter pruning and selection. Our experimental results demonstrate superior performance on various configurations and settings compared to popular sparse training and pruning methods. Besides the effectiveness in compression, DiSparse also provides a powerful tool to the multitask learning community. Surprisingly, we even observed better performance than some dedicated multitask learning methods in several cases despite the high model sparsity enforced by DiSparse. We analyzed the pruning masks generated with DiSparse and observed strikingly similar sparse network architecture identified by each task even before the training starts. We also observe the existence of a "watershed" layer where the task relatedness sharply drops, implying no benefits in continued parameters sharing. Our code and models will be available at: https://github.com/SHI-Labs/DiSparse-Multitask-Model-Compression.
updated: Thu Jun 09 2022 17:57:46 GMT+0000 (UTC)
published: Thu Jun 09 2022 17:57:46 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アソシエイト