HarDNet: A Low Memory Traffic Network
About
State-of-the-art neural network architectures such as ResNet, MobileNet, and DenseNet have achieved outstanding accuracy over low MACs and small model size counterparts. However, these metrics might not be accurate for predicting the inference time. We suggest that memory traffic for accessing intermediate feature maps can be a factor dominating the inference latency, especially in such tasks as real-time object detection and semantic segmentation of high-resolution video. We propose a Harmonic Densely Connected Network to achieve high efficiency in terms of both low MACs and memory traffic. The new network achieves 35%, 36%, 30%, 32%, and 45% inference time reduction compared with FC-DenseNet-103, DenseNet-264, ResNet-50, ResNet-152, and SSD-VGG, respectively. We use tools including Nvidia profiler and ARM Scale-Sim to measure the memory traffic and verify that the inference latency is indeed proportional to the memory traffic consumption and the proposed network consumes low memory traffic. We conclude that one should take memory traffic into consideration when designing neural network architectures for high-resolution applications at the edge.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Semantic segmentation | Cityscapes (test) | mIoU76 | 1145 | |
| Object Detection | PASCAL VOC 2007 (test) | mAP81.5 | 821 | |
| Object Detection | COCO v2017 (test-dev) | mAP36.8 | 499 | |
| Image Classification | ImageNet (val) | Top-1 Accuracy77.8 | 354 | |
| Semantic segmentation | Cityscapes (val) | mIoU77.4 | 287 | |
| Image Classification | ImageNet (test) | -- | 235 | |
| Semantic segmentation | Trans10K v2 (test) | mIoU56.19 | 104 | |
| Medical Image Segmentation | GLAS | Dice89.37 | 28 | |
| Semantic segmentation | Trans10K v2 | Accuracy90.19 | 27 | |
| Medical Image Segmentation | MoNu | Dice79.52 | 15 |