VeloxNet: Efficient Spatial Gating for Lightweight Embedded Image Classification
Organizations: Canizaro Livingston Gulf States Center for Environmental Informatics, the University of New Orleans, New Orleans, USA · Center for Geospatial Sciences, Naval Research Laboratory, Stennis Space Center, Hancock County, Mississippi, USA · US Army Corps of Engineers, Engineer Research and Development Center, Vicksburg, Mississippi, USA
Abstract
Deploying deep learning models on embedded devices for tasks such as aerial disaster monitoring and infrastructure inspection requires architectures that balance accuracy with strict constraints on model size, memory, and latency. This paper introduces VeloxNet, a lightweight CNN architecture that replaces SqueezeNet's fire modules with gated multi-layer perceptron (gMLP) blocks for embedded image classification. Each gMLP block uses a spatial gating unit (SGU) that applies learned spatial projections and multiplicative gating, enabling the network to capture spatial dependencies across the full feature map in a single layer. Unlike fire modules, which are limited to local receptive fields defined by small convolutional kernels, the SGU provides global spatial modeling at each layer with fewer parameters. We evaluate VeloxNet on three aerial image datasets: the Aerial Image Database for Emergency Response (AIDER), the Comprehensive Disaster Dataset (CDD), and the Levee Defect Dataset (LDD), comparing against eleven baselines including MobileNet variants, ShuffleNet, EfficientNet, and recent vision transformers. VeloxNet reduces the parameter count by 46.1% relative to SqueezeNet (from 740,970 to 399,366) while improving weighted F1 scores by 6.32% on AIDER, 30.83% on CDD, and 2.51% on LDD. These results demonstrate that substituting local convolutional modules with spatial gating blocks can improve both classification accuracy and parameter efficiency for resource-constrained deployment. The source code will be made publicly available upon acceptance of the paper.
Figures & tables
| SqueezeNet | VeloxNet | ||||
| Layer name/type | Output size | Number of parameters | Layer name/type | Output size | Number of parameters |
| input image | - | input image | - | ||
| conv1 (7 7) | 14,112 | conv1 (3 3) | 4,212 | ||
| batchnorm1 | 192 | batchnorm1 | 6 | ||
| maxpool1 | - | maxpool1 | - | ||
| fire2 | 12,064 | gMLP2 | 49,296 | ||
| Dataset | Class | Train | Validation | Testing | Total per Class |
|---|---|---|---|---|---|
| AIDER | Collapsed Building | 367 | 41 | 103 | 511 |
| Fire | 249 | 63 | 209 | 521 | |
| Flood | 252 | 63 | 211 | 526 | |
| Traffic | 232 | 59 | 194 | 485 | |
| Normal | 2107 | 527 | 1756 | 4390 | |
| Total Per Set | 3207 | 753 | 2473 | 6433 |
| Methods | F1 | Precision | Recall | Parameters | FLOPs | Size | FPS |
|---|---|---|---|---|---|---|---|
| (%) | (%) | (%) | ( ) | (M) | (MB) | ( ) | |
| MobileNetV1 | 73.13 | 74.13 | 73.20 | 3,212,101 | 588 | 12.34 | 411.30 |
| MobileNetV2 | 72.94 | 73.56 | 73.01 | 2,230,277 | 326 | 8.64 | 213.48 |
| MobileNetV3 | 63.96 | 65.46 | 63.88 | 3,660,573 | 233 | 14.06 | 160.52 |
| ShuffleNet | 73.45 | 73.73 | 73.59 | 1,258,729 | 152 | 4.86 | 175.96 |
| EfficientNet | 67.60 | 69.08 | 68.16 | 3,812,005 | 396 | 14.69 | 152.39 |
| Methods | F1 | Precision | Recall | Parameters | FLOPs | Size | FPS |
|---|---|---|---|---|---|---|---|
| (%) | (%) | (%) | ( ) | (M) | (MB) | ( ) | |
| MobileNetV1 | 62.26 | 64.28 | 62.43 | 3,212,101 | 590 | 12.34 | 426.02 |
| MobileNetV2 | 61.33 | 61.29 | 61.52 | 2,230,277 | 330 | 8.64 | 230.79 |
| MobileNetV3 | 52.68 | 53.11 | 53.08 | 3,660,573 | 230 | 14.06 | 169.88 |
| ShuffleNet | 60.65 | 61.25 | 60.99 | 1,258,729 | 150 | 4.86 | 177.18 |
| EfficientNet | 65.54 | 66.49 | 66.31 | 3,812,005 | 400 | 14.69 | 151.58 |
| Methods | F1 | Precision | Recall | Parameters | FLOPs | Size | FPS |
|---|---|---|---|---|---|---|---|
| (%) | (%) | (%) | ( ) | (M) | (MB) | ( ) | |
| MobileNetV1 | 82.71 | 83.58 | 82.94 | 3,212,101 | 590 | 12.34 | 414.29 |
| MobileNetV2 | 82.77 | 83.42 | 82.94 | 2,230,277 | 330 | 8.64 | 226.12 |
| MobileNetV3 | 77.04 | 77.44 | 77.25 | 3,660,573 | 230 | 14.06 | 163.09 |
| ShuffleNet | 80.95 | 81.66 | 81.18 | 1,258,729 | 150 | 4.86 | 182.29 |
| EfficientNet | 77.67 | 81.01 | 78.04 | 3,812,005 | 400 | 14.69 | 155.68 |
| Variant | F1 (%) | Params | Modification |
| Full model | |||
| VeloxNet (full) | 81.57 | 399K | Proposed architecture |
| SqueezeNet (fire modules) | 75.25 | 741K | All gMLP blocks replaced with fire modules |
| Component removal | |||
| w/o SGU | 77.83 | 385K | Remove spatial gating; keep channel projections only |
| w/o residual connections | 74.19 | 399K | Remove skip connections from gMLP blocks |
Image Classification
Image classification assigns semantic labels to images, enabling recognition of objects, scenes, conditions, and fine-grained categories.
See all 390 papers