رفتن به محتوای اصلی
کوهستان
توضیح کوهستانتفسیرپذیری مکانیکیمعماری ترنسفورمر· 8 دقیقه مطالعه· در هوش مصنوعی

مکانیسم یادگیری درجا: مدارهای القایی و ترکیب کلید در ترنسفورمرها چگونه کار می‌کنند؟

پژوهشگران تفسیرپذیری سازوکاری، سازوکار دقیق توجه دولایه‌ای را شناسایی کرده‌اند که به مدل‌های زبانی امکان می‌دهد الگوها را تشخیص داده و کپی کنند. این مدارها با انتقال داده‌ها از مسیر باقیمانده، به مدل اجازه می‌دهند بدون به‌روزرسانی وزن‌های پایه، خود را با دستورهای ورودی تازه وفق دهد.

به قلم آزاده ابراهیمی

به‌طور خلاصه

  1. مدارهای القایی به مدل‌های زبانی امکان می‌دهند تا با تشخیص و کپی مستقیم توالی شناسه‌ها از ورودی، فرایند یادگیری درجا را اجرا کنند.
  2. این سازوکار متکی بر ترکیب کلید است؛ جایی که سر توجه در لایه دوم از داده‌های نوشته‌شده توسط سر لایه اول برای یافتن الگوهای گذشته بهره می‌برد.
  3. این فرایند تماماً درون جریان باقیمانده رخ می‌دهد و به مدل امکان می‌دهد بدون تغییر در وزن‌های منجمد خود، با دستورهای تازه هماهنگ شود.

بزرگ‌ترین محدودیت بنیادین در هر مدل زبانی بزرگ مستقر، انجماد حافظه آن است. پس از پایان فرایند آموزش، میلیاردها وزن عصبی که هویت مدل را شکل می‌دهند دیگر تغییر نمی‌کنند؛ یعنی مدل نمی‌تواند با دستکاری در پارامترهای خود حقایق تازه‌ای بیاموزد. معماری در زمان استنتاج کاملاً ایستا و دست‌نخورده باقی می‌ماند.

با این حال، کافی است کاربری کلمه‌ای کاملاً ساختگی یا ساختار نگارشی نوظهوری را در ورودی تایپ کند؛ مدل بلافاصله آن را فرا می‌گیرد. این سازگاری شناور در قالب ورودی بدون کوچک‌ترین تغییر در پارامترها رخ می‌دهد؛ پدیده‌ای که «یادگیری درجا» یا درون‌زمینه نامیده می‌شود و امکان اجرای وظایفی را فراهم می‌کند که شبکه هرگز صراحتاً برای آن‌ها آموزش ندیده بود.

سال‌ها چگونگی یادگیری لحظه‌ای در یک ماتریس عددی ایستا برای مهندسان معمایی پوشیده بود. اکنون پژوهشگران حوزه «تفسیرپذیری سازوکاری» موفق شده‌اند موتور ریاضی دقیق پشت این پدیده را شناسایی و ایزوله کنند: سازوکاری دولایه‌ای به نام «مدار القایی». این یافته تبیینی عینی و ساختاری برای یکی از پیچیده‌ترین رفتارهای برآمده در مدل‌ها ارائه می‌دهد.[1][3]

با ردیابی جریان داده‌ها درون این مدارها، مشخص شده است که مدل‌ها با فرایندی موسوم به «ترکیب کلید» مسیرهای توجه خود را به‌طور موقت سیم‌کشی مجدد می‌کنند. این سازوکار به مدل اجازه می‌دهد توالی شناسه‌ها را مستقیماً درون متن ورودی رصد و کپی کند و عملاً یک حافظه کاری موقت پدید آورد.[2]

کالبدشکافی جریان باقیمانده

برای درک کارکرد مدارهای القایی، ابتدا باید به شریان اصلی مدل ترنسفورمر، یعنی «جریان باقیمانده» نگاهی بیندازیم. این شریان فضایی برداری با ابعاد بالاست که در آن هر شناسه در متن ورودی، وضعیتی جاری و پویا برای خود حفظ می‌کند که با عبور از شبکه مدام بازنویسی می‌شود.[2]

مکانیسم دومرحله‌ای مدار القایی متکی بر ترکیب کلید میان لایه‌هاست.

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

مقاله بنیادین سال ۲۰۲۱ اِلهِیج و همکاران درباره مدارهای ترنسفورمر توضیح می‌دهد: «تمامی بخش‌های یک ترنسفورمر از طریق خواندن و نوشتن در زیرفضاهای مختلف جریان باقیمانده با یکدیگر تعامل دارند.» این فرایند نوشتن جمعی، اساس پایه‌گذاری رفتارهای پیچیده در چند لایه مختلف است.[2]

سرهای توجه این جریان را از دو مسیر موازی هدایت می‌کنند: مدار «پرسش-کلید» با محاسبه امتیازهای توجه تعیین می‌کند سر توجه به کجا بنگرد، و مدار «خروجی-مقدار» مشخص می‌کند چه داده مشخصی از شناسه هدف کپی شده و به جریان باقیمانده شناسه جاری بازگردد.[2]

گام اول: سر شناسه پیشین

مدار القایی برای فعال شدن دست‌کم به دو لایه توجه نیاز دارد؛ به همین دلیل مدل‌های تک‌لایه از نظر ساختاری امکان یادگیری درجا را ندارند. این سازوکار در لایه اول توسط بخشی تخصصی موسوم به «سر شناسه پیشین» کلید می‌خورد که وظیفه‌ای معین و کپی‌محور دارد.[1][2]

هنگامی که سر شناسه پیشین به یک توالی می‌نگرد، مدار پرسش-کلید آن طوری برنامه‌ریزی شده که منحصراً به دقیقاً یک جایگاه عقب‌تر توجه کند. اگر متن حاوی توالی «گربه نشست» باشد، سر مستقر در جایگاه «نشست» فقط به شناسه پیشین یعنی «گربه» می‌نگرد و سایر بافت را نادیده می‌گیرد.[1]

به محض تمرکز روی شناسه پیشین، مدار خروجی-مقدار هویت آن شناسه را کپی کرده و در جریان باقیمانده شناسه فعلی می‌نویسد. پس از اجرای این لایه، شناسه «نشست» برچسبی ریاضی حمل می‌کند که نشان می‌دهد شناسه قبلی آن «گربه» بوده است؛ امری که بازنمایی آن را بنیادین دگرگون می‌کند.[1][2]

توانایی یادگیری درجا در قالب یک گذار فاز ناگهانی هم‌زمان با تشکیل مدارهای القایی پدیدار می‌شود.

این اقدام داده‌های عبوری از شبکه را غنی می‌سازد. مدل دیگر کلمات را مجزا نمی‌بیند؛ هر جایگاه اکنون دربرگیرنده برهم‌نهی از خود و کلمه ماقبل خویش است و زمینه را برای تطبیق الگوهای پیچیده‌تر در مرحله بعد فراهم می‌آورد.[4]

گام دوم: عملکرد ترکیب کلید در عمل

جادوی یادگیری درجا در لایه دوم رخ می‌دهد؛ جایی که «سر القایی» واقعی قرار دارد. این سر با تکیه بر اطلاعات تقویت‌شده از لایه اول، عملیات انطباق الگو موسوم به «ترکیب کلید» را اجرا می‌کند و دو لایه را به مداری واحد و منسجم پیوند می‌زند.[1][3]

ترکیب کلید زمانی رخ می‌دهد که سر موجود در لایه دوم، از خروجی نوشته‌شده توسط سر لایه اول برای ساخت ماتریس کلید خود استفاده کند. به‌جای وارسی شناسه خام، ماتریس‌های کلید سر القایی اکنون حاوی اطلاعات کلمه پیشین نیز هستند و مشخصات اعلامی هر موقعیت را دگرگون می‌کنند.[2]

تصور کنید در ادامه متن واژه «گربه» دوباره تکرار شود و مدل باید کلمه بعدی را پیش‌بینی کند. پرسشِ سر القایی در این جایگاه جدید «گربه»، متن را به سمت عقب جستجو می‌کند تا شناسه‌ای را بیابد که ماتریس کلید آن برچسب «شناسه پیشین = گربه» داشته باشد.[1]

از آنجا که سر شناسه پیشین پیش‌تر این برچسب را در جریان باقیمانده شناسه «نشست» نوشته بود، سر القایی به یک تطابق کامل ریاضی می‌رسد. مدار پرسش-کلید مستقیماً روی شناسه «نشست» قفل می‌شود و باقی متن را کنار می‌گذارد.[1][4]

بازنویسی و کپی کردن آینده

پس از تمرکز توجه بر شناسه پیشینِ درست، سر القایی حرکت پایانی را اجرا می‌کند. مدار خروجی-مقدار، هویت شناسه هدف یعنی «نشست» را کپی کرده و آن را در جریان باقیمانده موقعیت کنونی می‌نویسد تا شبکه برای تولید پاسخ مهیا شود.[1]

یادگیری درجا کاملاً درون جریان باقیمانده انجام می‌شود و به صفر تغییر در پارامترها نیاز دارد.

سپس مدل این جریان باقیمانده به‌روزشده را به خروجی نهایی بدل می‌کند و کلمه «نشست» را پیش‌بینی می‌نماید. مدل بدون تغییر حتی یک وزن از میلیاردها پارامتر ساکن خود، توانسته الگو را بشناسد، ادامه قبلی آن را بازیابی کند و بنویسد.[1][3]

مطالعه سال ۲۰۲۲ انتروپیک به رهبری اولسون که این سازوکار را فرمول‌بندی کرد، بیان می‌دارد: «سرهای القایی زمینه را برای یافتن نمونه‌های قبلی شناسه کنونی جستجو می‌کنند، به شناسه‌ای توجه نشان می‌دهند که در صورت تکرار الگو ظاهر می‌شود، و احتمال وقوع آن را افزایش می‌دهند.»[1]

این موتور چرایی مهارت بالای مدل‌ها را در روش پرامپت‌نویسی با چند نمونه آشکار می‌کند. وقتی کاربر سه نمونه ترجمه انگلیسی به فرانسوی ارائه می‌دهد، مدارهای القایی قالب ساختاری را تشخیص داده و الگوی متناظر را کپی می‌کنند و کار به‌آسانی انجام می‌پذیرد.[1][4]

تغییر فاز ناگهانی حین آموزش

کشف مدارهای القایی معمایی دیرینه درباره فرایند یادگیری مدل‌ها حین آموزش اولیه را نیز حل کرد. پژوهشگران متوجه شدند قابلیت یادگیری درجا برخلاف انتظار سنتی از شبکه‌های عصبی، آرام‌آرام در جریان میلیون‌ها گام آموزشی شکل نمی‌گیرد.[1]

بلکه مدل‌ها در اوایل دوره آموزش خود دستخوش یک «تغییر فاز» ناگهانی می‌شوند. طی بازه بسیار باریکی از گام‌های بهینه‌سازی، توان یادگیری از بستر ورودی جهشی تند پیدا می‌کند؛ امری که با افتی محسوس در خطای پردازش توالی‌های تکراری نمایان می‌شود.[1][3]

با ردیابی وزن‌های درونی مدل، محققان ثابت کردند این گذار فاز دقیقاً زمانی رخ می‌دهد که سرهای شناسه پیشین و سرهای القایی با هم جفت شده و مدارهای ترکیب کلید را پدید می‌آورند. قابلیت نوظهور درست در لحظه پیوند این دو لایه متولد می‌شود.[1][2]

پژوهشگران برای شناسایی سرهایی که مدارهای القایی را می‌سازند، شدت ترکیب کلید را ارزیابی می‌کنند.

در مدل‌هایی که فقط یک لایه توجه دارند، این گذار فاز هرگز ثبت نمی‌شود. چون ترکیب کلید نیازمند خواندن خروجی یک سر توسط سر دیگر است، معماری تک‌لایه توان تشکیل مدار القایی را ندارد و سقف توان استدلال آن هر چقدر هم آموزش ببیند محدود خواهد ماند.[1][2]

سنجش قدرت و اتصال مدارها

برای اثبات این‌که ترکیب کلید گرداننده اصلی این رفتار است، پژوهشگران به مشاهده صرف بسنده نمی‌کنند، بلکه دست به مداخله می‌زنند؛ آنان با حذف یا فرسایش ریاضی سرهای القایی در زمان استنتاج، افت توان مدل را اندازه می‌گیرند.[1]

وقتی سرهای القایی از یک مدل کوچک صرفاً متکی به توجه حذف می‌شوند، قابلیت یادگیری درجا تقریباً به‌کلی محو می‌گردد. مدل به یک پیش‌بینی‌کننده آماری ساده تنزل پیدا می‌کند که از فهم دستورهای جدید یا کشف الگوهای تکراری در متن ناتوان است.[1][3]

علاوه بر این، پژوهشگران شدت دقیق ترکیب کلید میان دو سر توجه را اندازه‌گیری می‌کنند. آنها نرم فروبنیوسِ حاصل‌ضرب ماتریسی مدار خروجی-مقدار سر اول و مدار پرسش-کلید سر دوم را محاسبه می‌نمایند که نشان‌دهنده مستقیم حجم تبادل اطلاعات است.[2]

امتیاز بالای ترکیب با دقت نشان می‌دهد کدام سرها در لایه‌ها با یکدیگر همکاری دارند. این دقت ریاضی به مهندسان اجازه می‌دهد دیاگرام محاسباتی ترنسفورمر را درست مانند نقشه سیم‌کشی یک برد الکترونیکی ترسیم کنند و هوش مصنوعی را از کیمیاگری به مهندسی ارتقا دهند.[2][4]

عبور از بن‌بست وزن‌های منجمد

شناخت ترکیب کلید نگاه پژوهشگران را به هوش مصنوعی دگرگون می‌کند. وزن‌های عصبی تنها پایگاهی از محفوظات نیستند، بلکه مجموعه‌ای از الگوریتم‌هایند که برای هدایت شناور جریان باقیمانده طراحی شده‌اند و برنامه‌های پویایی را در لحظه اجرا می‌کنند.[4]

تصویرسازی: جریان باقیمانده به مثابه حافظه موقت مشترک میان تمامی سرهای توجه در سراسر لایه‌ها عمل می‌کند.

با واگذاری وظیفه تطبیق الگو به بستر ورودی، مدل از تنگنای پارامترهای منجمد فراتر می‌رود. مدار القایی مثل یک حافظه رم موقت عمل می‌کند و بدون نیاز به ثبت داده در هارددیسک دائمیِ وزن‌ها، متغیرهای گذرای شناور را ذخیره و بازیابی می‌نماید.[1][4]

این شفافیت سازوکاری برای ایمنی و هم‌راستاسازی هوش مصنوعی حیاتی است. اگر مهندسان بدانند کدام سرهای توجه مسئول کپی کردن محتوا هستند، می‌توانند با پایش مستقیم همان مدارها، از بازتولید داده‌های محرمانه یا سمی پیش از خروج جلوگیری کنند.[3][4]

با رشد مدل‌ها تا تریلیون‌ها پارامتر، این مدارهای القایی پایه به ساختارهای پیچیده‌تری بدل می‌شوند که عهده‌دار استدلال‌های انتزاعی‌اند. با این همه، ریاضیات پایه تغییری نمی‌کند: لایه‌ای گذشته را ثبت می‌کند و لایه بعد با ترکیب کلید، آینده را بازمی‌نویسد.[1][2]

این تحلیل چگونه انجام شد

روش
ما مسیر ضرب ماتریسی را در دو لایه توجه بررسی کردیم تا نحوه تعامل مدارهای پرسش-کلید و خروجی-مقدار را حین ترکیب کلید مجزا کرده و وضعیت جریان باقیمانده را پیش و پس از ثبت داده توسط سر شناسه پیشین بسنجیم.
یافته
ترنسفورمر با پیوند مستقیم خروجی لایه نخست به محاسبات کلید در لایه دوم، یک جدول ارجاع پویا و وابسته به متن درون جریان باقیمانده پدید می‌آورد که نیاز به به‌روزرسانی وزن‌های ایستا را برای یادگیری ارتباط‌های جدید در زمان استنتاج برطرف می‌کند.
داده‌هایی که بر پایهٔ آن‌ها کار کردیم
  • خروجی سر شناسه پیشین لایه ۱ (مدار خروجی-مقدار): Residual stream enrichment — Transformer Circuits Thread
  • ماتریس کلید سر القایی لایه ۲ (مدار پرسش-کلید): Predecessor matching — arXiv
محدودیت‌های این تحلیل
این تحلیل بر پایه مدل‌های ساده‌شده و صرفاً مبتنی بر توجه است؛ در مدل‌های پیشرو با صدها لایه و بلوک‌های چندلایه پرسپترون، سرهای القایی با سازوکارهای دیگری تعامل دارند که هنوز نقشه‌برداری کاملی از آنها انجام نشده است.

اصطلاحات کلیدی

مدار القایی (Induction Circuit)
سازوکاری دولایه‌ای در مدل‌های ترنسفورمر که الگوهای تکرارشونده را در متن شناسایی کرده و توالی‌های بعدی را برای پیش‌بینی کلمه آینده کپی می‌کند.
ترکیب کلید (K-composition)
عملیاتی ریاضی که در آن سر توجه در لایه بالاتر از خروجی سر لایه پایین‌تر برای تشکیل ماتریس کلید خود استفاده می‌کند تا بر مبنای محتوای قبلی جستجو کند.
جریان باقیمانده (Residual Stream)
مسیر اصلی انتقال داده در مدل ترنسفورمر که هر شناسه در آن وضعیتی پویا دارد و سرهای توجه داده‌ها را در آن می‌خوانند یا می‌نویسند.
یادگیری درجا (In-context Learning)
توانایی مدل زبانی در انطباق با وظایف و الگوهای نو در متن ورودی، بدون اینکه وزن‌های پایه‌ای و منجمد آن تغییری پیدا کنند.
سر توجه (Attention Head)
بخشی از شبکه عصبی که وزن ارتباط و اهمیت بخش‌های مختلف توالی ورودی را نسبت به شناسه جاری محاسبه می‌کند.

پرسش‌های متداول

آیا ترنسفورمر تک‌لایه قادر به یادگیری درجا است؟

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

آیا مدارهای القایی وزن‌های مدل را تغییر می‌دهند؟

خیر. وزن‌های مدل حین استنتاج کاملاً منجمد و دست‌نخورده باقی می‌مانند. مدارهای القایی منحصراً با دستکاری داده‌های شناور در جریان باقیمانده فعالیت می‌کنند و بدون تغییر دائمی، امکان انطباق را به وجود می‌آورند.

پژوهشگران چگونه سرهای القایی را شناسایی می‌کنند؟

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

بررسی عمیق دیدگاه‌ها

پژوهشگران تفسیرپذیری سازوکاری

تمرکز بر مهندسی معکوس شبکه‌های عصبی به الگوریتم‌های شفاف و قابل درک.

این دیدگاه تأکید دارد که نباید به مدل‌های زبانی به چشم جعبه‌های سیاه نگریست. به باور این پژوهشگران، با تجزیه معماری مدل به اجزای گسسته و منطقی مانند مدارهای القایی، می‌توان عملکرد مدل را از منظر ریاضی به اثبات رساند. اولویت آنها یافتن دقیق ماتریس‌ها و اوزان مسئول رفتارهای مشخص است؛ چرا که ایمنی پایدار در هوش مصنوعی بدون شفافیت در سطح مدارها دست‌نیافتنی خواهد بود.

پژوهشگران قابلیت‌های مدل

نگاه به مدارهای القایی به عنوان سنگ‌بنای توسعه توان استدلال پیشرفته.

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

حامیان ایمنی هوش مصنوعی

اهمیت فهم ساختار مدارها برای جلوگیری از خروجی‌های مخرب یا نامطلوب.

حامیان ایمنی بر سطح مهارپذیری حاصل از تفسیرپذیری تکیه دارند. اگر مهندسان بدانند دقیقاً کدام سرهای توجه داده‌ها را کپی می‌کنند، قادر خواهند بود با نظارت بر همان مدارها یا غیرفعال کردن موقتشان، مانع از بازتولید اطلاعات محرمانه یا لحن زننده شوند. از این دیدگاه، آزمون‌های رفتار بیرونی کافی نیست و مهار حقیقی نیازمند اقدام مستقیم در سطح مدارهای محاسباتی مدل است.

پژوهشگران تفسیرپذیری سازوکاری 40%پژوهشگران قابلیت‌های مدل 30%حامیان ایمنی هوش مصنوعی 30%
پژوهشگران تفسیرپذیری سازوکاری
بر مهندسی معکوس شبکه‌های عصبی به الگوریتم‌های ملموس برای تضمین ایمنی تمرکز دارند.
پژوهشگران قابلیت‌های مدل
مدارهای القایی را سازه‌های بنیادین برای توسعه استدلال‌های پیشرفته‌تر می‌دانند.
حامیان ایمنی هوش مصنوعی
درک ساختار در سطح مدارها را شرط اساسی پیشگیری از رفتارهای خطرناک مدل می‌شمارند.

دیدگاه‌هایی که این گزارش پوشش نداده

  • مهندسان سخت‌افزار
  • توسعه‌دهندگان تجاری هوش مصنوعی

منابع

پوشش منابع

4 منبع

3 دیدگاه شناسایی‌شده

پژوهشگران تفسیرپذیری سازوکاری 40%پژوهشگران قابلیت‌های مدل 30%حامیان ایمنی هوش مصنوعی 30%
  1. [1]arXivپژوهشگران قابلیت‌های مدل

    In-context Learning and Induction Heads

    مطالعه در arXiv →
  2. [2]Transformer Circuits Threadپژوهشگران تفسیرپذیری سازوکاری

    A Mathematical Framework for Transformer Circuits

    مطالعه در Transformer Circuits Thread →
  3. [3]OpenReviewحامیان ایمنی هوش مصنوعی

    Induction Heads in Transformers

    مطالعه در OpenReview →
  4. [4]تیم سردبیری کوهستانپژوهشگران تفسیرپذیری سازوکاری

    تحلیل تیم سردبیری کوهستان

    مطالعه در تیم سردبیری کوهستان →

نظرات

همیشه در جریان باشید

هر زاویه. هر روز.

اخبار هوش مصنوعی با پوشش کامل منابع و تحلیل دیدگاه‌ها، هر روز و رایگان.