skip to content
LMMs-Lab

Search

Tags #audio

  • Pix-Pin-2025-04-28-12-16-06.gif

    What is Aero Audio?

    Github | Playground | Models | Evaluation Results | Cookbook

    Aero-1-Audio is a compact audio model adept at various audio tasks, including speech recognition, audio understanding, and following audio instructions. It is part of the Aero-1 series, the first generation of lightweight multimodal models developed by LMMs-Lab, with future expansions planned across additional modalities.

    1. Built upon the Qwen-2.5-1.5B language model, Aero delivers strong performance across multiple audio benchmarks while remaining parameter-efficient, even compared with larger advanced models like Whisper and Qwen-2-Audio and Phi-4-Multimodal, or commercial services like ElevenLabs/Scribe.

    2. Aero is trained within one day on 16 H100 GPUs using just 50k hours of audio data. Our insight suggests that audio model training could be sample efficient with high quality and filtered data.

    3. Aero can accurately perform ASR and audio understanding on continuous audio inputs up to 15 minutes in length, which we find the scenario is still a challenge for other models.

    ASR & Audio Understanding Performance

    We evaluate our model performance on multiple dimensions and different benchmarks. Let’s first take a look at its overall performance compare with other models

    ASR-Understanding-Compare.png ASR-Detail.png

    Our model achieves a balance between performance and parameter efficiency. We evaluate it across multiple ASR and audio understanding benchmarks. On ASR tasks, our model attains the lowest WER scores on datasets such as AMI, LibriSpeech, and SPGISpeech. It also demonstrates strong audio understanding capabilities on various comprehension benchmarks. As illustrated in the plotted graph, our model falls within the highlighted triangular region that represents an optimal trade-off between parameter efficiency and performance.

    Data Distribution

    We present the contributions of our data mixture here. Our SFT data mixture includes over 20 publicly available datasets, and comparisons with other models highlight the data’s lightweight nature.

    Data-distribution.png training-time.png

    *The hours of some training datasets are estimated and may not be fully accurate
    One of the key strengths of our training recipe lies in the quality and quantity of our data. Our training dataset consists of approximately 5 billion tokens, corresponding to around 50,000 hours of audio. Compared to models such as Qwen-Omni and Phi-4, our dataset is over 100 times smaller, yet our model achieves competitive performance. All data is sourced from publicly available open-source datasets, highlighting the sample efficiency of our training approach. A detailed breakdown of our data distribution is provided below, along with comparisons to other models.

    What’s insightful

    In this release, our primary focus is on developing an audio model capable of handling multiple audio tasks. The following examples showcase its core abilities across tasks such as audio understanding and speech recognition. Most notably, we highlight the model’s capability to perform long-form ASR, as demonstrated in the example below.

    Long ASR

    A common approach for current long-form ASR tasks is to split the audio into smaller, processable chunks and perform ASR on each segment individually. However, with the advancement of large language models (LLMs), long-context understanding has become increasingly important. We argue that a model’s ability to process long audio sequences continuously is essential for effective audio understanding and should be considered a critical capability. To demonstrate this, we set up a simple use case using examples from an NVIDIA conference and calculate the WER with respect to the auto-generated YouTube subtitles.

    Long-ASR-eval.png

    The image above presents a heatmap comparison of different models performing ASR tasks on a video with varying audio input lengths. As shown in the heatmap, Qwen-Omni and Phi-4 exhibit instability across different lengths and do not consistently produce the desired output.

    Note: The ground truth is derived from the auto-generated subtitles downloaded from YouTube. Therefore, the WER does not necessarily imply that our model achieves perfect results, but rather demonstrates that our model is comparable to the YouTube ASR pipeline.

    Model’s Output

    Qwen Omni (12 minutes chunk)

    When processing the audio in 12-minute chunks, Qwen-Omni failed to recognize the full speech content and was only able to capture portions of the audio.

    Qwen Omni (12 minutes chunk)
    that’s like what’s going on why does itfocused on um ai and parallel parallelizable workloads but it’s still general to an extent it’s not as use case specific as something like grock with a queue that’s really designed to you know spit out tokens as fast as possible and that like is a goldilocks zone where it’s flexible enough to handle different workloads but not um but still much faster than um a traditional cpu and that google is one of the only companies that has a scaled internal custom silicon effort

    Phi-4-Multimodal (full chunk)

    When processing the full audio without splitting, the Phi-4-Multimodal model began to ignore the instructions and instead generated an overall summary of the audio.

    Phi-4-Multimodal (full chunk)
    The conversation covered Nvidia’s focus on inference over training, the partnership with GM, the release of GUT-N1 for humanoid robotics, and the impact of China’s AI initiatives on global chip demand.

    Aero (full chunk)

    Aero Audio is able to generate the complete ASR output and accurately identify the full transcript.

    Aero (full chunk)
    Welcome to the brainstorm episode eighty two frank downing joining us recap of nvidia’s gtc conference that is the gpu technology conference frank what happened what were the big takeaways i on my side i saw a gm and in video partnering but we can circle back to that what was … right nice timing good timing all right we’ll see everyone next week see everyone thank you

    Results on LibriSpeech Unchunked

    In the previous release, LibriSpeech split their audio files into smaller chunks and calculated the overall Word Error Rate (WER) based on these segmented samples. However, as we observed, it is straightforward to concatenate the chunks back into their original form, thereby creating a simple long-form Audio Speech Recognition benchmark. We evaluated various models on these benchmarks and found that their performance generally declined compared to their results on shorter samples. Among the models tested, our model achieved the best performance, showing the smallest drop in accuracy relative to the chunked version.

    LS.Clean LS.Other LS.Clean(Long) LS.Other(Long) Avg Diff
    Phi-4 1.68 3.83 11.51 24.72 30.72
    Qwen2-Audio-Instruct 3.59 7.46 93.01 93.63 175.59
    Qwen2.5-Omni 1.80 3.40 13.03 13.29 21.12
    Aero-1-Audio 1.49 3.17 5.31 11.71 12.36

    We present the evaluation of various models on the unchunked LibriSpeech dataset. The average result is calculated by averaging the WER score differences across the same splits. All models show some degradation when handling longer audio, whereas our model exhibits the least amount of performance drop.

    Evaluation Result

    We then present the full evaluation result here with the evaluation scores

    ASR Benchmarks

    Model Parameters Automatic Speech Recognition Average
    AMI Earnings22 LibriSpeech
    Clean
    LibriSpeech
    Other
    SPGispeech Tedlium
    ElevenLabs/Scribe N/A 14.43 12.14 1.79 3.31 3.30 3.17 6.36
    REV.AI/Fusion N/A 10.93 12.09 2.88 6.23 4.05 2.80 6.50
    OpenAI/Whisper-large-v3 1.5B 15.95 11.29 2.01 3.91 2.94 3.86 6.66
    Assembly.AI/AssemblyBest N/A 15.64 13.54 1.74 3.11 1.81 3.43 6.55
    Alibaba/Qwen2.5-Omni 7B 12.41 12.74 1.80 3.40 2.35 3.11 5.97
    Microsoft/Phi-4-Multimodal 4B+1.6B 11.45 10.50 1.67 3.82 3.11 2.89 5.57
    LMMs-Lab/Aero-1-Audio 1.5B 10.53 13.79 1.49 3.17 1.97 2.87 5.64

    We evaluate our model on AMI, Earnings22, LibriSpeech, SPGISpeech, and TedLium. Our model achieves the second-best WER score compared to other models, while maintaining a small and efficient size.

    Audio Understanding Result

    We then test our model’s understanding result across 3 dimensions, Audio Analysis and Understanding, Speech Instruction, and Audio Scene Understanding

    Model Parameters Audio Analysis and Understanding Speech Instruction Audio Scene Understanding Average
    AIR-Chat MMAU OpenHermes Alpaca Audio AIR-Foundation
    Speech Sound Music Mix Avg testmini test test Speech Sound Music
    Alibaba/Qwen2-Audio-Instruct 7B 7.2 7.0 6.8 6.8 6.9 49.2 46.8 49.2 62.9 55.4 56.8 56.7
    Alibaba/Qwen2.5-Omni 7B 6.8 5.7 4.8 5.4 5.7 65.6 57.2 57.4 67.2 76.3 63.0 64.4
    Microsoft/Phi-4-Multimodal 4B+1.6B 7.5 7.0 6.7 6.8 7.0 65.0 57.8 62.6 48.3 40.6 35.5 52.8
    Tencent/Ola 7B 7.3 6.4 5.9 6.0 6.4 70.3 62.6 62.8 58.8 70.4 53.1 63.2
    Tencent/Vita 1.5 7B 4.8 5.5 4.9 2.9 4.5 35.5 9.6 7.0 31.5 24.1 25.5 28.6
    InspirAI/Mini-Omni2 0.5B 3.6 3.5 2.6 3.1 3.2 - - - - - - -
    LMMs-Lab/Aero-1-Audio 1.5B 5.7 5.3 4.7 5.8 5.4 59.4 40.0 45.4 48.0 57.6 44.2 50.5

    We conducted evaluations on AIR-Bench-Chat and MMAU for audio analysis and understanding. Our model achieved an average score of 5.35, outperforming Mini-Omni2 and Vita. For Audio Instruction Following, we evaluated on OpenHermes and Alpaca-Audio, following the same pipeline as AudioBench. Our model demonstrates a strong ability to understand instructions in speech and provide correct responses. Additionally, when evaluated on AIR-Bench-Foundation for Audio Scene Understanding, our model outperformed Phi-4-Multimodal in the sound and music dimensions. Overall, the average score of our model indicates strong performance relative to other models with larger parameter sizes.

    Training Techniques

    Dynamic Batch Size

    We implemented a dynamic batching strategy based on the estimated token length to control the batch size per device. In many cases, using a fixed batch size requires setting it conservatively small to avoid out-of-memory (OOM) errors on longer samples, which leads to underutilization of computing resources. To address this, we group samples into batches such that the total token length stays within a predefined threshold, thereby minimizing computational waste and improving efficiency.

    Sequence Packing

    To further optimize dynamic batching, we implemented sequence packing for both the audio encoder and the language model, enabling larger batch sizes and faster training. This operation was then fused with the Liger kernel to achieve even higher throughput and lower memory usage. With a fixed packing length of 4096 to regulate the dynamic batch size, the average Model FLOP Utilization (MFU) was limited to 0.03. However, with sequence packing enabled, the average MFU increased to approximately 0.34, demonstrating a significant improvement in training efficiency.

    Packing Length Sequence Packing Num GPUs Avg MFU Zero OOM
    4096 FALSE 64 0.03 2 No
    32768 FALSE 64 NA 2 Yes
    32768 TRUE 32 0.34 2 No

    We tested our implementations on different settings to demonstrate the efficiency of our implementation

    Contributor List

    alphabetical order

    *main contributors