2025-11-13T04:07:09.837900

Optimal Quantization for Matrix Multiplication

Ordentlich, Polyanskiy

Recent work in machine learning community proposed multiple methods for performing lossy compression (quantization) of large matrices. This quantization is important for accelerating matrix multiplication (main component of large language models), which is often bottlenecked by the speed of loading these matrices from memory. Unlike classical vector quantization and rate-distortion theory, the goal of these new compression algorithms is to be able to approximate not the matrices themselves, but their matrix product. Specifically, given a pair of real matrices $A,B$ an encoder (compressor) is applied to each of them independently producing descriptions with $R$ bits per entry. These representations subsequently are used by the decoder to estimate matrix product $A^\top B$. In this work, we provide a non-asymptotic lower bound on the mean squared error of this approximation (as a function of rate $R$) for the case of matrices $A,B$ with iid Gaussian entries. Algorithmically, we construct a universal quantizer based on nested lattices with an explicit guarantee of approximation error for any (non-random) pair of matrices $A$, $B$ in terms of only Frobenius norms $\|\bar{A}\|_F, \|\bar{B}\|_F$ and $\|\bar{A}^\top \bar{B}\|_F$, where $\bar{A},\bar{B}$ are versions of $A,B$ with zero-centered columns, respectively. For iid Gaussian matrices our quantizer achieves the lower bound and is, thus, asymptotically optimal. A practical low-complexity version of our quantizer achieves performance quite close to optimal. In addition, we derive rate-distortion function for matrix multiplication of iid Gaussian matrices, which exhibits an interesting phase-transition at $R\approx 0.906$ bit/entry, showing necessity of Johnson-Lindestrauss dimensionality reduction (sketching) in the low-rate regime.

academic

ম্যাট্রিক্স গুণনের জন্য সর্বোত্তম কোয়ান্টাইজেশন

মৌলিক তথ্য

পেপার আইডি: 2410.13780
শিরোনাম: ম্যাট্রিক্স গুণনের জন্য সর্বোত্তম কোয়ান্টাইজেশন
লেখক: Or Ordentlich (Hebrew University of Jerusalem), Yury Polyanskiy (MIT)
শ্রেণীবিভাগ: cs.IT cs.AI cs.CL cs.LG math.IT
প্রকাশনার সময়: ২০২৪ সালের অক্টোবর (arXiv প্রাক-প্রিন্ট)
পেপার লিঙ্ক: https://arxiv.org/abs/2410.13780

সারসংক্ষেপ

এই পেপারটি বৃহৎ আকারের ম্যাট্রিক্স গুণনের কোয়ান্টাইজেশন সমস্যার গভীর অধ্যয়ন করে। ঐতিহ্যবাহী ভেক্টর কোয়ান্টাইজেশনের বিপরীতে, এই গবেষণার লক্ষ্য ম্যাট্রিক্স নিজেই নয়, বরং তাদের ম্যাট্রিক্স গুণফল অনুমান করা। দুটি বাস্তব ম্যাট্রিক্স A এবং B দেওয়া হলে, এনকোডার প্রতিটি ম্যাট্রিক্সকে স্বাধীনভাবে সংকুচিত করে, প্রতিটি প্রবেশ R বিট ব্যবহার করে বর্ণিত হয়, তারপর ডিকোডার এই সংকুচিত প্রতিনিধিত্ব ব্যবহার করে ম্যাট্রিক্স গুণফল A⊤B অনুমান করে। পেপারটি স্বাধীন সমবিতরণ গাউসিয়ান প্রবেশ সহ ম্যাট্রিক্সের জন্য আনুমানিক গড় বর্গ ত্রুটির অ-অসীম নিম্ন সীমা প্রদান করে, নেস্টেড ল্যাটিস-ভিত্তিক সর্বজনীন কোয়ান্টাইজার তৈরি করে এবং R≈0.906 বিট/প্রবেশে একটি আকর্ষণীয় পর্যায় রূপান্তর আবিষ্কার করে, যা নিম্ন কোড-রেট ক্ষেত্রে Johnson-Lindenstrauss মাত্রা হ্রাসের প্রয়োজনীয়তা নির্দেশ করে।

গবেষণা পটভূমি এবং প্রেরণা

সমস্যার সংজ্ঞা

গভীর স্নায়ু নেটওয়ার্ক এবং বৃহৎ ভাষা মডেলের উত্থানের সাথে, ম্যাট্রিক্স গুণন গণনার প্রধান বাধা হয়ে উঠেছে। আধুনিক কম্পিউটিং হার্ডওয়্যার প্রায়শই গণনা ক্ষমতার চেয়ে মেমরি ব্যান্ডউইথ দ্বারা সীমাবদ্ধ। অতএব, মেমরি স্থানান্তর হ্রাস করার জন্য ম্যাট্রিক্সগুলির ক্ষতিপূর্ণ সংকোচন একটি গুরুত্বপূর্ণ সমস্যা।

ব্যবহারিক চাহিদা

বৃহৎ ভাষা মডেলের জন্য, লেখকরা প্রয়োজনীয় কোয়ান্টাইজেশন হার অনুমান করেছেন:

উৎপাদন পর্যায়ে, CPU গণনা সম্পদ পর্যাপ্তভাবে ব্যবহার করার জন্য 1-3 বিট/প্রবেশের কোয়ান্টাইজেশন হার প্রয়োজন
প্রাক-পূরণ পর্যায়ে, দ্রুত GPU-তে চলমান ছোট LLM-এর জন্য প্রায় 11.7 বিট/প্রবেশের কোয়ান্টাইজেশন হার প্রয়োজন

বিদ্যমান পদ্ধতির সীমাবদ্ধতা

ক্লাসিক্যাল ভেক্টর কোয়ান্টাইজেশন: ম্যাট্রিক্স A এবং B কে সরাসরি স্বাধীনভাবে কোয়ান্টাইজ করা এবং তারপর কোয়ান্টাইজড ম্যাট্রিক্সের গুণফল গণনা করা O(n²) ত্রুটির দিকে পরিচালিত করে
স্কেচিং পদ্ধতি: যদিও নিরপেক্ষ অনুমান প্রদান করে, তবুও বৈচিত্র্য O(n²)
নির্ধারক কোয়ান্টাইজার: গোলকের উপর ভেক্টরগুলির জন্য Ω(n²) এর নিম্ন সীমা বিদ্যমান

মূল অবদান

তাত্ত্বিক নিম্ন সীমা: iid গাউসিয়ান প্রবেশ সহ ম্যাট্রিক্সের জন্য ম্যাট্রিক্স গুণন কোয়ান্টাইজেশনের অ-অসীম নিম্ন সীমা প্রদান করে
সর্বজনীন কোয়ান্টাইজার: নেস্টেড ল্যাটিস-ভিত্তিক সর্বজনীন কোয়ান্টাইজার তৈরি করে, যেকোনো ম্যাট্রিক্সের জন্য স্পষ্ট ত্রুটি গ্যারান্টি সহ
অসীম সর্বোত্তমতা: প্রস্তাবিত কোয়ান্টাইজার iid গাউসিয়ান ম্যাট্রিক্সের জন্য নিম্ন সীমা অর্জন করে, তাই অসীম সর্বোত্তম
পর্যায় রূপান্তর ঘটনা: R≈0.906 বিট/প্রবেশে পর্যায় রূপান্তর আবিষ্কার করে, নিম্ন কোড-রেটে মাত্রা হ্রাসের প্রয়োজনীয়তা প্রকাশ করে
ব্যবহারিক অ্যালগরিদম: সর্বোত্তমতার কাছাকাছি কর্মক্ষমতা সহ কম জটিলতার বাস্তবায়ন প্রদান করে

পদ্ধতির বিস্তারিত বিবরণ

কাজের সংজ্ঞা

ম্যাট্রিক্স A ∈ R^(n×a) এবং B ∈ R^(n×b) দেওয়া হলে, লক্ষ্য হল এনকোডার f₁: R^(n×a) → 2^(naR) এবং f₂: R^(n×b) → 2^(nbR) এবং ডিকোডার g ডিজাইন করা যাতে:

$E\|A^⊤B - g(f_1(A), f_2(B))\|_F^2$

ন্যূনতম করা হয়, যেখানে প্রতিটি ম্যাট্রিক্স প্রবেশ R বিট ব্যবহার করে বর্ণিত হয়।

মূল ফাংশন Γ(R)

পেপারটি মূল হার-বিকৃতি ফাংশন সংজ্ঞায়িত করে:

$\Gamma(R) = \begin{cases} 1 - \left(1 - (2 \cdot 2^{-2R^*} - 2^{-4R^*})\right) \frac{R}{R^*} & R < R^* \\ 2 \cdot 2^{-2R} - 2^{-4R} & R \geq R^* \end{cases}$

যেখানে R* ≈ 0.906 স্থির বিন্দু সমীকরণ R = ½log₂(1 + 4R ln 2) এর সমাধান।

নেস্টেড ল্যাটিস কোয়ান্টাইজেশন স্কিম

1. অভ্যন্তরীণ পণ্য কোয়ান্টাইজেশন (মৌলিক নির্মাণ ব্লক)

গোলকের উপর ρ-সম্পর্কিত ভেক্টর U, V এর জন্য, নেস্টেড ল্যাটিস Λc ⊂ Λf ব্যবহার করে কোয়ান্টাইজ করুন:

এনকোডিং প্রক্রিয়া:

U এবং V এ যথাক্রমে স্বাধীন ডিদার ভেক্টর Z₁, Z₂ যোগ করুন
সূক্ষ্ম ল্যাটিস Λf এ কোয়ান্টাইজ করুন
মোটা ল্যাটিস Λc এ কোসেট প্রতিনিধিত্ব আউটপুট করুন

ডিকোডিং প্রক্রিয়া:

কোসেট থেকে কোয়ান্টাইজড পয়েন্ট পুনরুদ্ধার করুন
ডিদার সরান
অভ্যন্তরীণ পণ্য অনুমান গণনা করুন

2. ম্যাট্রিক্স গুণন কোয়ান্টাইজেশন

প্রাক-প্রক্রিয়াকরণ পদক্ষেপ:

শূন্য-কেন্দ্রীকরণ: Ā = A - (1/n)1·1^⊤A, B̄ = B - (1/n)1·1^⊤B গণনা করুন
নর্ম কোয়ান্টাইজেশন: প্রতিটি কলামের গড় এবং নর্ম উচ্চ নির্ভুলতায় বর্ণনা করুন
র্যান্ডম রোটেশন: Ā এবং B̄ এ একই অর্থোগোনাল ম্যাট্রিক্স S প্রয়োগ করুন

কোয়ান্টাইজেশন পদক্ষেপ:

ঘোরানো প্রতিটি কলামে অভ্যন্তরীণ পণ্য কোয়ান্টাইজার প্রয়োগ করুন
সময় ভাগাভাগি প্যারামিটার κ এবং MMSE স্কেলিং প্যারামিটার α ব্যবহার করুন

প্রযুক্তিগত উদ্ভাবনী পয়েন্ট

ডিদার প্রযুক্তি: কোয়ান্টাইজেশন ত্রুটি ইনপুট থেকে স্বাধীন করে, নির্ধারক কোয়ান্টাইজারের O(n²) ত্রুটি এড়ায়
নেস্টেড ল্যাটিস কাঠামো: সীমিত কোড-রেট প্রদান করার সময় ভাল কোয়ান্টাইজেশন কর্মক্ষমতা বজায় রাখে
সময় ভাগাভাগি: নিম্ন কোড-রেটে মাত্রা হ্রাসের মাধ্যমে সর্বোত্তম কর্মক্ষমতা অর্জন করে
র্যান্ডম রোটেশন: যেকোনো ভেক্টরকে গোলকে সমানভাবে বিতরণ করা রূপান্তরিত করে, বিশ্লেষণ সহজ করে

পরীক্ষামূলক সেটআপ

তাত্ত্বিক যাচাইকরণ পরীক্ষা

ডেটা উৎপাদন:

ম্যাট্রিক্স A, B ∈ R^(n×n), প্রবেশ iid N(0,1)
n = 3 × 2¹¹

বাস্তবায়ন বিবরণ:

মৌলিক ল্যাটিস: D₃ ল্যাটিস (3-মাত্রিক)
নেস্টেড অনুপাত: q = 6
লুকআপ টেবিল আকার: < 64KB (L1 ক্যাশে ফিট করতে পারে)
কার্যকর কোড-রেট: ≈ 3.015 বিট/প্রতীক

তুলনা পদ্ধতি

3-বিট স্কেলার কোয়ান্টাইজার (ℓ∞ নর্মালাইজেশন)
তাত্ত্বিক নিম্ন সীমা Γ(R)

পরীক্ষামূলক ফলাফল

প্রধান ফলাফল

কর্মক্ষমতা তুলনা:

প্রস্তাবিত পদ্ধতি: 1/n³ ∥Â⊤B - A⊤B∥²F ≈ 0.0593
3-বিট স্কেলার কোয়ান্টাইজেশন: ≈ 0.1668 (প্রায় 3 গুণ পার্থক্য)
তাত্ত্বিক নিম্ন সীমা: Γ(3.015) = 0.0304

মূল আবিষ্কার:

D₃ ল্যাটিস-ভিত্তিক স্কিম স্কেলার কোয়ান্টাইজেশনের চেয়ে উল্লেখযোগ্যভাবে ভাল
কর্মক্ষমতা তাত্ত্বিক সর্বোত্তমের কাছাকাছি (প্রায় 2 গুণ পার্থক্য)
n এর বৃদ্ধির সাথে তুলনায় কর্মক্ষমতা ব্যবধান আরও সংকুচিত হবে

জটিলতা বিশ্লেষণ

এনকোডিং জটিলতা: O(n log n) (দ্রুত Hadamard রূপান্তর ব্যবহার করে) ডিকোডিং জটিলতা: O(1) (লুকআপ টেবিল ব্যবহার করে) সংরক্ষণ ওভারহেড: প্রতিটি কোয়ান্টাইজারের জন্য স্কেলিং ফ্যাক্টর বর্ণনা করতে O(log n) অতিরিক্ত বিট প্রয়োজন

উপসংহার এবং আলোচনা

প্রধান উপসংহার

সর্বোত্তমতা: iid গাউসিয়ান ম্যাট্রিক্সের জন্য, প্রস্তাবিত স্কিম তথ্য-তাত্ত্বিক নিম্ন সীমা অর্জন করে
সর্বজনীনতা: যেকোনো ম্যাট্রিক্সের জন্য স্পষ্ট কর্মক্ষমতা গ্যারান্টি
পর্যায় রূপান্তর: R* ≈ 0.906 বিট/প্রবেশ একটি মূল থ্রেশহোল্ড
ব্যবহারিকতা: কম জটিলতার বাস্তবায়ন তাত্ত্বিক কর্মক্ষমতার কাছাকাছি

সীমাবদ্ধতা

ভাগ করা র্যান্ডমনেস: এনকোডার এবং ডিকোডারকে র্যান্ডম সিড ভাগ করতে হবে
ম্যাট্রিক্স শর্ত: ম্যাট্রিক্স প্রবেশ সীমাবদ্ধ হওয়া প্রয়োজন (M = n^(10^22000))
উচ্চ-মাত্রিক ল্যাটিস: তাত্ত্বিক সর্বোত্তমতার জন্য উচ্চ-মাত্রিক "ভাল" ল্যাটিস প্রয়োজন, ব্যবহারিকে নিম্ন-মাত্রিক ল্যাটিসের পণ্য ব্যবহার করা হয়
নির্ধারক স্কিম: এটি স্পষ্ট নয় যে র্যান্ডমনেস প্রয়োজন না করে সর্বোত্তম নির্ধারক স্কিম বিদ্যমান কিনা

ভবিষ্যত দিকনির্দেশনা

বহু-ম্যাট্রিক্স গুণফল: k>2 ম্যাট্রিক্সের গুণফলে সম্প্রসারণ
অন্যান্য দূরত্ব মেট্রিক্স: KL বিচ্যুতি ইত্যাদি অ-MSE মেট্রিক্স
নির্ধারক কোয়ান্টাইজার: ভাগ করা র্যান্ডমনেস প্রয়োজন না করে স্কিম অন্বেষণ করুন
গভীর নেটওয়ার্ক প্রয়োগ: বাস্তব LLM-এ স্থাপনা এবং অপ্টিমাইজেশন

গভীর মূল্যায়ন

সুবিধা

তাত্ত্বিক কঠোরতা: উপরের এবং নিম্ন সীমা সহ সম্পূর্ণ তথ্য-তাত্ত্বিক বিশ্লেষণ প্রদান করে
ব্যবহারিক মূল্য: LLM অনুমানে প্রকৃত বাধা সমাধান করে
প্রযুক্তিগত উদ্ভাবন: ল্যাটিস কোয়ান্টাইজেশন, র্যান্ডম রোটেশন এবং সময় ভাগাভাগি চতুরভাবে একত্রিত করে
সর্বজনীনতা: নির্দিষ্ট ম্যাট্রিক্স বিতরণ অনুমানের উপর নির্ভর করে না

অপূর্ণতা

জটিলতা: তাত্ত্বিক বিশ্লেষণ বেশ জটিল, বাস্তব বাস্তবায়নের জন্য একাধিক প্রযুক্তিগত উপাদান প্রয়োজন
ধ্রুবক ফ্যাক্টর: যদিও অসীম সর্বোত্তম, সীমিত নমুনায় ধ্রুবক বড় হতে পারে
হার্ডওয়্যার অভিযোজন: বিভিন্ন হার্ডওয়্যার প্ল্যাটফর্মের জন্য বাস্তবায়ন অপ্টিমাইজ করা প্রয়োজন
স্কেলেবিলিটি: দুটি ম্যাট্রিক্স থেকে একাধিক ম্যাট্রিক্সের গুণফলে সম্প্রসারণ অ-তুচ্ছ

প্রভাব

তাত্ত্বিক অবদান:

ম্যাট্রিক্স গুণন কোয়ান্টাইজেশনের তথ্য-তাত্ত্বিক ভিত্তি প্রতিষ্ঠা করে
পর্যায় রূপান্তর এবং মাত্রা হ্রাসের প্রয়োজনীয়তা প্রকাশ করে
ক্ষেত্রের জন্য গুরুত্বপূর্ণ তাত্ত্বিক বেঞ্চমার্ক প্রদান করে

ব্যবহারিক প্রয়োগ:

LLM কোয়ান্টাইজেশনের জন্য নতুন তাত্ত্বিক নির্দেশনা প্রদান করে
পরবর্তী কাজ NestQuant ইতিমধ্যে বাস্তব LLM-এ SOTA কর্মক্ষমতা অর্জন করেছে
হার্ডওয়্যার ত্বরণকারী ডিজাইনের জন্য তাত্ত্বিক ভিত্তি প্রদান করে

প্রযোজ্য দৃশ্যকল্প

বৃহৎ ভাষা মডেল অনুমান: ওজন এবং সক্রিয়করণের যৌথ কোয়ান্টাইজেশন
প্রান্ত কম্পিউটিং: মেমরি-সীমিত পরিবেশে ম্যাট্রিক্স অপারেশন
বিতরণ করা কম্পিউটিং: যোগাযোগ-সীমিত ম্যাট্রিক্স গুণন
বৈজ্ঞানিক কম্পিউটিং: বৃহৎ-স্কেল সংখ্যাগত রৈখিক বীজগণিত সমস্যা

রেফারেন্স

পেপারটি 44টি সম্পর্কিত সাহিত্য উদ্ধৃত করে, যা তথ্য তত্ত্ব, ল্যাটিস তত্ত্ব, র্যান্ডম লিনিয়ার অ্যালজেব্রা এবং স্নায়ু নেটওয়ার্ক কোয়ান্টাইজেশনের একাধিক ক্ষেত্রের গুরুত্বপূর্ণ কাজ অন্তর্ভুক্ত করে। বিশেষভাবে উল্লেখযোগ্য অন্তর্ভুক্ত:

Zamir এর ল্যাটিস এনকোডিং মনোগ্রাফ তাত্ত্বিক ভিত্তি প্রদান করে
Erez এবং Zamir এর নেস্টেড ল্যাটিস সম্পর্কিত যুগান্তকারী কাজ
OPTQ, QuIP ইত্যাদির মতো সাম্প্রতিক LLM কোয়ান্টাইজেশন পদ্ধতি
র্যান্ডম ম্যাট্রিক্স তত্ত্ব এবং গোলকীয় জ্যামিতির সম্পর্কিত ফলাফল

সামগ্রিক মূল্যায়ন: এটি তাত্ত্বিক এবং ব্যবহারিক উভয় ক্ষেত্রেই গুরুত্বপূর্ণ অবদান রাখা একটি উৎকৃষ্ট পেপার, যা ম্যাট্রিক্স গুণন কোয়ান্টাইজেশন সমস্যার জন্য দৃঢ় তথ্য-তাত্ত্বিক ভিত্তি প্রদান করে এবং সর্বোত্তমের কাছাকাছি ব্যবহারিক অ্যালগরিদম প্রদর্শন করে। এই কাজটি বৃহৎ-স্কেল মেশিন লার্নিং সিস্টেমে কোয়ান্টাইজেশন প্রযুক্তি বোঝা এবং উন্নত করার জন্য গুরুত্বপূর্ণ।