AI-hardware en softwarestack
Waarom maakt W4A16-kwantisatie, waarbij alleen de gewichten 4 bits krijgen en de activaties FP16 blijven, de decode bij de inferentie van grote taalmodellen bijzonder snel?
- AOmdat een 4-bits vermenigvuldiging nauwkeuriger is dan een 16-bits vermenigvuldiging
- BOmdat decode geheugenbegrensd is en de afname van het gelezen gewichtsvolume tot een vierde zich rechtstreeks in snelheid vertaalt
- COmdat het aantal bewerkingen afneemt als de activaties FP16 zijn
Antwoord en toelichting tonen
Antwoord: B. Omdat decode geheugenbegrensd is en de afname van het gelezen gewichtsvolume tot een vierde zich rechtstreeks in snelheid vertaalt
Decode leest voor elk token alle gewichten en wordt in snelheid bepaald door de geheugenbandbreedte. Met 4-bits gewichten wordt de gelezen hoeveelheid een vierde, en ook als ze in de chip weer naar FP16 worden omgezet voordat wordt gerekend, wordt de totale tijd sterk korter. Bij de rekenbegrensde prefill is dit effect klein en werkt kwantisatie waarbij de berekening zelf met lage precisie wordt uitgevoerd beter. Welke kwantisatiemethode effectief is, hangt dus af van waar de beperking ligt.