A Generalized Optimization Engine (GOE) for Edge AI Inference Acceleration
Organizations: DEVCOM Army Research Laboratory Aberdeen Proving Ground, MD, USA · University of West Florida Pensacola, FL, USA
Abstract
Artificial intelligence (AI) models have demonstrated remarkable capabilities across various domains, yet their widespread deployment is impeded by significant computational costs, particularly on resource-constrained devices. This paper explores the theoretical underpinnings of various AI model optimization techniques, algorithms, and abstractions, discussing their potential to reduce computational complexity, memory footprint, latency, and power consumption. Furthermore, we propose a comprehensive hardware (HW) and model-agnostic generalized optimization architecture that integrates these techniques for improved efficiency. Our study underscores the critical role of such a generalized optimization system in preparing model deployment over resource-constrained heterogeneous hardware in a tactical environment. As a concrete demonstration, we show that GOE-compressed language models deploy and run on a GPU-less edge CPU, and that the choice of compression method, not merely its nominal bit-width, determines whether task accuracy survives deployment.
Figures & tables
| Symbol | Meaning |
|---|---|
| Model from the repository (e.g., CNN variant, LLM family member). | |
| Quantization bit-width / policy (e.g., 4-, 8-, 16-bit; potentially mixed-precision). | |
| Pruning vector describing method and sparsity level(s). | |
| Training hyperparameters (e.g., learning rate) relevant for fine-tuning/QAT. | |
| Global pruning fraction (structured/unstructured). | |
| Distillation indicator: = none, = apply KD/self-distillation. |
| Model | Accuracy (%) | Size (Mb) |
|---|---|---|
| ResNet 50 | 80.56 | 97.7 |
| ViT B32 | 75.82 | 336.55 |
| Total | Pruned | Pruned | |||
|---|---|---|---|---|---|
| Runtime | Fine-Tune | Accuracy | Size | ||
| Model | (Min) | Epochs | Prune (%) | (%) | (Mb) |
| ResNet 50 | 16 | 5 | 25 | 96.93 | 55 |
| ResNet 50 | 19 | 5 | 50 | 96.87 | 24.52 |
| ResNet 50 | 19 | 5 | 75 | 96.99 | 6.19 |
| ViT B32 | 18 | 5 | 25 | 96.11 | 282.51 |
| Model | Base | Base | Quant | Quant |
|---|---|---|---|---|
| Accuracy (%) | Latency | Accuracy (%) | Latency | |
| ResNet 50 | 81.0 | 0.00139 | 81.0 | 0.00005 |
| MobileNet v2 | 72.0 | 0.00059 | 72.0 | 0.00002 |
| MobileNet v3 Small | 68.0 | 0.00015 | 38.0 | 0.00001 |
| MobileNet v3 Large | 76.0 | 0.00041 | 74.0 | 0.00003 |
| Model | Number of | Compression | Accuracy | Accuracy | Memory |
|---|---|---|---|---|---|
| Parameters | Ratio | (%) | Change | Usage (Mb) | |
| ofa-rn50 | 48,105,992 | 0 | 80.14 | – | 183.76 |
| Subnet 0 | 23,514,408 | 2.05 | 85.43 | +5.29 | 89.90 |
| Subnet 1 | 20,423,600 | 2.36 | 84.19 | +4.05 | 78.10 |
| Subnet 2 | 20,597,080 | 2.34 | 84.18 | +4.04 | 78.75 |
| Subnet 3 | 23,106,688 | 2.08 | 84.00 | +3.86 | 88.34 |
| Subnet | Quantized | Subnet | Quantized |
|---|---|---|---|
| Accuracy (%) | Accuracy (%) | Size (Mb) | Size (Mb) |
| 78.42 | 78.34 | 93.08 | 85.26 |
| 77.15 | 77.09 | 71.14 | 63.32 |
| 78.07 | 78.10 | 79.73 | 73.47 |
| 76.60 | 76.62 | 82.67 | 76.41 |
| 77.33 | 77.36 | 77.78 | 71.52 |
| Variant | MB | tok/s | ARC-e | PIQA | HS | CSQA |
|---|---|---|---|---|---|---|
| F16 | 2365 | 23.5 | .620 | .760 | .627 | .620 |
| Q8_0 | 1260 | 40.6 | .613 | .753 | .627 | .620 |
| Q4_K_M | 770 | 58.6 | .613 | .760 | .600 | .587 |
| pruned+Q4 | 627 | 72.0 | .440 | .640 | .500 | .193 |
| Variant | MB | tok/s | ARC-e | PIQA | HS | CSQA |
|---|---|---|---|---|---|---|
| F16 | 2950 | 18.0 | .747 | .780 | .667 | .807 |
| Q8_0 | 1570 | 30.5 | .753 | .773 | .667 | .807 |
| Q4_K_M | 940 | 43.3 | .747 | .747 | .633 | .733 |
Efficient Neural Network Inference
Efficient neural network inference reduces the time, memory, and energy needed to run trained models across hardware and deployment settings.
See all 471 papers