arXiv:2503.08719eess.IVcs.CV2025-03被引 3

将U-Net量化至4.24比特,实现低功耗可穿戴设备上的实时肿瘤分割。

QuantU-Net: Efficient Wearable Medical Imaging Using Bitwidth as a Trainable Parameter

  • 把模型精度设为可训练参数,用混合损失函数联合优化分割效果与模型大小。
  • 量化后模型仅4.24比特平均精度,准确率94.25%,比浮点基线低1.89%。
  • 训练时间从理论上数万亿次压缩到一次,适合嵌入FPGA等边缘硬件。

医学图像分割,尤其是肿瘤分割,在医学影像中至关重要,而U-Net是该任务广泛采用的卷积神经网络架构。然而,其高计算和内存需求限制了在资源受限设备(如可穿戴系统)上的部署。本文提出QuantU-Net,一种针对低功耗设备(如现场可编程门阵列FPGA)优化的量化U-Net版本。基于Brevitas框架,通过量化感知训练将模型精度降至平均4.24比特,验证准确率达94.25%,仅比浮点基线低1.89%。量化后模型体积减少约8倍,适合可穿戴设备的实时应用。采用结合二元交叉熵损失、Dice损失与比特宽度损失的自定义损失函数,将寻找最优比特宽度与准确率组合的训练次数从假设的6^23次压缩至单次训练完成。模型使用整数运算,凸显其在FPGA等专用AI加速器上的部署潜力。本工作推动了医学图像分割在资源受限设备上的应用,为可穿戴医疗中的实时低功耗诊断开辟新路径。

原文摘要 · Abstract (English)

Medical image segmentation, particularly tumor segmentation, is a critical task in medical imaging, with U-Net being a widely adopted convolutional neural network (CNN) architecture for this purpose. However, U-Net's high computational and memory requirements pose challenges for deployment on resource-constrained devices such as wearable medical systems. This paper addresses these challenges by introducing QuantU-Net, a quantized version of U-Net optimized for efficient deployment on low-power devices like Field-Programmable Gate Arrays (FPGAs). Using Brevitas, a PyTorch library for quantization-aware training, we quantize the U-Net model, reducing its precision to an average of 4.24 bits while maintaining a validation accuracy of 94.25%, only 1.89% lower than the floating-point baseline. The quantized model achieves an approximately 8x reduction in size, making it suitable for real-time applications in wearable medical devices. We employ a custom loss function that combines Binary Cross-Entropy (BCE) Loss, Dice Loss, and a bitwidth loss function to optimize both segmentation accuracy and the size of the model. Using this custom loss function, we have significantly reduced the training time required to find an optimal combination of bitwidth and accuracy from a hypothetical 6^23 number of training sessions to a single training session. The model's usage of integer arithmetic highlights its potential for deployment on FPGAs and other designated AI accelerator hardware. This work advances the field of medical image segmentation by enabling the deployment of deep learning models on resource-constrained devices, paving the way for real-time, low-power diagnostic solutions in wearable healthcare applications.

医学影像模型量化可穿戴设备FPGA

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