efficientnet_b4.ra2_in1k optimized for Arm-based Edge Linux

efficientnet_b4.ra2_in1k image classification model, optimized with W8A8 dynamic post-training quantization and exported as an ExecuTorch program for the XNNPACK CPU backend on Arm-based platforms.

This model was optimized using Arm AI Model Optimizer.

Summary

This repository contains an optimized version of timm/efficientnet_b4.ra2_in1k for image classification. The model is provided as an ExecuTorch .pte file using the XNNPACK with KleidiAI delegate, targeting Edge Linux systems. Precision: INT8 symmetric per-channel weights, INT8 dynamic activations.

Key results

Area Result
Model format ExecuTorch (.pte)
Target device class Edge Linux
Reference device Raspberry Pi 5 (16GB) (Cortex-A76, 4 cores)
Primary performance result 1.09x lower p50 latency (467.94 ms to 431.22 ms)
Accuracy result Top-1 accuracy 82.74% (-0.32 pp vs. baseline)
Size / memory result 3.66x smaller model (79.61 MB to 21.77 MB), 132.91 MB peak memory

Original model

Field Value
Original model timm/efficientnet_b4.ra2_in1k
Original source Hugging Face
Original developer timm
Original model card timm/efficientnet_b4.ra2_in1k
Original license BSD-3-Clause

Model files

File Description
model.pte Optimized INT8 ExecuTorch program for deployment
metadata.yaml Optimized model metadata
benchmarks/ Baseline and optimized benchmark records

Performance

Performance was measured on the reference configuration below. Results do not guarantee identical performance on every Arm-based system.

Reference configuration

Field Value
Device / platform Raspberry Pi 5 (16GB)
CPU / accelerator Cortex-A76, 4 cores, 2.4 GHz
OS Raspberry Pi OS Bookworm
Runtime ExecuTorch 1.1.0
Backend / delegate XNNPACK with KleidiAI, 4 threads
Batch size 1
Input resolution 384x384
Precision INT8 symmetric per-channel weights, INT8 dynamic activations
Runs 10 measured

Performance results

Metric Original / baseline Optimized Improvement
p50 end-to-end latency 467.94 ms 431.22 ms 1.09x
Throughput - 2.32 frames/sec -
Peak memory 188.92 MB 132.91 MB 1.42x less
Model size 79.61 MB 21.77 MB 3.66x smaller

Accuracy

Evaluation setup

Field Value
Dataset ImageNet-1k
Split val
Number of samples 5,000
Evaluation runtime ExecuTorch

Accuracy results

Metric Original / baseline Optimized Change
Top-1 accuracy 83.06% 82.74% -0.32 pp
Top-5 accuracy - 96.72% -

Users should re-evaluate the model on their own data before production use.

Optimization approach

Optimization area Applied? Notes
Model conversion Yes PyTorch model exported to ExecuTorch (.pte)
Quantization Yes INT8 symmetric per-channel weights, INT8 dynamic activations. Calibrated on 1,000 random samples (imagenet1k_calib_seed42_n1000)
Runtime/backend selection Yes XNNPACK with KleidiAI, 4 threads
Accuracy validation Yes ImageNet-1k, 5000 samples
Performance validation Yes Measured on Raspberry Pi 5 (16GB)

Intended use

This model is intended for developers evaluating image classification workloads on Arm-based Edge Linux platforms. It is suitable as a reference implementation for benchmarking, prototyping, and integration exploration.

Limitations

  • Performance depends on the target device, runtime version, backend support, memory configuration, and system load.
  • Accuracy was measured on a 5,000-image subset of the ImageNet-1k validation set, not the full 50,000 images.
  • This repository is not a replacement for the original model documentation.

About this version

Original Model: timm/efficientnet_b4.ra2_in1k by timm - Repository

Optimization/conversion: W8A8 dynamic post-training quantization and ExecuTorch .pte export with XNNPACK + KleidiAI backend support for execution on Arm-based platforms.

Converted/optimized by: Nota AI, using Arm AI Model Optimizer

License: The Original Model and the Optimized Model are subject to BSD-3-Clause.

This repository contains a converted or optimized version of the Original Model (the "Optimized Model"). The Original Model has been converted or optimized as described above for execution on Arm-based platforms.

No retraining or fine-tuning of the Original Model was performed as part of the conversion or optimization. The conversion or optimization was not intended to change the Original Model's behavior or intended use.

Original Model and Documentation

For information about the Original Model, including its development, training data, intended uses, limitations and other relevant information, please refer to the Original Model repository. Information in that repository was provided by the original developer or other third parties and, unless expressly stated otherwise, has not been independently verified by Arm.

Licenses and Third-Party Terms

Use of the Original Model and the Optimized Model is subject to the applicable licenses, usage restrictions and other terms identified above and in the relevant repositories. Publication of the Optimized Model does not grant any rights beyond those provided under the applicable license terms.

You are responsible for reviewing those terms and ensuring that your use of the Original Model and the Optimized Model is permitted.

Purpose of this Release

The Optimized Model is provided as a reference implementation to demonstrate and evaluate execution and performance on Arm-based systems. It is not a production-ready or supported solution.

Arm's publication of the Optimized Model does not constitute an endorsement or certification of the Original Model or a representation that the Optimized Model is suitable for production use or any particular purpose.

To the fullest extent permitted by applicable law (i) the Optimized Model is provided "as is." Arm makes no representations or warranties that the Original Model, the Optimized Model or their outputs are accurate, safe, secure, non-infringing, legally compliant, suitable for production use or fit for any particular purpose; and (ii) Arm will not be liable for any loss or damage arising from or in connection with the Optimized Model, its use or its outputs.

You are responsible for independently evaluating the Optimized Model, its outputs and its suitability for your intended use, including compliance with applicable legal, regulatory, safety and security requirements.

Arm does not commit to provide ongoing support, maintenance or updates for the Optimized Model. Any use of or reliance on the Optimized Model or its outputs is at your own risk.

Downloads last month
29
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for Arm/efficientnet-b4-int8-xnnpack-executorch-raspberrypi5

Quantized
(3)
this model