Commit
24b4f29
·
verified ·
1 Parent(s): db0cc39

Sync model repo (text/metadata)

Browse files
README.md ADDED
@@ -0,0 +1,157 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ pipeline_tag: image-classification
4
+ library_name: executorch
5
+ tags:
6
+ - arm
7
+ - edge-linux
8
+ - executorch
9
+ - xnnpack
10
+ - int8
11
+ - imagenet-1k
12
+ ---
13
+
14
+ # EfficientViT optimized for Arm-based Edge Linux
15
+
16
+ EfficientViT image classification model, optimized with INT8 quantization and exported as an ExecuTorch program for the XNNPACK CPU backend on Arm-based platforms.
17
+
18
+ This model was optimized using [Arm AI Model Optimizer](https://developer.arm.com/dev-docs/ai-model-optimizer/).
19
+
20
+ ## Summary
21
+
22
+ This repository contains an optimized version of EfficientViT for image classification. The model is provided as an ExecuTorch `.pte` file using the XNNPACK with KleidiAI delegate, targeting Edge Linux systems. Precision: INT8 weights (quantization details not recorded in metadata).
23
+
24
+ **Key results**
25
+
26
+ | Area | Result |
27
+ |---|---|
28
+ | Model format | ExecuTorch (.pte) |
29
+ | Target device class | Edge Linux |
30
+ | Reference device | Raspberry Pi 5 (16GB) (Cortex-A76, 4 cores) |
31
+ | Primary performance result | 1.24x lower p50 latency (26.01 ms to 20.90 ms) |
32
+ | Accuracy result | Top-1 accuracy 78.68% (-0.68 pp vs. baseline) |
33
+ | Size / memory result | 2.06x smaller model (37.39 MB to 18.12 MB), 31.69 MB peak memory |
34
+
35
+ ## Original model
36
+
37
+ | Field | Value |
38
+ |---|---|
39
+ | Original model | EfficientViT |
40
+ | Original source | GitHub |
41
+ | Original developer | MIT HAN Lab |
42
+ | Original model card | [EfficientViT](https://github.com/mit-han-lab/efficientvit) |
43
+ | Original license | [Apache-2.0](https://www.apache.org/licenses/LICENSE-2.0) |
44
+
45
+ ## Model files
46
+
47
+ | File | Description |
48
+ |---|---|
49
+ | `model.pte` | Optimized INT8 ExecuTorch program for deployment |
50
+ | `metadata.yaml` | Optimized model metadata |
51
+ | `benchmarks/` | Baseline and optimized benchmark records |
52
+
53
+ ## Performance
54
+
55
+ Performance was measured on the reference configuration below. Results do not guarantee identical performance on every Arm-based system.
56
+
57
+ **Reference configuration**
58
+
59
+ | Field | Value |
60
+ |---|---|
61
+ | Device / platform | Raspberry Pi 5 (16GB) |
62
+ | CPU / accelerator | Cortex-A76, 4 cores, 2.4 GHz |
63
+ | OS | Raspberry Pi OS Bookworm |
64
+ | Runtime | ExecuTorch 1.1.0 |
65
+ | Backend / delegate | XNNPACK with KleidiAI, 4 threads |
66
+ | Batch size | 1 |
67
+ | Input resolution | 224x224 |
68
+ | Precision | INT8 weights (quantization details not recorded in metadata) |
69
+ | Runs | 200 measured, 20 warm-up |
70
+
71
+ **Performance results**
72
+
73
+ | Metric | Original / baseline | Optimized | Improvement |
74
+ |---|---:|---:|---:|
75
+ | p50 end-to-end latency | 26.01 ms | 20.90 ms | 1.24x |
76
+ | Peak memory | 50.92 MB | 31.69 MB | 1.61x less |
77
+ | Model size | 37.39 MB | 18.12 MB | 2.06x smaller |
78
+
79
+ ## Accuracy
80
+
81
+ **Evaluation setup**
82
+
83
+ | Field | Value |
84
+ |---|---|
85
+ | Dataset | ImageNet-1k |
86
+ | Split | val |
87
+ | Number of samples | 5,000 |
88
+ | Evaluation runtime | ExecuTorch |
89
+
90
+ **Accuracy results**
91
+
92
+ | Metric | Original / baseline | Optimized | Change |
93
+ |---|---:|---:|---:|
94
+ | Top-1 accuracy | 79.36% | 78.68% | -0.68 pp |
95
+ | Top-5 accuracy | - | 94.34% | - |
96
+
97
+ Users should re-evaluate the model on their own data before production use.
98
+
99
+ ## Optimization approach
100
+
101
+ | Optimization area | Applied? | Notes |
102
+ |---|---|---|
103
+ | Model conversion | Yes | PyTorch model exported to ExecuTorch (.pte) |
104
+ | Quantization | Yes | INT8 weights (quantization details not recorded in metadata) |
105
+ | Runtime/backend selection | Yes | XNNPACK with KleidiAI, 4 threads |
106
+ | Accuracy validation | Yes | ImageNet-1k, 5000 samples |
107
+ | Performance validation | Yes | Measured on Raspberry Pi 5 (16GB) |
108
+
109
+ ## Intended use
110
+
111
+ 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.
112
+
113
+ ## Limitations
114
+
115
+ - Performance depends on the target device, runtime version, backend support, memory configuration, and system load.
116
+ - Accuracy was measured on a 5,000-image subset of the ImageNet-1k validation set, not the full 50,000 images.
117
+ - This repository is not a replacement for the original model documentation.
118
+
119
+ ## About this version
120
+
121
+ Original Model:
122
+ EfficientViT by MIT HAN Lab - [Repository](https://github.com/mit-han-lab/efficientvit)
123
+
124
+ Optimization/conversion:
125
+ INT8 quantization and ExecuTorch `.pte` export with XNNPACK + KleidiAI backend support for execution on Arm-based platforms.
126
+
127
+ Converted/optimized by:
128
+ Nota AI, using Arm AI Model Optimizer
129
+
130
+ License:
131
+ The Original Model and the Optimized Model are subject to [Apache-2.0](https://www.apache.org/licenses/LICENSE-2.0).
132
+
133
+ 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.
134
+
135
+ 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.
136
+
137
+ ### Original Model and Documentation
138
+
139
+ 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](https://github.com/mit-han-lab/efficientvit). 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.
140
+
141
+ ### Licenses and Third-Party Terms
142
+
143
+ 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.
144
+
145
+ You are responsible for reviewing those terms and ensuring that your use of the Original Model and the Optimized Model is permitted.
146
+
147
+ ### Purpose of this Release
148
+
149
+ 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.
150
+
151
+ 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.
152
+
153
+ 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.
154
+
155
+ 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.
156
+
157
+ 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.
benchmarks/efficientvit-raspberrypi5-fp32.yaml ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ version: 1.0.0
2
+ report_type: vision-classification
3
+ profile: Baseline
4
+ created_at: '2026-06-23T13:30:07Z'
5
+ context:
6
+ target_ref: raspberry-pi-5
7
+ runtime:
8
+ name: executorch
9
+ execution_backend: cpu
10
+ version: 1.1.0
11
+ config:
12
+ n_threads: 4
13
+ optimisations:
14
+ - XNNPACK
15
+ - KleidiAI
16
+ dataset:
17
+ name: ImageNet-1k
18
+ sample_count: 5000
19
+ slice: val
20
+ reference_url: https://www.image-net.org/
21
+ benchmark:
22
+ batch_size: 1
23
+ num_runs: 200
24
+ warmup_runs: 20
25
+ input_resolution: 224x224
26
+ performance:
27
+ end_to_end_latency_ms:
28
+ p50: 26.01
29
+ peak_memory_mb: 50.92
30
+ accuracy:
31
+ top_1_accuracy_pct: 79.36
benchmarks/efficientvit-raspberrypi5-int8.yaml CHANGED
@@ -1,16 +1,8 @@
1
  version: 1.0.0
2
  report_type: vision-classification
3
- created_at: '2026-06-23T13:30:07Z'
4
  profile: Arm-Optimized
 
5
  context:
6
- model:
7
- id: arm/efficientvit-arm-optimized
8
- base_model_id: TBD/efficientvit
9
- filename: model.pte
10
- profile: Arm-Optimized
11
- weight_dtype: int8
12
- model_size_mb: 18.62
13
- format: pte
14
  target_ref: raspberry-pi-5
15
  runtime:
16
  name: executorch
@@ -29,6 +21,7 @@ context:
29
  benchmark:
30
  batch_size: 1
31
  num_runs: 200
 
32
  input_resolution: 224x224
33
  performance:
34
  end_to_end_latency_ms:
 
1
  version: 1.0.0
2
  report_type: vision-classification
 
3
  profile: Arm-Optimized
4
+ created_at: '2026-06-23T13:30:07Z'
5
  context:
 
 
 
 
 
 
 
 
6
  target_ref: raspberry-pi-5
7
  runtime:
8
  name: executorch
 
21
  benchmark:
22
  batch_size: 1
23
  num_runs: 200
24
+ warmup_runs: 20
25
  input_resolution: 224x224
26
  performance:
27
  end_to_end_latency_ms:
metadata.yaml CHANGED
@@ -4,8 +4,8 @@ id: Arm/efficientvit-int8-xnnpack-executorch-raspberrypi5
4
  task_type: image-classification
5
  title: EfficientViT INT8
6
  filename: model.pte
7
- base_model_id: TBD/efficientvit
8
  weight_dtype: int8
9
  profile: Arm-Optimized
10
- model_size_mb: 18.62
11
  format: pte
 
4
  task_type: image-classification
5
  title: EfficientViT INT8
6
  filename: model.pte
7
+ base_model_id: https://github.com/mit-han-lab/efficientvit
8
  weight_dtype: int8
9
  profile: Arm-Optimized
10
+ model_size_mb: 18.12
11
  format: pte
np_recipe/calibration.json DELETED
@@ -1,11 +0,0 @@
1
- {
2
- "datasets": [
3
- {
4
- "ref_id": "595edc1f-9919-4027-b23d-1436459990fa",
5
- "dataset_name": "imagenet1k_calib_seed42_n1000",
6
- "split": "CALIBRATION",
7
- "num_samples": 1000,
8
- "original_dataset_alias": "imagenet1k"
9
- }
10
- ]
11
- }
 
 
 
 
 
 
 
 
 
 
 
 
np_recipe/evaluation.json DELETED
@@ -1,11 +0,0 @@
1
- {
2
- "datasets": [
3
- {
4
- "ref_id": "384e7ec8-2e03-4f88-9c7f-f7a248ae8403",
5
- "dataset_name": "imagenet1k_eval_seed42_n5000",
6
- "split": "EVALUATION",
7
- "num_samples": 5000,
8
- "original_dataset_alias": "imagenet1k"
9
- }
10
- ]
11
- }
 
 
 
 
 
 
 
 
 
 
 
 
np_recipe/project.json DELETED
@@ -1,24 +0,0 @@
1
- {
2
- "project_id": "5a6d8971-9ba1-448b-af43-f0a7ba744cc1",
3
- "name": "np_hpo_ee33516094",
4
- "target": {
5
- "device": "raspberry_pi_5_16gb",
6
- "runtime": "executorch",
7
- "backend": "xnnpack"
8
- },
9
- "gpu_device": "cuda:2",
10
- "base_model": {
11
- "model_id": "4d16fab8-a5e9-43a9-a3da-3d92117454b3",
12
- "alias": "efficientvit_pt2_pytorch_static_fp32",
13
- "modality": "CV",
14
- "task": "CLASSIFICATION"
15
- },
16
- "calibration": {
17
- "dataset_id": "595edc1f-9919-4027-b23d-1436459990fa",
18
- "dataset_name": "imagenet1k_calib_seed42_n1000"
19
- },
20
- "evaluation": {
21
- "dataset_id": "384e7ec8-2e03-4f88-9c7f-f7a248ae8403",
22
- "dataset_name": "imagenet1k_eval_seed42_n5000"
23
- }
24
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
np_recipe/run.detail.yaml DELETED
@@ -1,95 +0,0 @@
1
- detail:
2
- batch_size: 1
3
- gpu_device: cuda:2
4
- graph_optimize[0]:
5
- pattern_handlers:
6
- - PatternHandlerAdjustAdd
7
- - PatternHandlerAdjustNonContiguousConstantMathInput
8
- - PatternHandlerCanonicalizeConvolutionDefault
9
- - PatternHandlerCombineAddRelu
10
- - PatternHandlerCombineConvBN
11
- - PatternHandlerCombineConvBNRelu
12
- - PatternHandlerCombineConvRelu
13
- - PatternHandlerCombineConvTransposeBN
14
- - PatternHandlerCombineConvTransposeBNRelu
15
- - PatternHandlerCombineConvTransposeRelu
16
- - PatternHandlerCombineLinearRelu
17
- - PatternHandlerCombineMulRelu
18
- - PatternHandlerCommuteBNBeforeJoinOp
19
- - PatternHandlerCommuteBNBeforeLinearOp
20
- - PatternHandlerComposeErfFormulaIntoGelu
21
- - PatternHandlerComposeTanhFormulaIntoGelu
22
- - PatternHandlerDecomposeMaxDimKeepdimFalse
23
- - PatternHandlerDecomposePReluToReluMath
24
- - PatternHandlerFuseAffineInputToLinear
25
- - PatternHandlerFuseBNToConv
26
- - PatternHandlerFuseConsecutiveBatchNorm
27
- - PatternHandlerFuseConsecutiveCast
28
- - PatternHandlerFuseConsecutiveConcat
29
- - PatternHandlerFuseConsecutiveIdempotent
30
- - PatternHandlerFuseConsecutiveLogSoftmax
31
- - PatternHandlerFuseConsecutiveReshapeTranspose
32
- - PatternHandlerFuseConsecutiveSlice
33
- - PatternHandlerFuseConsecutiveUnsqueeze
34
- - PatternHandlerFuseConstantPath
35
- - PatternHandlerFuseInputScaleToConv
36
- - PatternHandlerFuseMathToConv
37
- - PatternHandlerFuseMathToLinear
38
- - PatternHandlerFusePadToConv
39
- - PatternHandlerFusePadToPool
40
- - PatternHandlerFuseReluPostfixToClamp
41
- - PatternHandlerFuseReluPrefixToClamp
42
- - PatternHandlerFuseMultiBranchReshapeTranspose
43
- - PatternHandlerFuseQKV
44
- - PatternHandlerFuseSplitConvConcat
45
- - PatternHandlerFuseUnsqueezeToReduce
46
- - PatternHandlerNormalizeSpatialMeanKeepdimFalse
47
- - PatternHandlerRemoveGraphAssertsPass
48
- - PatternHandlerRemoveBatchNormTracking
49
- - PatternHandlerRemoveCommonSubexpression
50
- - PatternHandlerRemoveDeadend
51
- - PatternHandlerRemoveDisabledAutocastWrapper
52
- - PatternHandlerRemoveDropout
53
- - PatternHandlerRemoveMonotoneBeforeRanking
54
- - PatternHandlerRemoveNoOpCast
55
- - PatternHandlerRemoveNoOpConcat
56
- - PatternHandlerRemoveNoOpExpand
57
- - PatternHandlerRemoveNoOpFlatten
58
- - PatternHandlerRemoveNoOpMath
59
- - PatternHandlerRemoveNoOpPad
60
- - PatternHandlerRemoveNoOpReshape
61
- - PatternHandlerRemoveNoOpSlice
62
- - PatternHandlerRemoveNoOpSplit
63
- - PatternHandlerRemoveNoOpSqueeze
64
- - PatternHandlerRemoveNoOpTranspose
65
- - PatternHandlerRemovePositive
66
- - PatternHandlerReplaceBatchNormWithConv
67
- - PatternHandlerReplaceBitwiseRightShiftWithDiv
68
- - PatternHandlerReplaceEinsumWithMatmul
69
- - PatternHandlerReplaceGatherWithConv
70
- - PatternHandlerReplaceGatherWithSliceReshape
71
- - PatternHandlerReplaceInplaceAdd
72
- - PatternHandlerReplaceInplaceHardswish
73
- - PatternHandlerReplaceInplaceIndexPut
74
- - PatternHandlerReplaceInplaceLeakyRelu
75
- - PatternHandlerReplaceInplaceMul
76
- - PatternHandlerReplaceInplaceRelu
77
- - PatternHandlerReplaceInplaceSilu
78
- - PatternHandlerReplaceMatmul2dWithLinear
79
- - PatternHandlerReplaceMatmul3dWithLinear
80
- - PatternHandlerReplaceMatmulWithLinear
81
- graph_quantize[0]:
82
- weight:
83
- granularity: CHANNEL
84
- calibration_method: MINMAX
85
- activation:
86
- granularity: TENSOR
87
- calibration_method: MINMAX
88
- calibration:
89
- num_samples: 128
90
- profile[0]:
91
- num_executions: 10
92
- timeout: 1800
93
- evaluate[0]:
94
- metrics:
95
- ACCURACY: {}
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
np_recipe/run.yaml DELETED
@@ -1,12 +0,0 @@
1
- version: '1.0'
2
- steps:
3
- - graph_optimize: null
4
- - graph_quantize:
5
- - scheme: W8-A8_dynamic
6
- automatic_mixed_precision: null
7
- profile:
8
- - OUTPUT_MODEL
9
- evaluate:
10
- - model: OUTPUT_MODEL
11
- metric:
12
- - ACCURACY