SpeechColab

non-profit

AI & ML interests

Machine Learning for Audio/Speech

Recent Activity

speechcolab's activity

sanchit-gandhiĀ 
posted an update 10 months ago
view post
Post
Why does returning timestamps help Whisper reduce hallucinations? šŸ§

Empirically, most practitioners have found that setting return_timestamps=True helps reduce hallucinations, particularly when doing long-form evaluation with Transformersā€™ ā€œchunkedā€ algorithm.

But why does this work?..

My interpretation is that forcing the model to predict timestamps is contradictory to hallucinations. Suppose you have the transcription:
The cat sat on the on the on the mat.

Where we have a repeated hallucination for ā€œon theā€. If we ask the model to predict timestamps, then the ā€œon theā€ has to contribute to the overall segment-level timing, e.g.:
<|0.00|> The cat sat on the on the on the mat.<|5.02|>

However, itā€™s impossible to fit 3 copies of ā€œon theā€ within the time allocation given to the segment, so the probability for this hallucinatory sequence becomes lower, and the model actually predicts the correct transcription with highest probability:
<|0.00|> The cat sat on the mat.<|5.02|>

In this sense, the end timestamp is of the opposite of the initial timestamp constraint they describe in Section 4.5 of the paper Robust Speech Recognition via Large-Scale Weak Supervision (2212.04356) ā†’ it helps the model remove extra words at the end of the sequence (rather than the initial timestamp which helps when the model ignores words at the start), but the overall principle is the same (using timestamps to improve the probability of more realistic sequences).

Leaving it open to you: why do you think timestamps reduces Whisper hallucinations?
Ā·