言語モデルがパラメーターとレイヤーで成長するにつれて、単一の GPU でそれらをトレーニングして推論することは非常に難しくなっています。これにより、GPT-3、BERT-Large、その他多くの大きな言語モデルの可用性が大幅に制限されています。この問題を解決するための一般的な手法は、トランス ヘッド、完全に接続されたウェイト、およびその他のモジュールを削除して、ネットワーク アーキテクチャを整理することです。主な課題は、重要なパラメーターと重要でないパラメーターを区別することです。私たちの目標は、そのようなパラメーターを特定するための強力な指標を見つけることです。したがって、重要度スコアを計算するために、GradCAM 解釈に基づく Cam-Cut と SmoothGrad に基づく Smooth-Cut という 2 つの戦略を提案します。この作業を通じて、スコアリング関数がより関連性の高いタスクベースのスコアをネットワークパラメーターに割り当てることができることを示します。したがって、特にBERTのより高い圧縮率で、両方のプルーニングアプローチが標準の重みおよび勾配ベースの戦略よりも大幅に優れています。ベースモデル。また、プルーニング マスクを分析したところ、標準的なメトリックを使用して取得したマスクとは大きく異なることがわかりました。
As language models have grown in parameters and layers, it has become much harder to train and infer with them on single GPUs. This is severely restricting the availability of large language models such as GPT-3, BERT-Large, and many others. A common technique to solve this problem is pruning the network architecture by removing transformer heads, fully-connected weights, and other modules. The main challenge is to discern the important parameters from the less important ones. Our goal is to find strong metrics for identifying such parameters. We thus propose two strategies: Cam-Cut based on the GradCAM interpretations, and Smooth-Cut based on the SmoothGrad, for calculating the importance scores. Through this work, we show that our scoring functions are able to assign more relevant task-based scores to the network parameters, and thus both our pruning approaches significantly outperform the standard weight and gradient-based strategies, especially at higher compression ratios in BERT-based models. We also analyze our pruning masks and find them to be significantly different from the ones obtained using standard metrics.