Aug, 2024
为设备视觉任务设计极为节省内存的卷积神经网络
Designing Extremely Memory-Efficient CNNs for On-device Vision Tasks
TL;DR本研究解决了资源受限的低端嵌入式和物联网设备在图像分类和对象检测等视觉任务中内存不足的问题。通过提出输入分割、补丁隧道和瓶颈重排等三种设计原则,显著降低了卷积神经网络的内存峰值使用,使其能够在仅63 KB的内存下实现竞争力的准确率(61.58%)。这一成果显示出该网络在内存使用上远小于现有的高效网络,具有重要的应用潜力。