Skip to content

Update diarizer timeline sync and LS-EEND finalization#421

Merged
Alex-Wengg merged 15 commits intomainfrom
codex/publicly-mutable-diarizer-timeline
Mar 25, 2026
Merged

Update diarizer timeline sync and LS-EEND finalization#421
Alex-Wengg merged 15 commits intomainfrom
codex/publicly-mutable-diarizer-timeline

Conversation

@SGD2718
Copy link
Copy Markdown
Collaborator

@SGD2718 SGD2718 commented Mar 24, 2026

Summary

  • add coverage for diarizer timeline synchronization, tentative timeline compatibility, and Sortformer streaming flush behavior
  • move LS-EEND tail-flush finalization into the streaming session so offline and streaming paths share the same finalize semantics
  • update API and diarization docs for explicit endingOnTime, timeline behavior, and finalization details

Verification

  • swift build
  • swift test --filter SortformerTimelineTests
  • swift test --filter SortformerStreamingIntegrationTests
  • swift test --filter LSEENDIntegrationTests.testDiarizerStreamingFinalizeMatchesProcessComplete
  • swift test --filter LSEENDIntegrationTests.testStreamingSessionMatchesOfflineInferenceOnRealFixtureAudio
  • swift test --filter LSEENDIntegrationTests.testDiarizerProcessEndingOnTimeAlignsVisibleRange

Open with Devin

Copilot AI review requested due to automatic review settings March 24, 2026 22:55
@SGD2718 SGD2718 self-assigned this Mar 24, 2026
@SGD2718 SGD2718 requested a review from Alex-Wengg March 24, 2026 22:57

This comment was marked as resolved.

Copy link
Copy Markdown
Contributor

@devin-ai-integration devin-ai-integration bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

✅ Devin Review: No Issues Found

Devin Review analyzed this PR and found no bugs or issues to report.

Open in Devin Review

@github-actions
Copy link
Copy Markdown

github-actions bot commented Mar 24, 2026

Sortformer High-Latency Benchmark Results

ES2004a Performance (30.4s latency config)

Metric Value Target Status
DER 33.4% <35%
Miss Rate 24.4% - -
False Alarm 0.2% - -
Speaker Error 8.8% - -
RTFx 9.3x >1.0x
Speakers 4/4 - -

Sortformer High-Latency • ES2004a • Runtime: 5m 23s • 2026-03-25T22:43:45.379Z

@github-actions
Copy link
Copy Markdown

github-actions bot commented Mar 24, 2026

Parakeet EOU Benchmark Results ✅

Status: Benchmark passed
Chunk Size: 320ms
Files Tested: 100/100

Performance Metrics

Metric Value Description
WER (Avg) 0.00% Average Word Error Rate
WER (Med) 0.00% Median Word Error Rate
RTFx 0.00x Real-time factor (higher = faster)
Total Audio 0.0s Total audio duration processed
Total Time 0.0s Total processing time

Streaming Metrics

Metric Value Description
Avg Chunk Time 0.000s Average chunk processing time
Max Chunk Time 0.000s Maximum chunk processing time
EOU Detections 0 Total End-of-Utterance detections

Test runtime: 0m36s • 03/25/2026, 06:36 PM EST

RTFx = Real-Time Factor (higher is better) • Processing includes: Model inference, audio preprocessing, state management, and file I/O

@github-actions
Copy link
Copy Markdown

github-actions bot commented Mar 24, 2026

PocketTTS Smoke Test ✅

Check Result
Build
Model download
Model load
Synthesis pipeline
Output WAV ✅ (183.8 KB)

Runtime: 0m36s

Note: PocketTTS uses CoreML MLState (macOS 15) KV cache + Mimi streaming state. CI VM lacks physical GPU — audio quality may differ from Apple Silicon.

@github-actions
Copy link
Copy Markdown

github-actions bot commented Mar 24, 2026

VAD Benchmark Results

Performance Comparison

Dataset Accuracy Precision Recall F1-Score RTFx Files
MUSAN 92.0% 86.2% 100.0% 92.6% 398.2x faster 50
VOiCES 92.0% 86.2% 100.0% 92.6% 413.7x faster 50

Dataset Details

  • MUSAN: Music, Speech, and Noise dataset - standard VAD evaluation
  • VOiCES: Voices Obscured in Complex Environmental Settings - tests robustness in real-world conditions

✅: Average F1-Score above 70%

@github-actions
Copy link
Copy Markdown

github-actions bot commented Mar 24, 2026

Offline VBx Pipeline Results

Speaker Diarization Performance (VBx Batch Mode)

Optimal clustering with Hungarian algorithm for maximum accuracy

Metric Value Target Status Description
DER 14.5% <20% Diarization Error Rate (lower is better)
RTFx 5.45x >1.0x Real-Time Factor (higher is faster)

Offline VBx Pipeline Timing Breakdown

Time spent in each stage of batch diarization

Stage Time (s) % Description
Model Download 11.450 5.9 Fetching diarization models
Model Compile 4.907 2.5 CoreML compilation
Audio Load 0.034 0.0 Loading audio file
Segmentation 21.333 11.1 VAD + speech detection
Embedding 191.744 99.6 Speaker embedding extraction
Clustering (VBx) 0.688 0.4 Hungarian algorithm + VBx clustering
Total 192.570 100 Full VBx pipeline

Speaker Diarization Research Comparison

Offline VBx achieves competitive accuracy with batch processing

Method DER Mode Description
FluidAudio (Offline) 14.5% VBx Batch On-device CoreML with optimal clustering
FluidAudio (Streaming) 17.7% Chunk-based First-occurrence speaker mapping
Research baseline 18-30% Various Standard dataset performance

Pipeline Details:

  • Mode: Offline VBx with Hungarian algorithm for optimal speaker-to-cluster assignment
  • Segmentation: VAD-based voice activity detection
  • Embeddings: WeSpeaker-compatible speaker embeddings
  • Clustering: PowerSet with VBx refinement
  • Accuracy: Higher than streaming due to optimal post-hoc mapping

🎯 Offline VBx Test • AMI Corpus ES2004a • 1049.0s meeting audio • 213.8s processing • Test runtime: 3m 44s • 03/25/2026, 06:50 PM EST

@github-actions
Copy link
Copy Markdown

github-actions bot commented Mar 24, 2026

ASR Benchmark Results ✅

Status: All benchmarks passed

Parakeet v3 (multilingual)

Dataset WER Avg WER Med RTFx Status
test-clean 0.57% 0.00% 5.36x
test-other 1.19% 0.00% 3.45x

Parakeet v2 (English-optimized)

Dataset WER Avg WER Med RTFx Status
test-clean 0.80% 0.00% 4.99x
test-other 1.00% 0.00% 3.25x

Streaming (v3)

Metric Value Description
WER 0.00% Word Error Rate in streaming mode
RTFx 0.55x Streaming real-time factor
Avg Chunk Time 1.630s Average time to process each chunk
Max Chunk Time 1.840s Maximum chunk processing time
First Token 1.931s Latency to first transcription token
Total Chunks 31 Number of chunks processed

Streaming (v2)

Metric Value Description
WER 0.00% Word Error Rate in streaming mode
RTFx 0.50x Streaming real-time factor
Avg Chunk Time 1.792s Average time to process each chunk
Max Chunk Time 2.107s Maximum chunk processing time
First Token 1.769s Latency to first transcription token
Total Chunks 31 Number of chunks processed

Streaming tests use 5 files with 0.5s chunks to simulate real-time audio streaming

25 files per dataset • Test runtime: 7m31s • 03/25/2026, 06:48 PM EST

RTFx = Real-Time Factor (higher is better) • Calculated as: Total audio duration ÷ Total processing time
Processing time includes: Model inference on Apple Neural Engine, audio preprocessing, state resets between files, token-to-text conversion, and file I/O
Example: RTFx of 2.0x means 10 seconds of audio processed in 5 seconds (2x faster than real-time)

Expected RTFx Performance on Physical M1 Hardware:

• M1 Mac: ~28x (clean), ~25x (other)
• CI shows ~0.5-3x due to virtualization limitations

Testing methodology follows HuggingFace Open ASR Leaderboard

@github-actions
Copy link
Copy Markdown

github-actions bot commented Mar 24, 2026

Speaker Diarization Benchmark Results

Speaker Diarization Performance

Evaluating "who spoke when" detection accuracy

Metric Value Target Status Description
DER 15.1% <30% Diarization Error Rate (lower is better)
JER 24.9% <25% Jaccard Error Rate
RTFx 24.94x >1.0x Real-Time Factor (higher is faster)

Diarization Pipeline Timing Breakdown

Time spent in each stage of speaker diarization

Stage Time (s) % Description
Model Download 9.431 22.4 Fetching diarization models
Model Compile 4.042 9.6 CoreML compilation
Audio Load 0.052 0.1 Loading audio file
Segmentation 12.619 30.0 Detecting speech regions
Embedding 21.031 50.0 Extracting speaker voices
Clustering 8.412 20.0 Grouping same speakers
Total 42.078 100 Full pipeline

Speaker Diarization Research Comparison

Research baselines typically achieve 18-30% DER on standard datasets

Method DER Notes
FluidAudio 15.1% On-device CoreML
Research baseline 18-30% Standard dataset performance

Note: RTFx shown above is from GitHub Actions runner. On Apple Silicon with ANE:

  • M2 MacBook Air (2022): Runs at 150 RTFx real-time
  • Performance scales with Apple Neural Engine capabilities

🎯 Speaker Diarization Test • AMI Corpus ES2004a • 1049.0s meeting audio • 42.1s diarization time • Test runtime: 4m 42s • 03/25/2026, 06:37 PM EST

@github-actions
Copy link
Copy Markdown

github-actions bot commented Mar 24, 2026

Qwen3-ASR int8 Smoke Test ✅

Check Result
Build
Model download
Model load
Transcription pipeline
Decoder size 571 MB (vs 1.1 GB f32)

Runtime: 4m23s

Note: CI VM lacks physical GPU — CoreML MLState (macOS 15) KV cache produces degraded results on virtualized runners. On Apple Silicon: ~1.3% WER / 2.5x RTFx.

devin-ai-integration[bot]

This comment was marked as resolved.

@SGD2718
Copy link
Copy Markdown
Collaborator Author

SGD2718 commented Mar 25, 2026

Why is it stuck

devin-ai-integration[bot]

This comment was marked as resolved.

@SGD2718 SGD2718 force-pushed the codex/publicly-mutable-diarizer-timeline branch from 445c82b to 59a63e2 Compare March 25, 2026 02:43
devin-ai-integration[bot]

This comment was marked as resolved.

devin-ai-integration[bot]

This comment was marked as resolved.

SGD2718 added 2 commits March 25, 2026 12:09
Updated comment to clarify the flushing of frames.
Updated the description of `finalizeSession()` to clarify its functionality.
devin-ai-integration[bot]

This comment was marked as resolved.

@Alex-Wengg Alex-Wengg merged commit d683525 into main Mar 25, 2026
12 checks passed
@Alex-Wengg Alex-Wengg deleted the codex/publicly-mutable-diarizer-timeline branch March 25, 2026 23:12
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants