בעידן שבו מודלי שפה גדולים (LLMs) זקוקים להיגיון מתקדם כדי להתמודד עם משימות מורכבות, למידה מחוזקת (RL) לאחר האימון הראשוני הוכיחה יכולת מרשימה להפעיל התנהגויות חשיבה. אולם, לצורך יעילות חישובית, תהליך זה משתמש בדרך כלל בגלגולים שנוצרו במחוץ-מדיניות – כלומר, בעזרת מדיניות דגימה ישנה יותר – כדי לעדכן את מדיניות היעד הנוכחית. הבעיה: הפער בין מדיניות הדגימה למדיניות היעד גורם ליציבות בעייתית, במיוחד כאשר הפער גדול. לפי המאמר החדש ב-arXiv, תיקון ברמת הטוקן, הנפוץ בשל פשטותו, מוביל לדינמיקות אימון לא יציבות.
החוקרים חוזרים לבסיס התיאורטי ומציגים כי התיקון הנכון והמדויק הוא יחס חשיבות מקדים (Prefix Importance Ratio) – יחס מצטבר לאורך הפרפיקס של הרצף. קירוב ברמת הטוקן, כפי שמשמשים ביעדים קיימים, גורם לחוסר יציבות כאשר הסטייה מחוץ-מדיניות גדולה. המאמר מדגים זאת באופן תיאורטי ומביא ראיות ניסוייות. בשיטות מסורתיות, יחסי החשיבות המצטברים עלולים להתפוצץ או להתכווץ, מה שפוגע בהתכנסות האימון.
כדי לפתור זאת, החוקרים מציעים יעד פשוט ואפקטיבי בשם Minimum Prefix Ratio (MinPRO). במקום יחס מקדים מצטבר לא יציב, MinPRO משתמש בתחליף לא-מצטבר: היחס המינימלי ברמת הטוקן שנצפה בפרפיקס הקודם. גישה זו מבטיחה יציבות נומרית ומחשבית, מבלי לוותר על הדיוק התיאורטי. ניסויים נרחבים על מודלים צפופים ותערובת-מומחים (MoE), במבחני היגיון מתמטי מרובים, מראים שיפור משמעותי ביציבות האימון ובביצועים השיא במצבי סטייה גדולים.
המשמעות של MinPRO חשובה במיוחד לפיתוח מודלי שפה גדולים, שבהם יעילות האימון היא גורם קריטי. שיטות קודמות סבלו מקריסות אימון כאשר השתמשו בגלגולים ישנים יותר להאצת התהליך. כעת, עם MinPRO, אפשר להגדיל את ה-off-policyness מבלי לסכן את היציבות, מה שמאפשר אימונים ארוכים יותר וחזקים יותר. בהקשר ישראלי, חברות כמו Mobileye או startups ב-AI עשויות להרוויח מכך בפיתוח מודלים מקומיים.
לסיכום, יחס חשיבות מקדים ו-MinPRO מסמנים צעד חשוב קדימה – או למעשה צעד אחורה לתיאוריה הנכונה – ביציבות אופטימיזציית מדיניות ב-LLMs. מנהלי טכנולוגיה צריכים לשים לב: האם השיטה הזו תשנה את אופן אימון הדור הבא של מודלי AI? קראו את המאמר המלא ב-arXiv כדי להעריך את ההשלכות העסקיות.