Gå til innholdet

Top-p-sampling

Valg av neste token fra en liten mengde sannsynlige kandidater som til sammen dekker en sannsynlighetsterskel.

Også kjent somNucleus Sampling
Maskinlæring

Hvilke tokener er igjen til utvalget?

Top-p-sampling, også kalt Nucleus Sampling, begrenser kandidatene til neste token ut fra deres samlede sannsynlighet. De mest sannsynlige tokenene tas med til den minste tilsvarende mengden minst dekker terskelen p. Deretter trekkes det fra mengden med omskalerte sannsynligheter. Antall kandidater kan endre seg ved hvert trinn. Samspillet med temperatur eller andre filtre avhenger av implementeringen; p er ikke andelen innholdsmessig riktige svar.

Et fiktivt regneeksempel omfatter bare kandidatene A, B og C med sannsynlighetene 0,5, 0,3 og 0,2. Ved p = 0,75 dekker A og B til sammen 0,8 og utgjør utvalgsmengden; C utelates. De gjenværende kandidatenes sannsynligheter normaliseres til summen én før trekningen. Dette beskriver ett enkelt utvalgstidspunkt. Ved neste trinn kan fordelingen se annerledes ut. Et tekstforslag til virksomheten må derfor fortsatt kontrolleres for innhold, udokumenterte tillegg og ønsket format.

Kilder