יום שלישי, 15 בספטמבר 2026 LIVE
AI־INFO

כתבה arXiv cs.LG ·

אימון מקוון לפיענוח ספקולטיבי

Online Draft Co-Training for Speculative Decoding in Large-Scale, Long-Context RL Post-Training
חוקרים פיתחו שיטה לאימון מקוון של דגמים גדולים עם הקשר ארוך. השיטה משפרת את מהירות האימון ומקטינה את צריכת הזיכרון. הקוד זמין בגיטהאב.
תקציר מקורי באנגליתarXiv:2609.07108v1 Announce Type: new Abstract: Speculative decoding accelerates rollout generation, which dominates the cost of reinforcement learning (RL) post-training. Online co-training can further increase the draft's accuracy, yielding greater speedups. However, scaling this approach to co-training on large models with long contexts poses two obstacles: (1) branch attention is unsupported by standard causal context-parallel (CP) implementations, and (2) target features span across pipeline-parallel (PP) stages. We address both with an end-to-end system for large-scale online draft co-training. For CP, we extend packed, load-balanced zigzag ring attention by merging rank-local branch attention with causal main-sequence attention. For PP, TapChannel transports intermediate target feat
קרא במקור המקורי