べき乗変換と符号保存スコア集約による大規模言語モデルの構造的プルーニング Structured Pruning of Large Language Models via Power Transformation and Sign-Preserving Score Aggregation with Adaptive Feature Retention
本研究はLLMの構造的プルーニングに際し、べき乗変換と符号保存スコア集約を組み合わせた新手法を提案し、精度を維持しながら効率的なモデル圧縮を実現する。
English summary
- This paper proposes a structured pruning method for LLMs combining power transformation and sign-preserving score aggregation with adaptive feature retention, enabling more accurate compression with less performance degradation.
大規模言語モデル(LLM)の巨大化が進むなか、精度をできるだけ損なわずにモデルを軽量化する「プルーニング(枝刈り)」技術への注目が高まっている。本稿で取り上げる研究は、べき乗変換と符号保存スコア集約を組み合わせた構造的プルーニングの新手法を提案し、従来より少ない性能劣化で効率的な圧縮を実現できるとしている。
プルーニングとは、モデル内の重要度が低いパラメータや構造を取り除いて計算量とメモリ使用量を削減する手法である。個々の重みを個別に削る非構造的プルーニングは高い圧縮率を得やすい一方、専用のハードウェアがないと実際の高速化につながりにくい。これに対し構造的プルーニングは、注意ヘッドや中間層のチャネルといったまとまった単位で削除するため、汎用的なGPU上でもそのまま推論速度の向上を得やすいという利点がある。
本手法の核となるのは、各構造の重要度を測る「スコア」の扱い方だ。まずべき乗変換によってスコアの分布を整え、極端に大きい値や小さい値の影響を調整することで、重要度の識別性を高めると見られる。さらに符号保存スコア集約では、複数の指標を統合する際に値の符号(正負)を保持することで、単純な絶対値の合算では失われがちな情報を残し、本当に不要な構造を見極めやすくする狙いがあると考えられる。加えて適応的な特徴保持により、モデルにとって重要な特徴を選択的に温存するという。
LLMの圧縮を巡っては、こうしたプルーニングのほかにも、重みの数値精度を落とす量子化や、大きなモデルの知識を小さなモデルへ移す蒸留など、さまざまなアプローチが並行して研究されている。プルーニング分野に限っても、SparseGPTやWanda、LLM-Pruner、SliceGPTなど多様な手法が提案されてきた。本研究はその系譜に連なるもので、スコアの前処理と集約という比較的軽量な工夫で精度維持を図る点に特徴がある。
こうした圧縮技術は、クラウドの推論コスト削減だけでなく、スマートフォンやエッジ機器といった資源の限られた環境でLLMを動かすうえでも重要性を増している。ただし、提案手法の有効性がどの程度のモデル規模やタスクにまで一般化するかは、今後の追試や実装公開を通じた検証が求められるだろう。
Large language models keep growing, and with them the cost of running inference, so techniques that shrink a trained network without retraining it from scratch have become an active research front. A new paper posted to arXiv proposes a structured pruning method for LLMs that combines a power transformation of importance scores with what the authors call sign-preserving score aggregation, along with adaptive feature retention. The stated goal is to remove parameters more aggressively while keeping accuracy degradation smaller than existing approaches, which matters because memory footprint, latency, and energy consumption remain the practical barriers to deploying capable models on constrained hardware.
Pruning generally comes in two flavors, and the distinction is important for understanding what this work targets. Unstructured pruning zeroes out individual weights, producing sparse matrices that can be very accurate but that mainstream GPUs and inference runtimes struggle to accelerate without specialized kernels. Structured pruning instead removes whole components such as attention heads, neurons, channels, or intermediate dimensions, yielding a smaller dense model that runs faster on standard hardware. The trade-off is that removing coarse-grained structures tends to hurt accuracy more, so the central challenge is deciding which structures are least important and can be dropped with minimal impact.
That decision hinges on importance scoring, and this is where the paper's contributions sit. Most pruning schemes assign each weight or unit a saliency score, often derived from weight magnitudes, activations, or gradient-based sensitivity, and then aggregate those scores across a dimension to rank entire structures for removal. The authors argue that naive aggregation can be distorted by the shape of the score distribution and by cancellation between positive and negative contributions. Applying a power transformation reshapes the distribution of scores before aggregation, which appears intended to reduce the dominance of a few outliers and produce more reliable rankings. Sign-preserving aggregation is meant to retain information about the direction of each contribution rather than collapsing everything into magnitudes, so that units whose effects would otherwise cancel out are not misjudged as unimportant. Adaptive feature retention rounds out the approach by keeping a variable set of features rather than pruning uniformly across the network.
The method belongs to a lineage of recent LLM-specific pruning work. LLM-Pruner popularized gradient-based structured pruning followed by lightweight recovery fine-tuning, while SliceGPT reduced the embedding dimension through a transformation of the weight matrices. On the unstructured and semi-structured side, SparseGPT and Wanda demonstrated that one-shot pruning using calibration data could reach high sparsity without full retraining, with Wanda in particular showing that combining weight magnitude with input activation norms is a strong and cheap saliency signal. The new paper's emphasis on how scores are transformed and combined can be read as a refinement of this scoring-and-aggregation step rather than a wholesale departure from it.
It is also worth situating pruning within the broader model-compression toolkit, since practitioners rarely rely on a single technique. Quantization, which lowers the numerical precision of weights and activations to 8-bit, 4-bit, or lower, is often the first lever pulled because it is well supported by libraries such as GPTQ, AWQ, and bitsandbytes. Knowledge distillation trains a smaller student model to imitate a larger teacher. These methods are largely complementary to pruning, and a compressed deployment pipeline may stack them, for example by pruning structurally and then quantizing the survivor. Structured pruning is attractive precisely because it produces a smaller dense model that composes cleanly with these other steps.
As with most preprints in this area, the claims should be read with appropriate caution until they are independently reproduced. Pruning results are sensitive to the choice of calibration data, the sparsity level, the model family, and whether any recovery fine-tuning is applied, and gains reported on perplexity or a handful of benchmarks do not always transfer to downstream tasks or to larger models. The reported combination of power transformation and sign-preserving aggregation is likely most valuable at moderate to high pruning ratios, where ranking errors are costliest. Readers interested in the specifics of the transformation, the calibration setup, and the evaluated model sizes should consult the paper directly, as those details determine how broadly the improvements are likely to hold.
本ページの本文・要約は AI による自動生成です。正確性は元記事 (arxiv.org) をご確認ください。