Structured Pruning of Large Language Models via Power Transformation and Sign-Preserving Score Aggregation with Adaptive Feature Retention

· Source: Artificial Intelligence · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Software Development & Engineering · Depth: Expert, quick

Summary

A new structured pruning method for large language models (LLMs) has been developed to overcome challenges encountered when adapting Adaptive Feature Retention (AFR) from unstructured pruning. This technique specifically addresses three key issues: distribution mismatch among heterogeneous pruning scores, the loss of crucial sign information indicating optimization direction, and the impact of outlier values. The proposed unified approach integrates power transformation for nonlinear distribution alignment, sign-preserving score aggregation, and percentile-based outlier removal. Experimental evaluations on Llama-3-8B, Vicuna-v1.5-13B, and LLaVA-v1.5-13B demonstrate that this method achieves accuracy comparable to unstructured pruning while delivering practical inference speedups through structured pruning.

Key takeaway

For Machine Learning Engineers optimizing LLM deployment, this structured pruning method offers a viable path to significant inference speedups without sacrificing model accuracy. If you are struggling with adapting unstructured pruning techniques like AFR to structured settings, consider implementing power transformation, sign-preserving score aggregation, and percentile-based outlier removal. This approach allows you to achieve practical performance gains on models like Llama-3-8B, making your deployments more efficient.

Key insights

Improved structured pruning for LLMs maintains accuracy while enabling practical inference speedups by addressing AFR adaptation challenges.

Principles

Method

The method combines power transformation for score alignment, sign-preserving score aggregation, and percentile-based outlier removal to adapt AFR for structured LLM pruning.

In practice

Topics

Best for: AI Engineer, Research Scientist, AI Scientist, Machine Learning Engineer, NLP Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.