about
TrimLLM: Progressive Layer Dropping for Domain-Specific LLMs (arxiv.org)
2 points by pulkitsh1234 on Jul 29, 2025 | hide | past | pdf | discuss on HN

In plain words: TrimLLM makes a language model shallower by gradually removing layers while it is tuned for one field, since layers specialize. It ran 2.1 to 5.7 times faster on everyday graphics cards than the best compression tools, with no accuracy loss at 50-60% compression.

Abstract

Specializing large language models (LLMs) for local deployment in domain-specific use cases is necessary for strong performance while meeting latency and privacy constraints. However, conventional task-specific adaptation approaches do not show simultaneous memory saving and inference speedup at deployment time. Practical compression techniques like quantization and pruning require dedicated hardware or kernel support to achieve measured inference speedup. We develop TrimLLM based on the layer-wise specialization phenomenon we empirically observed and verified on contemporary LLMs. TrimLLM reduces the depth of LLMs via progressive layer dropping. We show it retains LLMs' capacity in specific domains and achieves inference speedup irrespective of hardware and deep learning frameworks. We evaluated TrimLLM on LLMs of various sizes for inference; models adapted on medical, legal, and financial datasets all demonstrate $2.1-5.7\times$ inference speedup on consumer GPUs and up to $3.1\times$ speedup on A100 when compared to state-of-the-art model compression algorithms, with no loss in accuracy at 50$\sim$60\% model compression ratio.

Lanxiang Hu, Tajana Rosing, Hao Zhang
arXiv:2412.11242 · cs.LG, cs.AI, cs.CL · submitted Dec 15, 2024 · updated Dec 19, 2024
abstract · pdf · html

add comment on HN