English

Cascaded Multi-Granularity Pruning for On-Device LLM Inference in Industrial IoT

Computation and Language 2026-06-25 v1

Abstract

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.

Keywords

Cite

@article{arxiv.2606.26861,
  title  = {Cascaded Multi-Granularity Pruning for On-Device LLM Inference in Industrial IoT},
  author = {Jinghan Wang and Yanjun Chen and Wei Zhang and Xiaotong Huang and Tianchen Liu and Gaoliang Peng},
  journal= {arXiv preprint arXiv:2606.26861},
  year   = {2026}
}

Comments

This work has been submitted to the IEEE Internet of Things Journal for possible publication