Fast Generation of Pipek-Mezey Wannier Functions via the Co-Iterative Augmented Hessian Method

ملخص تقني: التوليد السريع لدوال وانير (Wannier) من نوع "بايبيك-ميزي" عبر طريقة الهيسيان المعزز التعاوني المتكرر

بيان المشكلة
توفر دوال وانير (WFs)، التي تمثل تمثيلاً موضعياً في الفضاء الحقيقي لمدارات بلوخ (Bloch orbitals)، أساساً لتقريب النطاقات، وتقييم الخصائص الاستجابية، وتنزيل الهاملتوني (Hamiltonian downfolding)، وطرق الجسيمات المتعددة ذات القياس المختزل. ومن بين معايير التوضع المختلفة، يعد مخطط "بايبيك-ميزي" (Pipek–Mezey) مفيداً بشكل خاص للأنظمة الدورية نظراً لصياغته القائمة على التجمعات الذرية، مما يحافظ على تناظرات $\sigma$ و $\pi$ ويؤدي إلى مدارات ذات دلالة كيميائية بديهية. ومع ذلك، فإن تحسين دوال وانير (PMWFs) في الأنظمة الدورية التي يتم أخذ عينات منها باستخدام شبكة $k$ منتظمة يفرض تحديات حوسبية كبيرة.

تقدم النهج القائمة على التدرج، مثل خوارزمية "برويين-فليتشر-جولدبرج-شانز" (BFGS) المنفذة في الفضاء المتبادل، تحسناً في التقارب مقارنة بطرق التدرج المتدرج (steepest-descent)، لكنها تظل محدودة بسبب تقارب الدرجة الأولى المتأصل فيها. وفي المقابل، فإن الطرق من الدرجة الثانية مثل خوارزمية الهيسيان المعزز التعاوني المتكرر (CIAH)، رغم إظهارها تقارباً تربيعياً، كانت تاريخياً مقتصرة على صياغات النقطة $\Gamma$ للخلية الفائقة. إن التطبيق المباشر لخوارزمية CIAH الجزيئية على الخلايا الفائقة الدورية دون استغلال التناظر الانتقالي يؤدي إلى قياسات غير مرغوب فيها تصل إلى $O(N_k^3 n^3)$ في كل من وقت وحدة المعالجة المركزية والذاكرة (حيث $N_k$ هو عدد نقاط $k$ و $n$ يمثل حجم الخلية الوحدة)، مما يجعلها محظورة حوسبياً للشبكات الكبيرة من $k$ والأنظمة التي تحتوي على آلاف المدارات.

المنهجية
يقدم المؤلفون k-CIAH، وهو امتداد لنقاط $k$ لخوارزمية الهيسيان المعزز التعاوني المتكرر من الدرجة الثانية المصممة خصيصاً لتوضع "بايبيك-ميزي" لدوال وانير. تتضمن المنهجية المكونات الرئيسية التالية:

1. الإطار النظري: يتم تمثيل دوال (PMWFs) في الفضاء المتبادل عن طريق تطبيق دورات وحدوية على مجموعة أولية من مدارات بلوخ. يسعى التحسين إلى تعظيم دالة الهدف (PM)، والتي تُصاغ كمسألة تقليل للدالة السالبة.
2. خوارزمية k-CIAH: تستخدم الطريقة نهج منطقة الثقة لـ "نيوتن" (trust-region Newton)، حيث يتم تحديد الخطوة في فضاء المولد عبر حل مسألة القيم الذاتية للهيسيان المعزز باستخدام خوارزمية "ديفيدسون" (Davidson). تتضمن قاعدة التحديث ضرب المصفوفة الوحدوية الحالية في أس المصفوفة المولدة.
3. التقييم الفعال لضرب الهيسيان في المتجه: الابتكار الحاسم هو الاشتقاق لتقييم فعال لضرب الهيسيان في المتجه. من خلال استغلال الشكل الموزع لمؤثرات الإسقاط، أعاد المؤلفون كتابة حاصل ضرب الإسقاط في المتجه لتجنب تخزين وحساب المصفوفات الوسيطة التي تزداد بنحو $O(N_k^3)$. بدلاً من ذلك، تمت إعادة هيكلة الحساب ليتناسب مع $O(N_k^2 n^3)$.
4. تحليل القياس: يوضح المؤلفون أنه مع هذا التقييم الفعال، يتناسب كل من وقت وحدة المعاللة المركزية ومتطلبات الذاكرة لـ k-CIAH مع $O(N_k^2 n^3)$. وهذا يطابق قياس النهج السابقة من الدرجة الأولى في فضاء $k$ (مثل k-BFGS) مع الاحتفاظ بخصائص التقارب التربيعي للطرق من الدرجة الثانية.
5. الاستقرار والتهيئة: يتضمن التنفيذ إجراءً لمحاذاة الطور للتهيئة الأولية لتثبيت درجات الحرية في القياس (gauge degrees of freedom)، وتحليلاً للاستقرار يعتمد على مسح "جاكوبي" (Jacobi-sweep) للهروب من النقاط المستقرة المحلية أو نقاط السرج، مما يضمن تقارباً قوياً.
6. تناظر عكس الزمن (TRS): بالنسبة لشبكات $k$ المغلقة تحت عملية الانعكاس، تفرض الطريقة تناظر عكس الزمن لتقليل عدد المعلمات المستقلة وضمان أن تكون الدوال حقيقية القيم.

المساهمات الرئيسية

• التطوير الخوارزمي: تعميم طريقة CIAH على مدارات بلوخ مع أخذ عينات من نقاط $k$، والتي سُميت k-CIAH.
• الكفاءة الحوسبية: اشتقاق تقييم فعال لضرب الهيسيان في المتجه، مما قلل القياس من $O(N_k^3 n^3)$ (النموذجي لنهج النقطة $\Gamma$ للخلية الفائقة) إلى $O(N_k^2 n^3)$، مما يجعل تحسين الدرجة الثانية ممكناً للأنظمة الدورية الكبيرة.
• التقارب القوي: إثبات أن k-CIAH يحقق تقارباً تربيعياً، مما يقلل بشكل كبير من عدد التكرارات المطلوبة مقارنة بطرق k-BFGS من الدرجة الأولى.

النتائج
أُجريت حسابات مرجعية على مجموعة متنوعة من المواد الصلبة، بما في ذلك العوازل (h-BN، الماس، MgO)، وأشباه الموصلات (السيليكون، SiO$_2$)، والمعادن (الألومنيوم)، والأسطح (CO/MgO(001)).

• سرعة التقارب: حققت k-CIAH تقارباً ثابتاً في 5-20 تكراراً عبر جميع الأنظمة، بينما تطلب k-BFGS ما بين 30 إلى 300 تكرار.
• التكلفة الحوسبية: بالنسبة للأنظمة التي تحتوي على 1,000 إلى 5,000 مدار، أظهرت k-CIAH كفاءة حوسبية إجمالية أعلى بنحو 2-3 مرات من k-BFGS، وأعلى بعدة مراتب من $\Gamma$-CIAH.
• سلوك القياس: أكدت مخططات (log-log) لوقت وحدة المعالجة المركزية مقابل عدد المدارات القياس المتوقع $O(N_{orb}^2)$ لكل من k-CIAH و k-BFGS، بينما أظهرت $\Gamma$-CIAH قياساً قدره $O(N_{orb}^3)$.
• تقريب وانير (Wannier Interpolation): تم التحقق من جودة دوال (PMWFs) الناتجة من خلال تقريب البنية النطاقية الإلكترونية. قدم التقريب القائم على PMWF بنى نطاقية دقيقة للغاية حتى مع شبكات $k$ خشنة نسبياً (مثل $5 \times 5$)، متفوقاً بشكل ملحوظ على التقريب القائم على مدارات "كوهن-شام" غير الموضعية. كما أظهرت عناصر مصفوفة "فوك" (Fock matrix) في الفضاء الحقيقي باستخدام أساس PMWF اضمحلالاً مكانياً أسرع بكثير، مما سهل التقريب الدقيق.

الأهمية والادعاءات
يزعم البحث أن k-CIAH يوفر استراتيجية سريعة وقوية لتوضع "بايبيك-ميزي" في الأنظمة الدورية، متجاوزاً قيود كل من طرق التدرج من الدرجة الأولى (البطيئة في التقارب) وطرق الدرجة الثانية التقليدية للنقطة $\Gamma$ (ذات القياس الضعيف). ومن خلال تحقيق قياس $O(N_k^2 n^3)$، تمكنت الطريقة من توضع آلاف المدارات بكفاءة حوسبية عالية. ويؤكد المؤلفون أن المدارات الموضعية عالية الجودة الناتجة ضرورية لتقريب وانير الموثوق، وتعد شرطاً أساسياً حاسماً لطرق الجسيمات المتعددة ذات القياس المختزل القائمة على التضمين الكمي ونظريات الارتباط المحلي. يرسخ هذا العمل k-CIAH كبديل تنافسي ومتفوق لتقنيات التحسين الحالية في فضاء $k$ للمواد الصلبة الدورية.