MCPcopy Index your code
hub / github.com/ASLP-lab/OSUM-Pangu

github.com/ASLP-lab/OSUM-Pangu @main

Chat with this repo
repository ↗ · DeepWiki ↗ · + Follow
1,397 symbols 4,425 edges 182 files 488 documented · 35%
What it actually does AI analysis from the code graph — generated when you open this
loading…
README

OSUM-Pangu: An Open-Source Multidimension Speech Understanding Foundation Model Built upon OpenPangu on Ascend NPUs

Yujie Liao, Xuelong Geng, Hongfei Xue, Shuiyuan Wang, Lei Xie

<img src="https://github.com/ASLP-lab/OSUM-Pangu/raw/main/images/OSUM-Pangu.jpg" width="400"/>







<a href="https://huggingface.co/ASLP-lab/OSUM-Pangu"> Ckpt | <a href="https://arxiv.org/abs/2603.10862"> Paper</a>

Recent advancements in Speech Large Language Models have significantly enhanced multi-dimensional speech understanding. However, the majority of high-performance frameworks are predominantly optimized for GPU centric ecosystems and proprietary backbones, creating a significant gap for deployment on non-CUDA computing infrastructures. In this paper, we present OSUM-Pangu, a fully open-source speech understanding foundation model developed on a completely non-CUDA software and hardware stack. By integrating an audio encoder with the openPangu-7B LLM backbone, we successfully implement the entire training and inference pipeline on the Ascend NPU platform. To facilitate efficient task alignment under non-CUDA resource constraints, we adopt a practical training process that sequentially bridges speech perception and user intent recognition. Experimental results demonstrate that OSUM-Pangu achieves task accuracy comparable to mainstream GPU-based models while maintaining robust natural language interaction capabilities. Our work provides a reproducible, non-CUDA baseline for the open-source speech community, promoting the independent evolution of multimodal intelligence.


Architecture

The overall architecture of OSUM-Pangu is shown below:

<img src="https://github.com/ASLP-lab/OSUM-Pangu/raw/main/images/architecture1.png" width="80%"/>

The model mainly consists of three components:

1. Speech Encoder

Whisper-medium
Responsible for extracting speech representations.

2. Adapter

Transforms acoustic features into tokens compatible with the LLM input space.

3. Intent-aware LLM

<a href="https://huggingface.co/FreedomIntelligence/openPangu-Embedded-7B-V1.1"> openPangu-Embedded-7B-V1.1 </a>

Responsible for: - Parsing natural language instructions - Identifying user intent - Determining which speech task to execute


Training Strategy

We adopt a a three-stage training proces, illustrated below:

<img src="https://github.com/ASLP-lab/OSUM-Pangu/raw/main/images/Strategy.png" width="80%"/>

Stage 1: Speech Understanding Alignment

Goal: Equip the model with multi-task speech understanding capability.

Characteristics:

  • Only speech-related modules are trained
  • Establish strong acoustic representation ability

Stage 2: Intent Understanding

Goal: Enable the model to understand natural language user instructions.

Examples:

Please transcribe this audio.
Analyze the speaker's emotion.
Identify what event happens in the audio.

The model learns:

  • Instruction semantic understanding
  • Task mapping capability

Stage 3: Joint Instruction Tuning

In the final stage, joint training allows the model to:

  • Automatically parse user instructions
  • Identify task types
  • Execute the corresponding speech understanding tasks

Without requiring fixed templates, such as:

What is the emotion of this speech?
Can you transcribe this audio?
What event happens in the audio?

The model can correctly understand and execute all of them.


Results

Dataset Configuration

Our experiments follow the task definitions of the OSUM framework. To maintain the linguistic reasoning capability of the backbone, we incorporate 2M entries from Alpaca-CoT for text-based interactions, with queries synthesized using CosyVoice 2. To evaluate the model's robustness in real-world scenarios, we utilize an Intent-Instruction Set (IIS) containing over 80k training samples and 4k test prompts, covering diverse colloquial user queries.


Multi-task Speech Understanding Performance

OSUM-Pangu demonstrates competitive performance across diverse tasks compared to GPU-based baselines Qwen2-Audio and OSUM, proving the effectiveness of the NPU-based pipeline.

Task Model Public Test Set Metric Public Result
ASR Qwen2-Audio WenetSpeech(n/m)

AISHELL-2(m/i/a)

LibriSpeech (c/o) | WER/CER (%) | 8.84 / 8.40

3.0 / 3.0 / 2.9

1.6 / 3.6 | | | OSUM | | | 6.46 / 5.34

2.81 / 2.75 / 2.73

2.19 / 5.53 | | | OSUM-Pangu | | | 7.40 / 10.49

3.01 / 2.98 / 2.95

3.51 / 8.36 | | VED | Qwen2-Audio | VocalSound | ACC (%) | 93.3 | | | OSUM | | | 82.58 | | | OSUM-Pangu | | | 73.04 | | SER | Qwen2-Audio | MELD-test

MER2023 | ACC (%) | 55.3 / -- | | | OSUM | | | 53.38 / 86.43 | | | OSUM-Pangu | | | 36.40 / 89.19 | | SGC | Qwen2-Audio | Kaggle-CommonVoice test | ACC (%) | 97.25 | | | OSUM | | | 99.41 | | | OSUM-Pangu | | | 97.48 | | SAP | Qwen2-Audio | Kaggle-CommonVoice test | ACC (%) | 35.53 | | | OSUM | | | 76.52 | | | OSUM-Pangu | | | 83.31 |


Instruction Following Performance (IFR)

Instruction Following Rate (IFR) measures the ability of the model to parse natural language instructions and execute the corresponding tasks.

The metric is defined as:

$$ IFR = \left( \frac{N_{correct}}{N_{total}} \right) \times 100\% $$

where:

  • $N_{correct}$ represents the number of correctly executed instructions
  • $N_{total}$ represents the total number of evaluation samples Compared with mainstream open-source models, OSUM-Pangu achieves significantly better performance:
Model IFR (\%)
Qwen2Audio-Instruct 71.3
OSUM-Pangu (Ours) 90.2

Flexibility vs Accuracy

We evaluate whether natural language instructions (NL) degrade performance compared to fixed instructions (FI).

Results show that the model maintains strong flexibility while preserving task accuracy.

Task Test FI NL $\Delta$
ASR test-net/librispeech-clean 7.36/3.64 7.40/3.51 +0.04/-0.13
SER Testemotion 67.39 67.41 +0.02
SGC Testgender 97.04 96.02 -1.02
SRWT Testalign 22.39 17.52 -4.87
SSR Teststyle 62.79 58.05 -4.74
VED Testevent 77.74 73.04 -4.70
SAP Testage 71.75 72.86 +0.11

Conclusion:

Only minor performance drops appear in relatively niche tasks such as:

  • Style recognition
  • Event detection

Core tasks such as:

  • ASR
  • SER
  • SAP

remain almost unchanged, validating the effectiveness of the three-stage training process.


Speech-to-Text Chat (STTC) Capability

We further evaluate the model in conversational reasoning scenarios.

OSUM-Pangu outperforms GLM-4-Voice on the TriviaQA and WebQ benchmarks.

Model LLaMA Q TriviaQA Web Q
ChatGPT-4o 71.7 69.7 51.6
GLM-4-Voice 50.7 26.5 15.9
DeepTalk 59.7 27.5 23.1
OSUM-EChat 55.3 33.7 30.4
OSUM-Pangu 44.6 28.9 29.5

How to Use the OSUM-Pangu Framework for Training and Inference

Environment Setup

Before starting, please ensure that your device supports NPU and the Python environment is properly configured.

We recommend running the code on a Linux system.

If Conda is not installed, please refer to: https://blog.csdn.net/qq_41636123/article/details/130266232

# Create a new conda environment
conda create -n osum_pangu python=3.10
conda activate osum_pangu

# Clone the repository
git clone https://github.com/ASLP-lab/OSUM-Pangu.git
cd OSUM-Pangu

# Install dependencies
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

Model Download

from huggingface_hub import snapshot_download

snapshot_download(
    repo_id="ASLP-lab/OSUM-Pangu",
    local_dir="path",
    local_dir_use_symlinks=False,
    endpoint="https://hf-mirror.com"
)

Inference

This project provides batch inference scripts for all tasks under in :OSUM-Pangu/infer_code:

python infer_ASR.py

Training

To ensure a smooth training process, please follow the steps below.

1. Data Preparation

Data can be prepared in three formats:

raw、shard、combine

Recommended: shard format

After preparing the dataset, write the generated data index into the following configuration file:

OSUM-Pangu/conf/data_s2t_tmp.yaml

2. Start Training

Run the main training script:

OSUM-Pangu/train.sh

Citation

@misc{liao2026osumpanguopensourcemultidimensionspeech,
      title={OSUM-Pangu: An Open-Source Multidimension Speech Understanding Foundation Model Built upon OpenPangu on Ascend NPUs}, 
      author={Yujie Liao and Xuelong Geng and Hongfei Xue and Shuiyuan Wang and Lei Xie},
      year={2026},
      eprint={2603.10862},
      archivePrefix={arXiv},
      primaryClass={cs.SD},
      url={https://arxiv.org/abs/2603.10862}, 
}

Contact Us

If you are interested in leaving a message to our research team, feel free to email liaoyujie@mail.nwpu.edu.cn.

<a href="http://www.nwpu-aslp.org/">
    <img src="https://github.com/ASLP-lab/OSUM-Pangu/raw/main/images/aslp.png" width="400"/>
</a>

Core symbols most depended-on inside this repo

to_numpy
called by 55
wenet/bin/export_onnx_cpu.py
decode
called by 49
wenet/transformer/asr_model.py
make_pad_mask
called by 42
wenet/utils/mask.py
get_feat_from_wav_path
called by 40
common_utils/utils4infer.py
log_softmax
called by 32
wenet/transformer/ctc.py
update
called by 29
wenet/dataset/dataset.py
keys
called by 20
tools/compute-cer.py
open_or_fd
called by 16
wenet/dataset/kaldi_io.py

Shape

Method 654
Function 544
Class 199

Languages

Python100%

Modules by API surface

wenet/utils/scheduler.py51 symbols
wenet/bin/export_onnx_bpu.py46 symbols
wenet/dataset/datapipes.py41 symbols
wenet/dataset/process/processor_language_think.py36 symbols
wenet/dataset/process/processor.py36 symbols
wenet/dataset/process/processor_tag_think.py35 symbols
wenet/dataset/kaldi_io.py35 symbols
wenet/dataset/process/processor_tmp.py34 symbols
wenet/transducer/predictor.py33 symbols
wenet/finetune/lora/layers.py33 symbols
wenet/utils/train_utils.py25 symbols
wenet/transformer/subsampling.py25 symbols

For agents

$ claude mcp add OSUM-Pangu \
  -- python -m otcore.mcp_server <graph>

⬇ download graph artifact