Műszaki ÚTMUTATÓ

SwiGLU és kapuzott aktiválások

A SwiGLU egy kapuzott aktiválási funkció, amely megszorozza a bemenet egy lineáris vetületét egy Swish által aktivált második vetülettel, tanulható, adatfüggő kapuként működik a transzformátor előrecsatolt rétegeiben.

Áttekintés

A SwiGLU egy kapuzott aktiválási funkció, amely megszorozza a bemenet egy lineáris vetületét egy Swish által aktivált második vetülettel, tanulható, adatfüggő kapuként működik a transzformátor előrecsatolt rétegeiben. Folyamatosan javítja a nyelvi modell minőségét, ezért szinte minden modern LLM használja.

A SwiGLU és a Gated Activations egy olyan műszaki építőelem, amely nagymértékben befolyásolja a modell minőségét, az infrastruktúra költségeit, a késleltetést és a megbízhatóságot.

Mély merülés

A szabványos transzformátor-előrecsatoló blokk két lineáris rétegből áll, közöttük egy ReLU vagy GELU. Kapuzott lineáris egységek, Dauphin et al. 2016-ban az első vetületet osszuk két részre, és az egyik felét használjuk a másik kapuhoz elemenkénti szorzással. A Noam Shazeer által 2020-ban népszerűsített SwiGLU a Swish (SiLU) függvényt használja ehhez a kapuhoz: output = (Swish(xW) * (xV)) W2, kettő helyett három súlymátrixszal. A kapuzás lehetővé teszi a hálózat számára, hogy dimenziónként szelektíven továbbítsa vagy elnyomja az információkat. Mivel a harmadik mátrix hozzáadása növeli a paramétereket, az implementációk nagyjából kétharmadára csökkentik a rejtett dimenziót, így a teljes számítás összehasonlítható marad a GELU MLP-vel. Shazeer kísérletei mérhető zavartságnövekedést mutattak, és a LLaMA, a PaLM és a Mistral is ezt alkalmazta.

Technikai betekintés

A Swish egy x * sigmoid(beta*x), egy sima, nem monoton függvény, amely a ReLU-val ellentétben kis negatív értékeket enged át. A SwiGLU-ban a 'gate' ág Swish(xW) 0-hoz vagy 1-hez közeli értékeket állít elő, amelyek elemenként megszorozzák az 'érték' ágat xV, így minden rejtett egység hozzájárulását egy tanult, bemenettől függő jel modulálja. A harmadik súlymátrix a költség; a kétharmadnyi rejtett méretű trükk a FLOP költségvetését egy vanília előrecsatoló réteghez igazítja.

A SwiGLU és a kapuzott aktiválások elsajátítása

A SwiGLU egy kapuzott aktiválási funkció, amely megszorozza a bemenet egy lineáris vetületét egy Swish által aktivált második vetülettel, tanulható, adatfüggő kapuként működik a transzformátor előrecsatolt rétegeiben. Folyamatosan javítja a nyelvi modell minőségét, ezért szinte minden modern LLM használja. A SwiGLU és a Gated Activations egy olyan műszaki építőelem, amely nagymértékben befolyásolja a modell minőségét, az infrastruktúra költségeit, a késleltetést és a megbízhatóságot. A mély megértés érdekében a SwiGLU-t és a Gated Activations-t működési modellként kell kezelni, nem pedig egyetlen funkcióként: határozza meg a kívánt eredményeket, tisztázza a feltételezéseket, és válassza szét azt, amit a rendszer megbízhatóan képes elvégezni, attól, ami még szakértői megítélést igényel.

A gyakorlatban a SwiGLU-t és a Gated Activations-t használó erős csapatok optimalizálják az architektúrát, az adatokat és az infrastruktúrát a megbízhatóság és a költségek szempontjából. Dokumentálják az explicit sikerkritériumokat, tesztelik a valósághű adatokat és munkafolyamatokat, és a megfigyelt hibaminták alapján iterálnak, nem pedig egyszeri benchmark győzelmek alapján. Ez az a hely, ahol az elméleti megértés tartós képességgé válik a termék, a politika és a műveletek között.

Az építészeti döntések évekig növelik a teljesítményt és a működési költségeket. Ugyanakkor az egyik benchmark optimalizálása elrejtheti a rendszer általános gyengeségeit. A legrugalmasabb megközelítés a kísérleti sebesség és az irányítási fegyelem kombinálása: kísérleti kísérletek futtatása, bizonyítékok rögzítése, döntési naplók közzététele és a biztosítékok folyamatos frissítése a modell viselkedésének, a felhasználói elvárásoknak és a szabályozási követelményeknek megfelelően.

Stratégiai hatás

Az építészeti döntések évekig növelik a teljesítményt és a működési költségeket.

Az építészeti döntések évekig növelik a teljesítményt és a működési költségeket. A kiváló minőségű telepítéseknél ez mérhető működési szabályokká, tulajdonosi határokká és ismétlődő felülvizsgálati rituálékká alakul át, így a csapatok növelhetik a bizalmat a kétértelműség skálázása helyett.

A technikai oktatás segít a csapatoknak a megfelelő verem kiválasztásában, nem csak a legújabb készletben.

A technikai oktatás segít a csapatoknak a megfelelő verem kiválasztásában, nem csak a legújabb készletben. A kiváló minőségű telepítéseknél ez mérhető működési szabályokká, tulajdonosi határokká és ismétlődő felülvizsgálati rituálékká alakul át, így a csapatok növelhetik a bizalmat a kétértelműség skálázása helyett.

A jobb mérnöki döntések csökkentik a termelés megbízhatósági incidenseit.

A jobb mérnöki döntések csökkentik a termelés megbízhatósági incidenseit. A kiváló minőségű telepítéseknél ez mérhető működési szabályokká, tulajdonosi határokká és ismétlődő felülvizsgálati rituálékká alakul át, így a csapatok növelhetik a bizalmat a kétértelműség skálázása helyett.

A SwiGLU és a kapuzott aktiválások jövője

A SwiGLU az alapértelmezett MLP a nyílt súlyú LLM-ekben, és nem valószínű, hogy hamarosan kiszorítják. Az aktív irányok közé tartoznak a GeGLU és a ReGLU változatok, az egyesített GPU-kernelek, amelyek mindkét vetületet egy menetben számítják ki, valamint a kapuzott MLP-k és a szakértők keverékének kombinálása, így minden szakértő maga SwiGLU blokk. A kutatók azt is vizsgálják, hogy a kapuzás miért segíti az optimalizálást, még olcsóbb kapuk tervezése a cél.

Valós megvalósítás

A LLaMA, a PaLM és a Mistral a GELU előrecsatolt réteget SwiGLU-ra cseréli, hogy csökkentse a zavartságot egyenlő számítás mellett

A rejtett dimenzió körülbelül kétharmadára (8/3 d) van méretezve, így az extra kapuzási mátrix nem növeli a FLOP-okat

A szakértői keverék modellek, mint például a Mixtral, a SwiGLU blokkokat használják szakértőnkénti előrecsatoló hálózatként

A Vision és a multimodális transzformátorok GeGLU/SwiGLU kapuzást kölcsönöznek MLP-alrétegeik fejlesztéséhez

Megvalósítási minták

SwiGLU és kapuzott aktiválások a gyakorlatban

A LLaMA, a PaLM és a Mistral a GELU előrecsatoló réteget SwiGLU-ra cseréli, hogy csökkentse a zavartságot egyenlő számítás mellett.

A LLaMA, a PaLM és a Mistral a GELU előrecsatolási réteget SwiGLU-ra cseréli, hogy csökkentse a zavartságot egyenlő számítás mellett. A csapatok általában jobb eredményeket érnek el, ha előre meghatározzák a minőségi küszöböket, megtartják az emberi eszkalációs útvonalat az éles esetekben, és nyomon követik a termelékenység növekedését és a hibaköltségeket az idő múlásával.

SwiGLU és kapuzott aktiválások a gyakorlatban

A rejtett dimenzió körülbelül kétharmadára (8/3 d) van méretezve, így az extra kapuzási mátrix nem növeli a FLOP-okat.

A rejtett dimenzió körülbelül kétharmadára (8/3 d) van méretezve, így az extra kapuzási mátrix nem növeli a FLOP-okat. A csapatok általában jobb eredményeket érnek el, ha előre meghatározzák a minőségi küszöböket, megtartják az emberi eszkalációs útvonalat az éles esetekben, és nyomon követik mind a termelékenységnövekedést, mind a hibaköltségeket az idő múlásával.

SwiGLU és kapuzott aktiválások a gyakorlatban

A szakértői keverék modellek, mint például a Mixtral, a SwiGLU blokkokat használják szakértőnkénti előrecsatolási hálózatként.

A vegyes szakértői modellek, mint például a Mixtral, a SwiGLU blokkokat használják szakértőnkénti előrecsatolási hálózatként A csapatok általában jobb eredményeket érnek el, ha előre meghatározzák a minőségi küszöbértékeket, emberi eszkalációs utat tartanak a szélsőséges eseteknél, és nyomon követik a termelékenység növekedését és a hibaköltségeket az idő múlásával.

SwiGLU és kapuzott aktiválások a gyakorlatban

A Vision és a multimodális transzformátorok GeGLU/SwiGLU kapuzást kölcsönöznek MLP-alrétegeik fejlesztéséhez.

A Vision és a multimodális transzformátorok GeGLU/SwiGLU kapuzást kölcsönöznek MLP-alrétegeik fejlesztéséhez. A csapatok általában jobb eredményeket érnek el, ha előre meghatározzák a minőségi küszöböket, megtartják az emberi eszkalációs utat az éles esetekben, és nyomon követik mind a termelékenységnövekedést, mind a hibaköltségeket az idő múlásával.

Kockázatok és védőkorlátok

!

Egy benchmark optimalizálása elrejtheti a rendszer általános hiányosságait.

!

Az infrastrukturális és karbantartási költségeket gyakran alábecsülik.

!

A biztonsági és megfigyelhetőségi hiányosságok a rendszerek bonyolultabbá válásával nőhetnek.

Végrehajtási ütemterv

1

A megvalósítás előtt határozza meg a késleltetési, minőségi és költségcélokat.

A megvalósítás előtt határozza meg a késleltetési, minőségi és költségcélokat. Minden lépést bizonyítékkapuként kell kezelni: ha a feltételek nem teljesülnek, szüneteltesse a közzétételt, zárja be a rést, és csak ezután bővítse a felhasználást.

2

Benchmark reális terhelési és adatviszonyok mellett.

Benchmark reális terhelési és adatviszonyok mellett. Minden lépést bizonyítékkapuként kell kezelni: ha a feltételek nem teljesülnek, szüneteltesse a közzétételt, zárja be a rést, és csak ezután bővítse a felhasználást.

3

Műszerfigyelés a hibák, az eltolódás és a felhasználói hatások szempontjából.

Műszerfigyelés a hibák, az eltolódás és a felhasználói hatások szempontjából. Minden lépést bizonyítékkapuként kell kezelni: ha a feltételek nem teljesülnek, szüneteltesse a közzétételt, zárja be a rést, és csak ezután bővítse a felhasználást.

4

A méretezés előtt készítse elő a visszagörgetési és az incidensre adott válaszútvonalakat.

A méretezés előtt készítse elő a visszagörgetési és az incidensre adott válaszútvonalakat. Minden lépést bizonyítékkapuként kell kezelni: ha a feltételek nem teljesülnek, szüneteltesse a közzétételt, zárja be a rést, és csak ezután bővítse a felhasználást.

Folytassa a felfedezést