آرشیو مقاله‌ها همه‌ی مقاله‌هاموضوع‌هالابراتوارهاجستجودرباره
همه‌ی مقاله‌هاموضوع‌هالابراتوارهاجستجودرباره
خانه مقاله‌ها مقاله‌ی 2610.02185

مقاله‌ی arXiv:2610.02185 · منتشرشده در کانال

Decoding Looped Transformers Better for (Almost) Free

رمزگشایی بهتر ترنسفورمرهای حلقه‌ای تقریباً رایگان با مقایسه‌ی حلقه‌های اولیه و نهایی

تاریخ
(۹ مهر ۱۴۰۵) 2026-10-01 تأییدشده از روی تاریخ ارسال نسخه‌ی اول در آرکایو
نویسنده‌ها
Weihao Liu, Huangjie Zheng, Tianrong Chen, Rohit Dilip, Richard He Bai, Yizhu Jiao, Yuyang Wang, Ruixiang Zhang
نهاد نویسنده‌ها
Apple
برچسب لابراتوار
Apple

خلاصه‌ی فارسی

مسئله

ترنسفورمرهای حلقه‌ای با اجرای مکرر یک بلوک مشترک در چند حلقه کارایی پارامتری به دست می‌آورند. هر حلقه یک بازنمایی میانی قابل رمزگشایی برای همان توکن بعدی می‌دهد، اما رمزگشایی استاندارد حالت‌های قبلی را دور می‌ریزد. چون حلقه‌های اولیه محاسبه‌ی کمتری دارند، این ساختار خود به خود جفت پیش‌بینی ضعیف و قوی هم‌تراز می‌سازد.

روش

چارچوب LoopCD یک رمزگشایی مقایسه‌ای بدون آموزش است که انتخاب توکن را با مقایسه‌ی پیش‌بینی نهایی با یک گذر حلقه‌ای قبلی هدایت می‌کند، یا در فضای لاجیت با یک گذر خروجی اضافه یا در فضای حالت پنهان بدون هیچ سربار خروجی.

یافته‌ها

  • در چهار خانواده‌ی ترنسفورمر حلقه‌ای بهبودهای قابل توجه و پیوسته داد: نسخه‌ی لاجیت pass@۱ مدل Ouro-2.6B-Thinking روی AIME ۲۰۲۴ را از ۶۱٫۸۸ به ۷۳٫۳۳ درصد رساند.
  • نسخه‌ی حالت پنهان pass@۱ مدل Huginn روی HumanEval را از ۲۲٫۵۶ به ۳۱٫۷۱ درصد رساند.
  • با نصف کردن تعداد حلقه‌ها همچنان با پایه‌ی کامل بدون هدایت برابر یا بهتر بود و محاسبات پیش‌رو را ۲۲٫۵ تا ۴۸٫۲ درصد کم کرد.

محدودیت‌ها

مقاله بخش جداگانه‌ای برای محدودیت‌ها ندارد؛ روش فقط برای ترنسفورمرهای حلقه‌ای کاربرد دارد.

چرا مهم است

ترنسفورمرهای حلقه‌ای راهی برای مدل‌های کوچک‌تر با استدلال عمیق‌ترند. این کار نشان می‌دهد اطلاعاتی که در حلقه‌های میانی دور ریخته می‌شود، می‌تواند هم دقت را بالا ببرد و هم محاسبه را کم کند.