Limitations of Normalization in Attention Mechanism
Mudarisov, Burtsev, Petrova et al.
This paper investigates the limitations of the normalization in attention mechanisms. We begin with a theoretical framework that enables the identification of the model's selective ability and the geometric separation involved in token selection. Our analysis includes explicit bounds on distances and separation criteria for token vectors under softmax scaling. Through experiments with pre-trained GPT-2 model, we empirically validate our theoretical results and analyze key behaviors of the attention mechanism. Notably, we demonstrate that as the number of selected tokens increases, the model's ability to distinguish informative tokens declines, often converging toward a uniform selection pattern. We also show that gradient sensitivity under softmax normalization presents challenges during training, especially at low temperature settings. These findings advance current understanding of softmax-based attention mechanism and motivate the need for more robust normalization and selection strategies in future attention architectures.
এই পেপারটি মনোযোগ প্রক্রিয়ায় নর্মালাইজেশন পদ্ধতির তাত্ত্বিক সীমাবদ্ধতা গভীরভাবে অধ্যয়ন করে। লেখকরা মডেলের নির্বাচন ক্ষমতা এবং টোকেন নির্বাচনে জড়িত জ্যামিতিক বিচ্ছেদ চিহ্নিত করার জন্য একটি তাত্ত্বিক কাঠামো প্রতিষ্ঠা করেছেন। বিশ্লেষণে সফটম্যাক্স স্কেলিং এর অধীনে টোকেন ভেক্টর দূরত্ব এবং বিচ্ছেদ মানদণ্ডের স্পষ্ট সীমানা অন্তর্ভুক্ত রয়েছে। প্রশিক্ষিত GPT-2 মডেলে পরীক্ষা-নিরীক্ষার মাধ্যমে, লেখকরা তাত্ত্বিক ফলাফল অভিজ্ঞতামূলকভাবে যাচাই করেছেন এবং মনোযোগ প্রক্রিয়ার মূল আচরণ বিশ্লেষণ করেছেন। গবেষণা দেখায় যে নির্বাচিত টোকেনের সংখ্যা বৃদ্ধির সাথে সাথে, মডেল তথ্যপূর্ণ টোকেন আলাদা করার ক্ষমতা হ্রাস পায় এবং প্রায়শই সমান নির্বাচন প্যাটার্নে রূপান্তরিত হয়। গবেষণা আরও দেখায় যে সফটম্যাক্স নর্মালাইজেশনের অধীনে গ্রেডিয়েন্ট সংবেদনশীলতা প্রশিক্ষণে চ্যালেঞ্জ নিয়ে আসে, বিশেষত কম তাপমাত্রার সেটিংসে।
এই গবেষণা যে মূল সমস্যাটি সমাধান করতে চায় তা হল মনোযোগ প্রক্রিয়ায় সফটম্যাক্স নর্মালাইজেশনের অন্তর্নিহিত সীমাবদ্ধতা, বিশেষত "অদৃশ্য মনোযোগ" (vanishing attention) ঘটনা। প্রসঙ্গ দৈর্ঘ্য L বৃদ্ধির সাথে সাথে, মনোযোগের ওজন 1/L এর দিকে প্রবণ হয়, যা মডেলকে তথ্যপূর্ণ এবং অ-তথ্যপূর্ণ টোকেন কার্যকরভাবে আলাদা করতে অক্ষম করে তোলে।
লেখকরা মনোযোগ প্রক্রিয়াকে ক্ষমতা-সীমাবদ্ধ পুনরুদ্ধারকারী (capacity-limited retriever) হিসাবে পুনর্নির্ধারণ করেছেন, প্রথম নীতি থেকে নর্মালাইজেশনের অন্তর্নিহিত সীমাবদ্ধতা বিশ্লেষণ করেছেন এবং আরও শক্তিশালী মনোযোগ স্থাপত্য ডিজাইনের জন্য তাত্ত্বিক নির্দেশনা প্রদান করেছেন।
দূরত্ব সীমানা তত্ত্ব: নির্বাচিত এবং অ-নির্বাচিত টোকেনের মধ্যে প্রতিনিধিত্ব দূরত্বের অ-অ্যাসিম্পটোটিক উপরের সীমানা প্রকাশ করা (উপপাদ্য 1), প্রমাণ করে যে যখন শীর্ষ-N সেট L এর সাথে সমানুপাতিকভাবে বৃদ্ধি পায়, দূরত্ব অনিবার্যভাবে ভেঙে পড়ে, "সফটম্যাক্স বোতলজনক" আনুষ্ঠানিক করে
জ্যামিতিক বিচ্ছেদ সীমানা: মৃদু গোলকীয় অনুমানের অধীনে, প্রমাণ করে যে একটি একক মনোযোগ মাথা একযোগে প্রায় 80% শীর্ষ-N টোকেন আলাদা করতে পারে (উপপাদ্য 2), একক-মাথা প্রতিনিধিত্ব ক্ষমতার কঠোর সীমা পরিমাপ করে
গ্রেডিয়েন্ট সংবেদনশীলতা বিশ্লেষণ: সাধারণ নর্মালাইজারের জ্যাকোবিয়ান নর্ম সীমাবদ্ধ করা (লেম্মা 2), সফটম্যাক্সে বিশেষায়িত করা যখন ক্লাসিক 1/(4T) অস্থিতিশীলতা পুনরুদ্ধার করা, আক্রমণাত্মক তাপমাত্রা স্কেলিংয়ের অপ্টিমাইজেশন অসুবিধা ব্যাখ্যা করা
অভিজ্ঞতামূলক যাচাইকরণ: GPT-2 এ পরীক্ষা-নিরীক্ষা তিনটি পূর্বাভাস সবই নিশ্চিত করে: দূরত্ব ভেঙে পড়া, বিচ্ছেদযোগ্যতা সম্পৃক্ততা এবং 1/T গ্রেডিয়েন্ট বৃদ্ধি
দৈর্ঘ্য L এর টোকেন এমবেডিং সিকোয়েন্স X = {xi}Li=1 দেওয়া, যেখানে xi ∈ Rd, টোকেন নির্বাচন এবং বিচ্ছেদে বিভিন্ন নর্মালাইজেশন পদ্ধতির তাত্ত্বিক সীমাবদ্ধতা বিশ্লেষণ করা।
6% নিয়ম: শুধুমাত্র প্রায় 6% টোকেন নির্বাচিত হওয়ার প্রয়োজন, এই অনুপাত অতিক্রম করে অভিজ্ঞতামূলক বিতরণ এবং প্রত্যাশিত বিতরণ পরিসংখ্যানগতভাবে আলাদা হয়ে যায়
80% উপরের সীমা: একটি একক মনোযোগ মাথার জ্যামিতিক বিচ্ছেদ ক্ষমতা প্রায় 80% এর একটি কঠোর উপরের সীমা রয়েছে
বহু-মাথার প্রয়োজনীয়তা: তাত্ত্বিক ব্যাখ্যা করে কেন প্রসঙ্গের বিভিন্ন অংশ কভার করার জন্য একাধিক মনোযোগ মাথার প্রয়োজন
পেপারটি মনোযোগ প্রক্রিয়া, ট্রান্সফর্মার স্থাপত্য, দীর্ঘ সিকোয়েন্স প্রক্রিয়াকরণ ইত্যাদি ক্ষেত্রের মূল সাহিত্য উদ্ধৃত করে, যার মধ্যে রয়েছে:
ভাসওয়ানি এবং অন্যদের মূল ট্রান্সফর্মার পেপার
বিভিন্ন দীর্ঘ সিকোয়েন্স প্রক্রিয়াকরণ পদ্ধতি (স্পার্স ট্রান্সফর্মার, লংফর্মার ইত্যাদি)
বিকল্প নর্মালাইজেশন পদ্ধতি (স্পার্সম্যাক্স, স্কেলেবল-সফটম্যাক্স ইত্যাদি)
সম্পর্কিত তাত্ত্বিক বিশ্লেষণ কাজ (সফটম্যাক্স বোতলজনক ইত্যাদি)
সামগ্রিক মূল্যায়ন: এটি একটি উচ্চ মানের তাত্ত্বিক বিশ্লেষণ পেপার, যা মনোযোগ প্রক্রিয়ার নর্মালাইজেশনের জন্য প্রথমবারের মতো একটি সিস্টেমেটিক গাণিতিক কাঠামো প্রদান করে। তাত্ত্বিক ফলাফল কঠোর এবং ব্যবহারিক মূল্য রয়েছে, পরীক্ষা-নিরীক্ষা যাচাইকরণ পর্যাপ্ত। পেপারটি শুধুমাত্র বিদ্যমান পদ্ধতির সীমাবদ্ধতা ব্যাখ্যা করে না, বরং ভবিষ্যত উন্নতির জন্য স্পষ্ট দিকনির্দেশনাও প্রদান করে। ট্রান্সফর্মার স্থাপত্য বোঝা এবং উন্নত করার জন্য এটি গুরুত্বপূর্ণ।