Zipformer

0 downloads 0 likes on ModelScope updated

Model card

Zipformer: Optimized for Qualcomm Devices

Zipformer streaming ASR (Automatic Speech Recognition) model is a state-of-the-art system designed for transcribing spoken language into written text streamingly. This model is based on the transformer architecture and has been optimized for edge inference by replacing linear layers with convolutional (conv) layers. It exhibits robust performance in realistic, noisy environments, making it highly reliable for real-world applications. Specifically, it excels in long-form transcription, capable of accurately transcribing audios. Time to the first token is the encoder's latency, while time to each additional token is joiner's latency, where we assume a max decoded length specified below.

This is based on the implementation of Zipformer found here. This repository contains pre-exported model files optimized for Qualcomm® devices. You can use the Qualcomm® AI Hub Models library to export with custom configurations. More details on model performance across various devices, can be found here.

Qualcomm AI Hub Models uses Qualcomm AI Hub Workbench to compile, profile, and evaluate this model. Sign up to run these models on a hosted Qualcomm® device.

Getting Started

There are two ways to deploy this model on your device:

Option 1: Download Pre-Exported Models

Below are pre-exported model assets ready for deployment.

Runtime Precision Chipset SDK Versions Download
PRECOMPILED_QNN_ONNX float Snapdragon® X2 Elite QAIRT 2.45, ONNX Runtime 1.27.1 Download
PRECOMPILED_QNN_ONNX float Snapdragon® X Elite QAIRT 2.45, ONNX Runtime 1.27.1 Download
PRECOMPILED_QNN_ONNX float Snapdragon® 8 Gen 3 Mobile QAIRT 2.45, ONNX Runtime 1.27.1 Download
PRECOMPILED_QNN_ONNX float Snapdragon® 8 Gen 1 Mobile QAIRT 2.45, ONNX Runtime 1.27.1 Download
PRECOMPILED_QNN_ONNX float Qualcomm® Dragonwing™ IQ-8275 QAIRT 2.45, ONNX Runtime 1.27.1 Download
PRECOMPILED_QNN_ONNX float Qualcomm® Dragonwing™ QCS8550 (Proxy) QAIRT 2.45, ONNX Runtime 1.27.1 Download
PRECOMPILED_QNN_ONNX float Qualcomm® Dragonwing™ IQ-9075 QAIRT 2.45, ONNX Runtime 1.27.1 Download
PRECOMPILED_QNN_ONNX float Snapdragon® 8 Elite Mobile QAIRT 2.45, ONNX Runtime 1.27.1 Download
PRECOMPILED_QNN_ONNX float Snapdragon® 8 Elite Gen 5 Mobile QAIRT 2.45, ONNX Runtime 1.27.1 Download
QNN_CONTEXT_BINARY float Snapdragon® X2 Elite QAIRT 2.45 Download
QNN_CONTEXT_BINARY float Snapdragon® X Elite QAIRT 2.45 Download
QNN_CONTEXT_BINARY float Snapdragon® 8 Gen 3 Mobile QAIRT 2.45 Download
QNN_CONTEXT_BINARY float Snapdragon® 8 Gen 1 Mobile QAIRT 2.45 Download
QNN_CONTEXT_BINARY float Qualcomm® Dragonwing™ IQ-8275 QAIRT 2.45 Download
QNN_CONTEXT_BINARY float Qualcomm® Dragonwing™ QCS8550 (Proxy) QAIRT 2.45 Download
QNN_CONTEXT_BINARY float Qualcomm® SA8775P QAIRT 2.45 Download
QNN_CONTEXT_BINARY float Qualcomm® Dragonwing™ IQ-9075 QAIRT 2.45 Download
QNN_CONTEXT_BINARY float Qualcomm® SA7255P QAIRT 2.45 Download
QNN_CONTEXT_BINARY float Qualcomm® SA8295P QAIRT 2.45 Download
QNN_CONTEXT_BINARY float Snapdragon® 8 Elite Mobile QAIRT 2.45 Download
QNN_CONTEXT_BINARY float Snapdragon® 8 Elite Gen 5 Mobile QAIRT 2.45 Download

For more device-specific assets and performance metrics, visit Zipformer on Qualcomm® AI Hub.

Option 2: Export with Custom Configurations

Use the Qualcomm® AI Hub Models Python library to compile and export the model with your own:

  • Custom weights (e.g., fine-tuned checkpoints)
  • Custom input shapes
  • Target device and runtime configurations

This option is ideal if you need to customize the model beyond the default configuration provided here.

See our repository for Zipformer on GitHub for usage instructions.

Model Details

Model Type: Model_use_case.speech_recognition

Model Stats:

  • Input resolution: 80x71 (0.71 seconds audio)
  • Max decoded sequence length: 200 tokens
  • Model checkpoint: pfluo/k2fsa-zipformer-chinese-english-mixed
  • Model size (decoder) (float): 13.2 MB
  • Model size (encoder) (float): 242 MB
  • Model size (joiner) (float): 12.2 MB
  • Number of parameters (decoder): 3.47M
  • Number of parameters (encoder): 63.2M
  • Number of parameters (joiner): 3.21M

Performance Summary

Model Runtime Precision Chipset Inference Time (ms) Peak Memory Range (MB) Primary Compute Unit
decoder PRECOMPILED_QNN_ONNX float Snapdragon® X2 Elite 0.092 ms 1 - 1 MB NPU
decoder PRECOMPILED_QNN_ONNX float Snapdragon® X Elite 0.142 ms 7 - 7 MB NPU
decoder PRECOMPILED_QNN_ONNX float Snapdragon® 8 Gen 3 Mobile 0.158 ms 0 - 7 MB NPU
decoder PRECOMPILED_QNN_ONNX float Snapdragon® 8 Gen 1 Mobile 0.25 ms 0 - 14 MB NPU
decoder PRECOMPILED_QNN_ONNX float Qualcomm® Dragonwing™ IQ-8275 0.357 ms 0 - 4 MB NPU
decoder PRECOMPILED_QNN_ONNX float Qualcomm® Dragonwing™ QCS8550 (Proxy) 0.191 ms 0 - 8 MB NPU
decoder PRECOMPILED_QNN_ONNX float Qualcomm® QCS8450 0.25 ms 0 - 14 MB NPU
decoder PRECOMPILED_QNN_ONNX float Qualcomm® Dragonwing™ IQ-9075 0.274 ms 0 - 3 MB NPU
decoder PRECOMPILED_QNN_ONNX float Qualcomm® Dragonwing™ IQ-X7181 0.142 ms 7 - 7 MB NPU
decoder PRECOMPILED_QNN_ONNX float Qualcomm® Dragonwing™ Q-8750 0.139 ms 0 - 10 MB NPU
decoder PRECOMPILED_QNN_ONNX float Snapdragon® 8 Elite Mobile 0.139 ms 0 - 10 MB NPU
decoder PRECOMPILED_QNN_ONNX float Snapdragon® 8 Elite Gen 5 Mobile 0.137 ms 0 - 7 MB NPU
decoder QNN_CONTEXT_BINARY float Snapdragon® X2 Elite 0.176 ms 0 - 0 MB NPU
decoder QNN_CONTEXT_BINARY float Snapdragon® X Elite 0.241 ms 0 - 0 MB NPU
decoder QNN_CONTEXT_BINARY float Snapdragon® 8 Gen 3 Mobile 0.15 ms 15 - 22 MB NPU
decoder QNN_CONTEXT_BINARY float Snapdragon® 8 Gen 1 Mobile 0.24 ms 0 - 9 MB NPU
decoder QNN_CONTEXT_BINARY float Qualcomm® Dragonwing™ IQ-8275 0.334 ms 0 - 2 MB NPU
decoder QNN_CONTEXT_BINARY float Qualcomm® Dragonwing™ QCS8550 (Proxy) 0.197 ms 0 - 1 MB NPU
decoder QNN_CONTEXT_BINARY float Qualcomm® SA8775P 0.324 ms 0 - 10 MB NPU
decoder QNN_CONTEXT_BINARY float Qualcomm® SA8650P 0.324 ms 0 - 10 MB NPU
decoder QNN_CONTEXT_BINARY float Qualcomm® SA8255P 0.324 ms 0 - 10 MB NPU
decoder QNN_CONTEXT_BINARY float Qualcomm® QCS8450 0.24 ms 0 - 9 MB NPU
decoder QNN_CONTEXT_BINARY float Qualcomm® Dragonwing™ IQ-9075 0.24 ms 0 - 2 MB NPU
decoder QNN_CONTEXT_BINARY float Qualcomm® Dragonwing™ IQ-X7181 0.241 ms 0 - 0 MB NPU
decoder QNN_CONTEXT_BINARY float Qualcomm® Dragonwing™ Q-8750 0.133 ms 0 - 12 MB NPU
decoder QNN_CONTEXT_BINARY float Qualcomm® SA7255P 0.432 ms 0 - 8 MB NPU
decoder QNN_CONTEXT_BINARY float Qualcomm® SA8295P 0.456 ms 0 - 5 MB NPU
decoder QNN_CONTEXT_BINARY float Snapdragon® 8 Elite Mobile 0.133 ms 0 - 12 MB NPU
decoder QNN_CONTEXT_BINARY float Snapdragon® 8 Elite Gen 5 Mobile 0.131 ms 0 - 8 MB NPU
encoder PRECOMPILED_QNN_ONNX float Snapdragon® X2 Elite 7.921 ms 10 - 10 MB NPU
encoder PRECOMPILED_QNN_ONNX float Snapdragon® X Elite 13.362 ms 142 - 142 MB NPU
encoder PRECOMPILED_QNN_ONNX float Snapdragon® 8 Gen 3 Mobile 8.885 ms 9 - 21 MB NPU
encoder PRECOMPILED_QNN_ONNX float Snapdragon® 8 Gen 1 Mobile 16.148 ms 9 - 23 MB NPU
encoder PRECOMPILED_QNN_ONNX float Qualcomm® Dragonwing™ IQ-8275 13.723 ms 8 - 13 MB NPU
encoder PRECOMPILED_QNN_ONNX float Qualcomm® Dragonwing™ QCS8550 (Proxy) 12.508 ms 0 - 159 MB NPU
encoder PRECOMPILED_QNN_ONNX float Qualcomm® QCS8450 16.148 ms 9 - 23 MB NPU
encoder PRECOMPILED_QNN_ONNX float Qualcomm® Dragonwing™ IQ-9075 14.85 ms 8 - 12 MB NPU
encoder PRECOMPILED_QNN_ONNX float Qualcomm® Dragonwing™ IQ-X7181 13.362 ms 142 - 142 MB NPU
encoder PRECOMPILED_QNN_ONNX float Qualcomm® Dragonwing™ Q-8750 7.533 ms 9 - 22 MB NPU
encoder PRECOMPILED_QNN_ONNX float Snapdragon® 8 Elite Mobile 7.533 ms 9 - 22 MB NPU
encoder PRECOMPILED_QNN_ONNX float Snapdragon® 8 Elite Gen 5 Mobile 6.975 ms 10 - 16 MB NPU
encoder QNN_CONTEXT_BINARY float Snapdragon® X2 Elite 8.417 ms 3 - 3 MB NPU
encoder QNN_CONTEXT_BINARY float Snapdragon® X Elite 13.06 ms 3 - 3 MB NPU
encoder QNN_CONTEXT_BINARY float Snapdragon® 8 Gen 3 Mobile 8.737 ms 2 - 12 MB NPU
encoder QNN_CONTEXT_BINARY float Snapdragon® 8 Gen 1 Mobile 16.003 ms 2 - 13 MB NPU
encoder QNN_CONTEXT_BINARY float Qualcomm® Dragonwing™ IQ-8275 13.446 ms 3 - 8 MB NPU
encoder QNN_CONTEXT_BINARY float Qualcomm® Dragonwing™ QCS8550 (Proxy) 12.442 ms 3 - 4 MB NPU
encoder QNN_CONTEXT_BINARY float Qualcomm® SA8775P 14.123 ms 3 - 12 MB NPU
encoder QNN_CONTEXT_BINARY float Qualcomm® SA8650P 14.123 ms 3 - 12 MB NPU
encoder QNN_CONTEXT_BINARY float Qualcomm® SA8255P 14.123 ms 3 - 12 MB NPU
encoder QNN_CONTEXT_BINARY float Qualcomm® QCS8450 16.003 ms 2 - 13 MB NPU
encoder QNN_CONTEXT_BINARY float Qualcomm® Dragonwing™ IQ-9075 14.075 ms 3 - 7 MB NPU
encoder QNN_CONTEXT_BINARY float Qualcomm® Dragonwing™ IQ-X7181 13.06 ms 3 - 3 MB NPU
encoder QNN_CONTEXT_BINARY float Qualcomm® Dragonwing™ Q-8750 7.486 ms 3 - 15 MB NPU
encoder QNN_CONTEXT_BINARY float Qualcomm® SA7255P 28.792 ms 3 - 11 MB NPU
encoder QNN_CONTEXT_BINARY float Qualcomm® SA8295P 17.343 ms 0 - 5 MB NPU
encoder QNN_CONTEXT_BINARY float Snapdragon® 8 Elite Mobile 7.486 ms 3 - 15 MB NPU
encoder QNN_CONTEXT_BINARY float Snapdragon® 8 Elite Gen 5 Mobile 6.928 ms 3 - 11 MB NPU
joiner PRECOMPILED_QNN_ONNX float Snapdragon® X2 Elite 0.08 ms 1 - 1 MB NPU
joiner PRECOMPILED_QNN_ONNX float Snapdragon® X Elite 0.144 ms 6 - 6 MB NPU
joiner PRECOMPILED_QNN_ONNX float Snapdragon® 8 Gen 3 Mobile 0.155 ms 0 - 6 MB NPU
joiner PRECOMPILED_QNN_ONNX float Snapdragon® 8 Gen 1 Mobile 0.238 ms 0 - 14 MB NPU
joiner PRECOMPILED_QNN_ONNX float Qualcomm® Dragonwing™ IQ-8275 0.269 ms 0 - 3 MB NPU
joiner PRECOMPILED_QNN_ONNX float Qualcomm® Dragonwing™ QCS8550 (Proxy) 0.188 ms 0 - 8 MB NPU
joiner PRECOMPILED_QNN_ONNX float Qualcomm® QCS8450 0.238 ms 0 - 14 MB NPU
joiner PRECOMPILED_QNN_ONNX float Qualcomm® Dragonwing™ IQ-9075 0.27 ms 0 - 3 MB NPU
joiner PRECOMPILED_QNN_ONNX float Qualcomm® Dragonwing™ IQ-X7181 0.144 ms 6 - 6 MB NPU
joiner PRECOMPILED_QNN_ONNX float Qualcomm® Dragonwing™ Q-8750 0.134 ms 0 - 7 MB NPU
joiner PRECOMPILED_QNN_ONNX float Snapdragon® 8 Elite Mobile 0.134 ms 0 - 7 MB NPU
joiner PRECOMPILED_QNN_ONNX float Snapdragon® 8 Elite Gen 5 Mobile 0.135 ms 0 - 7 MB NPU
joiner QNN_CONTEXT_BINARY float Snapdragon® X2 Elite 0.204 ms 0 - 0 MB NPU
joiner QNN_CONTEXT_BINARY float Snapdragon® X Elite 0.238 ms 1 - 1 MB NPU
joiner QNN_CONTEXT_BINARY float Snapdragon® 8 Gen 3 Mobile 0.149 ms 0 - 7 MB NPU
joiner QNN_CONTEXT_BINARY float Snapdragon® 8 Gen 1 Mobile 0.229 ms 0 - 9 MB NPU
joiner QNN_CONTEXT_BINARY float Qualcomm® Dragonwing™ IQ-8275 0.242 ms 0 - 2 MB NPU
joiner QNN_CONTEXT_BINARY float Qualcomm® Dragonwing™ QCS8550 (Proxy) 0.187 ms 0 - 1 MB NPU
joiner QNN_CONTEXT_BINARY float Qualcomm® SA8775P 0.327 ms 0 - 9 MB NPU
joiner QNN_CONTEXT_BINARY float Qualcomm® SA8650P 0.327 ms 0 - 9 MB NPU
joiner QNN_CONTEXT_BINARY float Qualcomm® SA8255P 0.327 ms 0 - 9 MB NPU
joiner QNN_CONTEXT_BINARY float Qualcomm® QCS8450 0.229 ms 0 - 9 MB NPU
joiner QNN_CONTEXT_BINARY float Qualcomm® Dragonwing™ IQ-9075 0.234 ms 2 - 4 MB NPU
joiner QNN_CONTEXT_BINARY float Qualcomm® Dragonwing™ IQ-X7181 0.238 ms 1 - 1 MB NPU
joiner QNN_CONTEXT_BINARY float Qualcomm® Dragonwing™ Q-8750 0.134 ms 0 - 13 MB NPU
joiner QNN_CONTEXT_BINARY float Qualcomm® SA7255P 0.475 ms 0 - 8 MB NPU
joiner QNN_CONTEXT_BINARY float Qualcomm® SA8295P 0.461 ms 0 - 5 MB NPU
joiner QNN_CONTEXT_BINARY float Snapdragon® 8 Elite Mobile 0.134 ms 0 - 13 MB NPU
joiner QNN_CONTEXT_BINARY float Snapdragon® 8 Elite Gen 5 Mobile 0.134 ms 0 - 8 MB NPU

License

  • The license for the original implementation of Zipformer can be found here.

References

Community