We propose to model parallel streams of data, such as overlapped speech, using shuffles. Specifically, this paper shows how the shuffle product and partial order finite-state automata (FSAs) can be used for alignment and speaker-attributed transcription of overlapped speech. We train using the total score on these FSAs as a loss function, marginalizing over all possible serializations of overlapping sequences at subword, word, and phrase levels. To reduce graph size, we impose temporal constraints by constructing partial order FSAs. We address speaker attribution by modeling (token, speaker) tuples directly. Viterbi alignment through the shuffle product FSA directly enables one-pass alignment. We evaluate performance on synthetic LibriSpeech overlaps. To our knowledge, this is the first algorithm that enables single-pass alignment of multi-talker recordings. All algorithms are implemented using k2 / Icefall.
Figures & tables
Figure 1: FSAs representing different utterance group serializations. Tuples on states represent indices into the two serialized sequences. Colors represent unique sources. (a) The shuffle product FSA between two sequences y_{1}=abc,y_{2}=\color[rgb]{1,0,0}xy . (b) The pruned shuffle FSA obtainable from application of partial order constraints (see Section 3.2 ). (c) The subgraph corresponding to utterance-level serialized output training (SOT). (d) The subgraph corresponding to token-level SOT, assuming token order a\leq\color[rgb]{1,0,0}x\color[rgb]{0,0,0}\leq b\leq\color[rgb]{1,0,0}y\color[rgb]{0,0,0}\leq c .
Figure 2: The compact selfless label topology from [ 33 ] . Note that, different from the traditional CTC-topology, it forces tokens to appear on exactly one frame, and forces ⊘ between every token in the transcript, i.e., between “a” and a subsequent “b”.
Statistics of the Avg. Utterance Group by Datset
Dataset
# Segs
# Spks
Dur (s)
Spk / s
% Ovlp
2spk Ovlp
8
2
39.1
1.4
42.9
3spk Ovlp
8
3
35.4
1.7
55.0
Libri2Mix
2
2
8.4
1.6
64.2
Libri3Mix
3
3
9.0
2.2
72.2
Synth
2
2
9.2
1.5
52.7
Table 1: Statistics of evaluation datasets. Spk / s denotes the speaker density, or average number of active speakers per frame; overlap percentage denotes the fraction of time with more than one active speaker.
Token Order
Synth
LibriMix Test
LibriSpeech
Avg
3 Clean
2 Clean
2 Both
Clean
Other
SOT
43.7
71.9
50.5
65.8
6.4
12.0
41.7
κ = 0s
36.6
69.6
39.3
52.8
7.0
12.2
36.3
κ = 1s
20.7
61.4
24.1
41.5
6.0
10.6
27.4
κ = 2s
19.2
60.5
22.3
38.7
5.8
10.5
26.2
κ = 4s
18.5
60.6
21.1
39.7
5.7
10.4
26.0
Table 2: Test set results with tcpWER collar = 5s for WavLM Base at 25 Hz using the factored speaker attribution model p(σt∣x,⊘ˉ)×p(υt∣x) . This table compares serialization strategies (Token Order). All values are tcpWER (%).
System
WavLM
Loss
Spk Order
Synth
LibriMix Test
LibriSpeech Test
Average
3 Clean
2 Clean
2 Both
Clean
Other
Large
shuffle
start
12.7
60.9
17.6
30.2
5.4
7.9
22.5
sd-ctc
13.3
58.5
16.8
32.3
5.2
8.2
22.4
shuffle
length
11.8
44.3
8.1
25.3
5.3
8.2
17.2
sd-ctc
11.1
43.8
7.8
28.2
5.3
8.2
17.4
Table 3: Test set results with tcpWER (collar = 5s). All models use the factored speaker attribution model p(σt∣x,⊘ˉ)×p(υt∣x) and a 50 Hz frame rate. The table compares speaker ordering strategies and training losses. All values are WER (%).
Align Graph
Collar ( κ )
Model
2spk overlap
3spk overlap
IoU ↑
BE ↓
τ % ↓
IoU ↑
BE ↓
τ % ↓
G(Y′)
2s
CTC
56.2
89
13.8
45.0
138
41.2
G(Y′)
2s
Shuffle
61.8
69
6.3
58.2
81
16.9
32s
61.7
70
6.2
-
-
-
2s
SD-CTC
63.4
66
6.2
59.5
78
16.9
32s
63.2
66
6.2
-
-
-
Table 4: Alignment results on synthetic LibriSpeech mixtures. Performance in terms of the IoU, BE in milliseconds, and the Kendall- τ distance, ( τ ). Alignment error is measured against MFA word-level alignments on the corresponding single-speaker recordings.
Decoding
Synth
LibriMix Test
LibriSpeech
Avg
3 Clean
2 Clean
2 Both
Clean
Other
Greedy
11.8
44.3
8.1
25.3
5.3
8.2
17.2
tgt spk HLG
9.6
42.5
6.2
24.2
3.6
6.1
14.8
+ oracle # spks
9.6
42.5
6.2
24.2
3.6
6.1
14.8
UME [ 53 ]
-
15.9
6.4
19.6
-
-
-
Yang MTASR [ 54 ]
-
-
5.1
-
-
-
-
Table 5: Test set word error rates (%). All results are tcpWER (collar=5s).
The Viterbi algorithm has been previously used to perform forced alignment of audio to text to mine training data from online resources. However, many existing implementations have quadratic time and space complexity, scaling poorly to long input sequences. We propose two optimizations to address this issue. First, we apply the Hirschberg algorithm to perform the alignment in place using linear memory. Second, we model the alignment between speech and text as a constrained random walk, allowing us to prune the search space with arbitrary confidence while accounting for transcription errors. The Hirschberg optimization reduces memory usage from 140 GB to 5 MB for three-hour inputs while producing identical alignments in one-third the time of torchaudio when both run on a CPU. We achieve an additional 2x speedup with pruning on inputs longer than 20 minutes while preserving alignment accuracy in more than 98% of tested cases.
Lawry Sorenson, Michael Crandall, Eric K. Ringger +1
Brigham Young University Computer Science Department Provo, UT, United States
Word-level forced alignment estimates when each transcript word occurs in an audio recording. It underpins text-based media editing, subtitling, speech-data curation, and phonetic analysis. Existing evaluations understate the difficulty of forced alignment by relying on short, clean speech, perfect transcripts, and metrics that obscure consequential alignment errors. In contrast, real-world media and data-processing pipelines operate on long and diverse recordings. Additionally, forced aligners often operate on error-prone automatic speech recognition (ASR) output. We address these gaps with improved evaluation metrics, a scoring protocol for real ASR transcripts, and AlignBench, a benchmark spanning diverse speaker, acoustic, and text conditions. We further introduce FuseAlign, a transformer-based aligner trained on large-scale pseudo-labeled speech with online label correction. FuseAlign performs joint contextualization of audio and text for the localization of coarse words. The model then refines boundaries at millisecond resolution and detects missing transcript words in the audio without lexicon-based or Viterbi decoding. On AlignBench, FuseAlign substantially outperforms all baselines and remains robust under real ASR transcripts. Ablations show that convolutional upsampling and EMA-snapshot label correction matter more than model properties such as parameter count.
We study timestamped speaker-attributed automatic speech recognition (SA-ASR) for long-form, multi-party speech with overlap. In this setting, chunk-wise inference must preserve meeting-level speaker identity consistency while producing time-stamped, speaker-labeled transcripts. Prior Speech-LLM systems tend to prioritize either local diarization or global labeling, lacking the ability to jointly model fine-grained temporal boundaries and robust cross-chunk identity linking. We propose G-STAR, an end-to-end framework that couples a cache-conditioned speaker-tracking module with a Speech-LLM transcription backbone. The tracker provides structured speaker cues with temporal grounding, and the LLM generates attributed text conditioned on these cues. G-STAR supports component-wise optimization and joint end-to-end training, enabling flexible learning under heterogeneous supervision and domain shift. Under chunk-wise decoding protocols, experiments on both oracle-segmented local evaluation and full-meeting global evaluation show strong speaker-attributed transcription performance.
Jing Peng, Ziyi Chen, Haoyu Li +7
Shanghai Jiao Tong University · Shenzhen Research Institute of Big Data · Nanjing University +2