4-bit量化让深度模型在边缘设备实时运行
QuartDepth: Post-Training Quantization for Real-Time Depth Estimation on the Edge
- 4bit量化权重与激活,压缩模型体积并降低计算开销
- 引入激活精修与权重重建,误差下降12.3%且精度接近原模型
- 适配ASIC的可编程加速器,支持算子融合与高效推理
单目深度估计(MDE)在计算机视觉中至关重要,但部署高精度模型于资源受限的边缘设备(尤其是专用集成电路ASIC)面临巨大挑战。现有基础模型虽性能优异,却加剧了硬件部署难度。本文提出QuartDepth,采用后训练量化技术对MDE模型进行4位量化,同时设计支持算子融合与自定义指令的可编程硬件加速器。为缓解量化带来的性能损失,引入激活精修与补偿算法,并提出权重重建方法以减小量化误差。实验表明,该框架在ASIC上实现接近原始模型的精度,同时推理速度更快、能效更高,有效弥合高性能深度估计与边缘实用性的差距。
原文摘要 · Abstract (English)
Monocular Depth Estimation (MDE) has emerged as a pivotal task in computer vision, supporting numerous real-world applications. However, deploying accurate depth estimation models on resource-limited edge devices, especially Application-Specific Integrated Circuits (ASICs), is challenging due to the high computational and memory demands. Recent advancements in foundational depth estimation deliver impressive results but further amplify the difficulty of deployment on ASICs. To address this, we propose QuartDepth which adopts post-training quantization to quantize MDE models with hardware accelerations for ASICs. Our approach involves quantizing both weights and activations to 4-bit precision, reducing the model size and computation cost. To mitigate the performance degradation, we introduce activation polishing and compensation algorithm applied before and after activation quantization, as well as a weight reconstruction method for minimizing errors in weight quantization. Furthermore, we design a flexible and programmable hardware accelerator by supporting kernel fusion and customized instruction programmability, enhancing throughput and efficiency. Experimental results demonstrate that our framework achieves competitive accuracy while enabling fast inference and higher energy efficiency on ASICs, bridging the gap between high-performance depth estimation and practical edge-device applicability. Code: https://github.com/shawnricecake/quart-depth
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。