כתבה
arXiv cs.LG ·
פיצול חישובי תחזית חינם
Almost Free State Prediction Separation
פיצול חישובי תחזית חינם. המחקר מציע שיטה לפיצול תחזית וחישובי קוד במודלי למידת מכונה, כדי לצמצם את זמן החישוב.
תקציר מקורי באנגליתarXiv:2609.03807v3 Announce Type: replace Abstract: State--prediction separation (SPS) relieves a language model's hidden state of two competing burdens---summarizing the context and predicting the next token---by splitting the forward pass into a state stream and a prediction stream. The separation works, but it is expensive: the prediction stream is a second pass over the whole backbone, costing $\sim$1.9$\times$ the pretraining FLOPs, and even more in terms of wall-clock time when using a flexible attention mask. This paper makes state--prediction separation almost free. We take the separation to its limit with a free pause token: a prediction stream that writes no keys or values at all and so rides the sequence's existing positions. It improves next-token prediction of a standard Trans
קרא במקור המקורי
arxiv.org
פתח כתבה מקורית