Keten-van-Model Leren voor TaalmodellenChain-of-Model Learning for Language Model
In dit artikel introduceren we een nieuw leerparadigma, genaamd Chain-of-Model (CoM), dat de causale relatie incorporeert in de verborgen toestanden van elke laag in een kettingstijl, waardoor een grote schaalbaarheidsefficiëntie wordt geïntroduceerd in modeltraining en flexibiliteit in inferentie tijdens implementatie. We introduceren het concept van Chain-of-Representation (CoR), dat de verborgen toestanden in elke laag formuleert als een combinatie van meerdere subrepresentaties (d.w.z. kettingen) op het niveau van de verborgen dimensie. In elke laag kan elke ketting van de uitvoerrepresentaties alleen al zijn voorgaande kettingen in de invoerrepresentaties zien. Hierdoor kan het model dat is gebouwd op het CoM-framework progressief opschalen door kettingen toe te voegen op basis van eerdere modellen (d.w.z. kettingen), en biedt het meerdere submodellen van verschillende groottes voor elastische inferentie door verschillende aantallen kettingen te gebruiken. Op basis van dit principe ontwikkelen we Chain-of-Language-Model (CoLM), dat het idee van CoM incorporeert in elke laag van de Transformer-architectuur. Op basis van CoLM introduceren we verder CoLM-Air door een KV-delingmechanisme te introduceren, dat alle sleutels en waarden binnen de eerste ketting berekent en vervolgens deelt over alle kettingen. Dit ontwerp toont aanvullende uitbreidingsmogelijkheden, zoals het mogelijk maken van naadloze LM-switching, prefilling-versnelling en meer. Experimentele resultaten tonen aan dat onze CoLM-familie vergelijkbare prestaties kan bereiken als de standaard Transformer, terwijl tegelijkertijd meer flexibiliteit wordt geboden, zoals progressief schalen om de trainings efficiëntie te verbeteren en meerdere modelgroottes te bieden voor elastische inferentie, wat een nieuwe weg opent naar het bouwen van taalmmodellen. Onze code zal in de toekomst worden vrijgegeven op: https://github.com/microsoft/CoLM.