Relative Time Intervals Representation for Word-level Timestamping with Masked Training
Organizations: Soochow University · University of Queenland · AISpeech Ltd · Jiangsu Key Lab of Language Computing
Abstract
Although Speech Large Language Models (SpeechLLMs) excel at speech understanding and generation, their capacity for fine-grained, temporally aligned outputs remains underexplored. Our work addresses this gap by enabling SpeechLLMs to jointly model speech content and temporal structure, effectively transforming them from content understanding machines" into temporal-aware content understanding machines". Specifically, we replace traditional absolute timestamps with relative timestamps, achieving a more compact vocabulary and stronger generalization capabilities. To efficiently infuse timestamp prediction ability into pre-trained large language models, we introduce a hybrid fine-tuning strategy: full-parameter fine-tuning of the timestamp-augmented embedding layer and language model head, combined with LoRA fine-tuning of the decoder layers. Moreover, we design a masked timestamp training objective, preventing the model from over-relying on ground-truth timestamps, and thereby enhancing robustness against noisy real-world annotations. Extensive experiments demonstrate that our approach achieves significant improvements in timestamp prediction accuracy while maintaining strong speech transcription performance.
Figures & tables
| Model | Tolerance (ms) | Librispeech | Wenet-Meeting | ||||
|---|---|---|---|---|---|---|---|
| P (%) | R (%) | Avg. Diff (ms) | P(%) | R (%) | Avg. Diff (ms) | ||
| Qwen2-Audio [ 4 ] | 80 | 0.00 | 0.00 | 1031.23 | 60.81 | 56.06 | 59.03 |
| 160 | 0.01 | 0.00 | 1031.18 | 76.07 | 70.13 | 49.11 | |
| 240 | 0.16 | 0.05 | 1030.42 | 82.30 | 75.87 | 43.07 | |
| WhisperTimestamed [ 15 ] | 80 | 2.70 | 2.36 | 131.92 | 3.54 | 1.51 | 164.09 |
| 160 | 10.09 | 8.83 | 120.52 | 24.30 | 10.38 | 134.03 | |
| Configuration | AISHELL-2 iOS (Chinese) | Common Voice (English) | ||||
|---|---|---|---|---|---|---|
| WER (%) | Precision | Recall | WER (%) | Precision | Recall | |
| Absolute Timestamp | 2.87 | 0.9544 | 0.9546 | 16.41 | 0.7861 | 0.7838 |
| - TS Loss | 2.96 | 0.9506 | 0.9447 | 18.63 | 0.7428 | 0.7407 |
| Relative Timestamp | 2.15 | 0.9763 | 0.9634 | 11.63 | 0.8770 | 0.8546 |
| - TS Loss | 2.31 | 0.9715 | 0.9548 | 12.66 | 0.8712 | 0.8463 |
| - Timestamp Masking | 2.56 | 0.9658 | 0.9578 | 14.47 | 0.8067 | 0.7853 |
| Method | AS-1 | AS-2 | Wenet | Libri | CV |
|---|---|---|---|---|---|
| (CN) | (CN) | (CN) | (EN) | (EN) | |
| Qwen2 Audio | 1.62 | 3.38 | 21.99 | 43.32 | 74.37 |
| SenseVoiceSmall | 3.10 | 3.87 | 7.54 | 3.80 | 15.97 |
| WhisperTimestamped | 12.46 | 8.08 | 56.66 | 20.50 | 38.09 |
| Absolute Timestamp | 1.38 | 2.87 | 11.30 | 9.11 | 16.41 |
| Ours (Relative Timestamp) | 1.26 | 2.15 | 5.56 | 2.78 | 11.63 |
Speech Language Models
Speech language models connect spoken audio with large language models for understanding, transcription, translation, and real-time dialogue.
See all 312 papers