提出可训练的位级软量化层,实现输入特征压缩以降低边缘设备数据传输量。
Trainable Bitwise Soft Quantization for Input Feature Compression
- 用sigmoid近似阶梯函数,实现可训练的量化阈值。
- 在多个数据集上实现5到16倍的压缩比,性能接近32位全精度模型。
- 适合资源受限的物联网场景,尤其适用于需远程计算的边缘设备。
物联网环境下机器学习应用对算力和内存资源的需求日益增长,亟需新方法优化资源使用。尽管模型压缩与效率提升已取得显著进展,许多应用仍依赖远程服务器提供计算资源。然而,将边缘设备数据传至远程服务器可能受限于带宽、延迟或能耗。本文提出一种面向任务的可训练特征量化层,用于压缩神经网络的输入特征,显著减少设备向远程服务器传输的数据量。该层允许每个输入特征被量化至用户定义的比特数,实现在数据采集时的简单本地压缩。通过用sigmoid逼近阶跃函数,实现可训练的量化阈值;通过拼接多个sigmoid输出(即位级软量化),在集成神经网络时获得可训练的量化值。实验表明,本方法优于标准量化基线,在保持接近全精度模型性能的同时,相比32位输入可实现5×至16×的压缩比。
原文摘要 · Abstract (English)
The growing demand for machine learning applications in the context of the Internet of Things calls for new approaches to optimize the use of limited compute and memory resources. Despite significant progress that has been made w.r.t. reducing model sizes and improving efficiency, many applications still require remote servers to provide the required resources. However, such approaches rely on transmitting data from edge devices to remote servers, which may not always be feasible due to bandwidth, latency, or energy constraints. We propose a task-specific, trainable feature quantization layer that compresses the input features of a neural network. This can significantly reduce the amount of data that needs to be transferred from the device to a remote server. In particular, the layer allows each input feature to be quantized to a user-defined number of bits, enabling a simple on-device compression at the time of data collection. The layer is designed to approximate step functions with sigmoids, enabling trainable quantization thresholds. By concatenating outputs from multiple sigmoids, introduced as bitwise soft quantization, it achieves trainable quantized values when integrated with a neural network. We compare our method to full-precision inference as well as to several quantization baselines. Experiments show that our approach outperforms standard quantization methods, while maintaining accuracy levels close to those of full-precision models. In particular, depending on the dataset, compression factors of $5\times$ to $16\times$ can be achieved compared to $32$-bit input without significant performance loss.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。