arXiv:2510.13760eess.IV2025-10中稿 · 2025 IEEE/ACM Inte…

将视觉Transformer压缩至三值量化,在边缘设备上实现医疗影像高效分析

Invited Paper: BitMedViT: Ternary-Quantized Vision Transformer for Medical AI Assistants on the Edge

  • 采用三值量化线性层与多查询注意力,保持模型稳定
  • 在MedMNIST上达86%准确率,模型大小缩小43倍
  • 适合医疗边缘设备部署,能效比超现有模型41倍

视觉变压器(ViTs)在解析复杂医学影像数据方面表现优异,但其高计算和内存需求限制了在临床环境中资源受限的移动与可穿戴设备上的实时部署。我们提出BiTMedViT,一种新型边缘视觉变压器,作为医疗AI助手,可在边缘端直接对医学图像进行结构化分析。该模型采用针对医学影像定制的三值量化线性层,并结合多查询注意力训练机制,在低精度激活下保持稳定性。此外,通过任务感知蒸馏从高容量教师模型恢复因极端量化损失的精度。最后,我们构建了一个将三值化ViT映射到自定义CUDA内核的流水线,提升了Jetson Orin Nano上的内存带宽利用率和降低延迟。最终,BiTMedViT在MedMNIST跨12个数据集上达到86%诊断准确率(89% SOTA),模型尺寸缩小43倍,内存流量减少39倍,推理延迟降至16.8毫秒,能效高达41倍于现有最优模型,在183.62 GOPs/J下运行。结果表明,这为极端精度的医学影像ViT在边缘部署提供了可行且科学的路径,缩小了算法进展与可部署临床工具之间的差距。

原文摘要 · Abstract (English)

Vision Transformers (ViTs) have demonstrated strong capabilities in interpreting complex medical imaging data. However, their significant computational and memory demands pose challenges for deployment in real-time, resource-constrained mobile and wearable devices used in clinical environments. We introduce, BiTMedViT, a new class of Edge ViTs serving as medical AI assistants that perform structured analysis of medical images directly on the edge. BiTMedViT utilizes ternary- quantized linear layers tailored for medical imaging and com- bines a training procedure with multi-query attention, preserving stability under ternary weights with low-precision activations. Furthermore, BiTMedViT employs task-aware distillation from a high-capacity teacher to recover accuracy lost due to extreme quantization. Lastly, we also present a pipeline that maps the ternarized ViTs to a custom CUDA kernel for efficient memory bandwidth utilization and latency reduction on the Jetson Orin Nano. Finally, BiTMedViT achieves 86% diagnostic accuracy (89% SOTA) on MedMNIST across 12 datasets, while reducing model size by 43x, memory traffic by 39x, and enabling 16.8 ms inference at an energy efficiency up to 41x that of SOTA models at 183.62 GOPs/J on the Orin Nano. Our results demonstrate a practical and scientifically grounded route for extreme-precision medical imaging ViTs deployable on the edge, narrowing the gap between algorithmic advances and deployable clinical tools.

边缘计算三值量化医疗AIViT

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