Search papers, labs, and topics across Lattice.
This paper introduces a cascaded multi-granularity pruning framework designed to enhance the deployment of large language models (LLMs) on Industrial Internet of Things (IIoT) edge devices by addressing the limitations of existing structured pruning methods at high compression ratios. By systematically removing layers, attention heads, and feed-forward channels while employing low-rank recovery for importance re-estimation, the framework achieves a remarkable 13.8 times compression on Multi-Head Attention (MHA)+GELU architectures, improving accuracy by 3.70 percentage points over the strongest baseline. In practical applications, this approach significantly reduces inference latency by up to 67.2% and peak memory usage by 62.5%, proving its effectiveness for real-world IIoT scenarios.
Achieving 13.8x compression on LLMs while improving accuracy by 3.70 percentage points could redefine on-device inference for IIoT applications.
Deploying large language models (LLMs) on Industrial Internet of Things (IIoT) edge devices demands extreme compression, yet existing structured pruning methods collapse at high compression ratios due to one-shot importance estimation, and their cross-architecture behavior remains unpredictable. This article presents a cascaded multi-granularity pruning framework that removes layers, attention heads, and feed-forward channels in coarse-to-fine order, with lightweight low-rank recovery between stages to re-estimate component importance. An information-theoretic analysis motivates this ordering, and the Structural Independence Assumption (SIA) is formalized as a checkable condition predicting whether per-component pruning criteria are reliable for a given architecture: Multi-Head Attention (MHA)+GELU designs satisfy the SIA, whereas Grouped Query Attention (GQA)+SwiGLU designs violate it. On bearing fault diagnosis spanning 88M to 6.25B-parameter models, the framework extends achievable compression to 13.8 times on MHA+GELU architectures with 83.82% accuracy (+3.70 percentage points (pp) over the strongest baseline), while exposing a ~74pp accuracy collapse on GQA+SwiGLU architectures that violate the SIA. Deployed on an industrial slewing bearing fault diagnosis platform with NVIDIA DGX Spark, compressed models reduce inference latency by up to 67.2% and peak memory by 62.5%, demonstrating viability for IIoT edge inference.
