Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Run, Don't Walk: Chasing Higher FLOPS for Faster Neural Networks

About

To design fast neural networks, many works have been focusing on reducing the number of floating-point operations (FLOPs). We observe that such reduction in FLOPs, however, does not necessarily lead to a similar level of reduction in latency. This mainly stems from inefficiently low floating-point operations per second (FLOPS). To achieve faster networks, we revisit popular operators and demonstrate that such low FLOPS is mainly due to frequent memory access of the operators, especially the depthwise convolution. We hence propose a novel partial convolution (PConv) that extracts spatial features more efficiently, by cutting down redundant computation and memory access simultaneously. Building upon our PConv, we further propose FasterNet, a new family of neural networks, which attains substantially higher running speed than others on a wide range of devices, without compromising on accuracy for various vision tasks. For example, on ImageNet-1k, our tiny FasterNet-T0 is $2.8\times$, $3.3\times$, and $2.4\times$ faster than MobileViT-XXS on GPU, CPU, and ARM processors, respectively, while being $2.9\%$ more accurate. Our large FasterNet-L achieves impressive $83.5\%$ top-1 accuracy, on par with the emerging Swin-B, while having $36\%$ higher inference throughput on GPU, as well as saving $37\%$ compute time on CPU. Code is available at \url{https://github.com/JierunChen/FasterNet}.

Jierun Chen, Shiu-hong Kao, Hao He, Weipeng Zhuo, Song Wen, Chul-Ho Lee, S.-H. Gary Chan• 2023

Related benchmarks

TaskDatasetResultRank
Object DetectionCOCO 2017 (val)--
2454
Instance SegmentationCOCO 2017 (val)
APm0.399
1144
Oriented Object DetectionDOTA v1.0 (test)--
378
Object DetectionCOCO 2017
AP (Box)39.9
279
Instance SegmentationCOCO 2017
APm36.9
199
Image ClassificationImageNet-1k 1.0 (test)
Top-1 Accuracy81.3
197
Image ClassificationImageNet-1k 1.0 (test)
Top-1 Accuracy0.789
191
Image ClassificationImageNet-C (val)
mCE70.8
97
Image ClassificationImageNet-R (val)
Accuracy40.5
82
Oriented Object DetectionDOTA v1.5 (test)
mAP71.07
58
Showing 10 of 20 rows

Other info

Code

Follow for update