MobileNet V3
9 memberslucid.models.vision.mobilenet_v3MobileNet v3 family — Howard et al., 2019.
Howard, Andrew, et al. "Searching for MobileNetV3." Proceedings of the IEEE/CVF International Conference on Computer Vision, 2019, pp. 1314–1324.
MobileNet-v3 is the result of an explicit neural architecture search layered on top of the inverted-residual block of v2. A platform-aware NAS (MnasNet-style) selects the per-stage expansion ratio, kernel size, and channel count to minimise a latency-and-accuracy joint objective on a real mobile CPU, after which a NetAdapt pass fine-tunes the channel counts of each layer individually under a hard latency budget. The output is two hand-tuned variants — Large and Small — targeting different parts of the latency/accuracy frontier.
Two architectural refinements give v3 its accuracy gain over v2. First, a lightweight squeeze-and-excitation module is inserted inside selected bottleneck blocks; it pools the spatial dimensions and re-weights the channel responses, providing a cheap channel-attention mechanism. Second, a new activation function called hard-swish,
replaces swish/SiLU in the deeper half of the network. Hard-swish matches swish closely in shape but uses only piecewise-linear primitives, making it dramatically cheaper to evaluate on mobile-friendly fixed-point hardware.
A further trick is rewriting the expensive final layers of v2. The expansion before the global average pool is moved after the pool, where it operates on a single spatial location, saving roughly of total latency. Combined, these changes give MobileNet-v3-Large higher ImageNet top-1 accuracy than v2 at lower latency on a Pixel-1 CPU.
Classes
Functions
mobilenet_v3_large→ MobileNetV3MobileNet-v3-Large feature-extracting backbone.
mobilenet_v3_large_cls→ MobileNetV3ForImageClassificationMobileNet-v3-Large image classifier with the redesigned head.
mobilenet_v3_small→ MobileNetV3MobileNet-v3-Small feature-extracting backbone.
mobilenet_v3_small_cls→ MobileNetV3ForImageClassificationMobileNet-v3-Small image classifier with the redesigned head.