arXiv:2509.25439cs.LGcs.NE2025-09中稿 · Asilomar 2025

提出Norm-Q方法,用8位量化压缩HMM模型无损,99%压缩率适配神经符号系统。

Norm-Q: Effective Compression Method for Hidden Markov Models in Neuro-Symbolic Applications

  • 设计归一化量化感知训练流程,降低概率模型计算开销。
  • 4096状态HMM可压缩至8位无损失,最多3位有可接受误差。
  • 适合部署在定制硬件的轻量级神经符号系统,开源代码可用。

隐马尔可夫模型(HMM)广泛应用于生成任务,在具有马尔可夫特性的神经符号系统中表现出色。但其与神经网络结合时会放大计算密集和数据传输瓶颈。本文提出Norm-Q,一种针对概率符号模型(如HMM)的归一化线性量化方法。通过最小化位宽损失,显著缓解内存与带宽压力,支持在专用硬件上部署。该方法引入归一化量化感知的期望最大化训练过程。实验表明,相比传统量化方法,Norm-Q实现更高压缩率且得分损失更小。在大语言模型的受限生成任务中,4096个隐藏状态的HMM可无损压缩至8位,最多3位时损失仍可接受。特别地,该方法对HMM权重的压缩率可达99%。代码已开源:https://github.com/superstarghy/Norm-Q。

原文摘要 · Abstract (English)

Hidden Markov models (HMM) are commonly used in generation tasks and have demonstrated strong capabilities in neuro-symbolic applications for the Markov property. These applications leverage the strengths of neural networks and symbolic reasoning to create robust and interpretable AI systems. However, they may inherit and amplify the shortcomings of both approaches. Both components require dense computation and data transfer, and their communication further hinders performance. This paper proposes Norm-Q, a normalized linear quantization approach for compressing probabilistic symbolic models, such as HMMs. We reduce the bit width of the data with minimal impact, thereby alleviating memory and bandwidth stress and enabling deployment on potential custom hardware. Our method introduces a normalized quantization-aware expectation maximization process for probabilistic model training. The experimental results show that Norm-Q achieves a higher compression rate with reasonable score loss compared to traditional quantization methods. In the case of the constrained generation task of large language models, we successfully quantize an HMM of 4096 hidden states to 8 bits without loss and, at most, 3 bits with acceptable loss. Notably, the Norm-Q method can achieve a compression rate of 99% for the weights of the HMM. The code is open source at https://github.com/superstarghy/Norm-Q.

模型压缩隐马尔可夫量化神经符号

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。