]> git.djapps.eu Git - pkg/ggml/sources/ggml/commit
whisper : improve beam search candidate diversity (whisper/1947)
authorJosh Bleecher Snyder <redacted>
Sun, 10 Mar 2024 14:54:43 +0000 (07:54 -0700)
committerGeorgi Gerganov <redacted>
Sun, 10 Mar 2024 18:05:25 +0000 (20:05 +0200)
commitc623d5d73767891ad07455b1fb4a0f174484f830
treeb23c3a7c7c8a9c18cd494645938808d7b39c3690
parent86961f54d35b39919780111dff69a1a040c81d0a
whisper : improve beam search candidate diversity (whisper/1947)

As of #1486, whisper.cpp uses a unified KV cache with KQ masking.
As a result, depending on their location in the batch,
identical sequences in a batch can have slightly different outputs
due to floating point rounding errors during reduction.
See the discussion in #1941 for more details.

The beam search code used "has identical sum of log probabilities"
as a shorthand for "is an identical token sequence". However, per above,
identical tokens do not necessarily result in identical probabilities.

Instead, explicitly compare on sequences.
This is linear in cost when they are identical,
but the lengths are always small and the comparisons are cheap.

This increases diversity during beam search.

This improves output quality for some short samples I've been working
with, at no detectable performance cost.
I haven't checked against larger corpuses.

Fixes #1941
examples/whisper/whisper.cpp