UAlign: Pushing the Limit of Template-free Retrosynthesis Prediction with Unsupervised SMILES Alignment
يُعد UAlign مساراً مبتكراً لتحويل الرسوم البيانية إلى تسلسلات (graph-to-sequence) يعتمد على القوالب للتنبؤ بالتخليق الرجعي، حيث يستفيد من الشبكات العصبية الرسومية، ونماذج المحولات (Transformers)، وتقنية محاذاة SMILES غير الخاضعة للإشراف ليتفوق بشكل كبير على الأساليب الحديثة مع مضاهاة النهج القائمة على القوالب الراسخة.
المؤلفون الأصليون:Kaipeng Zeng, Bo yang, Xin Zhao, Yu Zhang, Fan Nie, Xiaokang Yang, Yaohui Jin, Yanyan Xu
تخيل أنك شيف ماهر يحاول إعادة ابتكار طبق شهير ومعقد بمجرد النظر إلى الطبق النهائي. أنت تعلم أن المكونات وتقنيات الطهي موجودة، ولكن عليك اكتشاف كيفية تفكيك الوجبة بدقة للوصول إلى مكوناتها الأولية. هذا هو التحدي اليومي للكيميائيين، وهو مجال يُعرف باسم التخليق العضوي. يتعين عليهم العمل بشكل عكسي من دواء أو مادة مرغوبة للوصول إلى المكونات البسيطة والرخيصة اللازمة لبنائها. تُسمى هذه العملية "التخليق الرجعي" (retrosynthesis). لعقود من الزمن، حاولت الحواسيب المساعدة في ذلك، لكنها غالباً ما تتعثر. تعتمد بعض الأساليب على كتاب وصفات ضخم وثابت للوصفات المعروفة (القوالب)، وهو ما يفشل عندما يظهر طبق جديد وغريب. أما الأساليب الأخرى، فتحاول تخمين المكونات من الصفر، مثل طاهٍ لم يرَ الطبق من قبل، وغالباً ما ينتهي به الأمر بوصفة لا معنى لها أو تستخدم مكونات غير موجودة. السؤال الكبير هو: هل يمكن للحاسوب أن يتعلم كيف "يعكس عملية الطهي" لجزيء ما بكفاءة دون الحاجة إلى كتاب وصفات مكتوب مسبقاً، مع استيعاب القواعد العميقة للكيمياء في الوقت ذاته؟
إليك UAlign، المساعد الرقمي الجديد المصمم لحل هذا اللغز. لقد بنى الباحثون وراء هذه الورقة البحثية، كايبينج زينج وفريقه، نظاماً يعمل مثل محقق فائق الذكاء لا يكتفي بتخمين المكونات فحسب، بل يفهم بالفعل هيكل الطبق النهائي. فبدلاً من معاملة الجزيئات كسلاسل عشوائية من الحروف (كما تقرأها الحواسيب عادةً)، ينظر UAlign إلى الجزيء كخريطة ثلاثية الأبعاد من الاتصالات، تماماً مثل نظام مترو الأنفاق حيث المحطات هي الذرات والمسارات هي الروابط الكيميائية.
الحيلة الذكية التي يستخدمها UAlign هي مفهوم "محاذاة SMILES غير الخاضعة للإشراف". فكر في الأمر بهذه الطريقة: إذا قمت بتفكيك قلعة من قطع "ليغو" (Lego)، فإن معظم الطوب سيبقى في مكانه تماماً؛ فقط قطع قليلة هي التي تتغير أو تُزال. يدرك UAlign أنه في التفاعل الكيميائي، تكون الأجزاء "غير المتغيرة" من الجزيء هي الأسهل في الرصد. لذا، بدلاً من محاولة إعادة بناء الشيء بأكمله من الصفر، يقوم بمحاذاة الأجزاء غير المتغيرة من المكونات الأولية مع المنتج النهائي تلقائياً، دون الحاجة إلى تدخل بشري لتسمية كل اتصال على حدة. إنه يشبه امتلاك ماسح ضوئي سحري يسلط الضوء فوراً على الأجزاء التي لم تتحرك في قلعة الليغو، مما يسمح للحاسوب بتركيز قدراته الذهنية فقط على القطع القليلة التي تغيرت بالفعل.
النتائج مثيرة للإعجاب. فعند اختباره على قواعد بيانات ضخمة من التفاعلات الكيميائية، أثبت UAlign أنه أفضل بكثير في التنبؤ بالمكونات الأولية الصحيحة من الأساليب السابقة "الخالية من القوالب". في الواقع، كان أداؤه جيداً لدرجة أنه استطاع اللحاق بالأساليب القديمة التي تعتمد على كتب الوصفات الجامدة، بل وتفوق عليها أحياناً. على سبيل المثال، في إحدى مجموعات الاختبار الرئيسية، تنبأ بالمواد الأولية العشر الممكنة بشكل صحيح بنسبة 90% من الوقت، وهي قفزة كبيرة مقارنة بمنافسيه. كما أظهر قدرته على تخطيط وصفات متعددة الخطوات للأدوية المعقدة في العالم الحقيقي، مثل Mitapivat وPacritinib، من خلال تفكيكها بنجاح إلى سلائف أبسط.
ومع ذلك، يوضح المؤلفون بحذر أنه على الرغم من كون UAlign أداة جديدة قوية، إلا أنه ليس عصا سحرية تحل كل شيء. فهو لا يزال يعاني قليلاً في توليد مجموعة متنوعة من الإجابات المختلفة (التنوع)، ولا يشرح تماماً لماذا اتخذ خياراً كيميائياً معيناً، وهو أمر قد يصعب على الكيميائيين البشريين الوثوق به. ولكن من خلال الجمع بين الفهم العميق للأشكال الجزيئية وطريقة ذكية لمحاذاة القطع، يدفع UAlign حدود ما يمكن للحواسيب القيام به في المختبر، مما يجعل الرحلة من جزيء مستهدف إلى دواء حقيقي في العالم الواقعي أقل شبهاً بلعبة التخمين وأكثر شبهاً بلغز تم حله.
ملخص تقني: UAlign
بيان المشكلة
يعد التنبؤ بالتخليق الرجعي أحادي الخطوة تحديًا جوهريًا في التخطيط الاصطناعي بمساعدة الحاسوب (CASP)، والذي يهدف إلى التنبؤ بالمتفاعلات المطلوبة لتخليق جزيء المنتج المستهدف. وبينما حقق التعلم العميق تقدمًا في هذا المجال، تواجه الأساليب الحالية قيودًا كبيرة:
الأساليب القائمة على القوالب (Template-based): تعتمد على قواعد بيانات قوالب التفاعل، مما يوفر قابلية للتفسير ولكنها تعاني من ضعف القدرة على التوسع وعدم القدرة على تغطية الحالات الخارجة عن مكتبة القوالب.
الأساليب شبه القائمة على القوالب (Semi-template-based): تقسم المهمة إلى مهام فرعية (التنبؤ بالسينثون وإكمال المتفاعلات)، مما يزيد من تعقيد الاستدلال وصعوبة التدريب؛ حيث تؤدي الإخفاقات في الخطوات المبكرة إلى انتقال الخطأ إلى المخرجات النهائية.
الأساليب غير القائمة على القوالب (Template-free): وهي النماذج التوليدية التي تولد المتفاعلات مباشرة من سلاسل SMILES للمنتج. ومع ذلك، غالبًا ما تتجاهل هذه الأساليب المعلومات الطوبولوجية والكيميائية الغنية للروابط الموجودة في الرسوم البيانية الجزيئية. علاوة على ذلك، فهي تولد المتفاعلات عادةً من الصفر، مما يفشل في الاستفيد من الحدس الكيميائي الذي يقتضي بأن معظم الهياكل الجزيئية تظل دون تغيير أثناء التفاعل. إن المحاولات الحالية لاستخدام هذا عبر محاذاة SMILES الخاضعة للإشراف تتطلب وسمًا معقدًا للبيانات ويمكن أن تقيد تنوع النموذج.
المنهجية
يقترح البحث UAlign، وهو مسار جديد (graph-to-sequence) غير قائم على القوالب للتنبؤ بالتخليق الرجعي أحادي الخطوة. تتكون البنية من إطار عمل "مشفر-مفكك" (encoder-decoder) يتضمن المكونات الرئيسية التالية:
1. المشفر: +EGAT
بدلاً من استخدام نماذج Transformer القياسية أو الشبكات العصبية الرسومية (GNNs) بشكل منفصل، يستخدم UAlign شبكة انتباه رسومية معدلة تسمى +EGAT.
تمرير الرسائل المدرك للحواف (Edge-Aware Message Passing): يدمج +EGAT معلومات الروابط الكيميائية (ميزات الحواف) صراحةً في عملية تمرير الرسائل جنبًا إلى جنب مع ميزات العقد.
الآلية: يطبق الانتباه الذاتي على العقد وجيرانها من خطوة واحدة، ويحسب معاملات الانتباه بناءً على كل من ميزات العقد والحواف. يسمح هذا للنموذج بالتقاط الخصائص الهيكلية وخصائص الروابط الكيميائية بشكل أكثر فعالية من شبكات GNN القياسية.
المخرج: ينتج المشفر ميزات العقد H التي تمثل الرسم البياني لجزيء المنتج.
2. محاذاة SMILES غير الخاضعة للإشراف
الابتكار الأساسي هو آلية محاذاة SMILES غير الخاضعة للإشراف، والتي تعالج مشكلة توليد البنى الفرعية غير المتغيرة من الصفر.
المفهوم: نظرًا لأن المتفاعلات والمنتجات تشترك في تداخل هيكلي كبير، يهدف النموذج إلى توليد رموز (tokens) SMILES للمتفاعلات بترتيب يتوافق مع ترتيب ذرات المنتج.
ترتيب SMILES للمتفاعلات المحافظ على الترتيب: تحدد الطريقة "ترتيبًا متوافقًا مع البحث بالعمق" (DFS-corresponding order) للمتفاعلات يقلل من عكس رتب الذرات بالنسبة لترتيب البحث بالعمق (DFS) للمنتج.
التدريب غير الخاضع للإشراف: على عكس أساليب المحاذاة الخاضعة للإشراف التي تتطلب تسميات صريحة لمطابقة الذرات (والتي قد تسرب معلومات أثناء الاستدلال)، يقوم UAlign بتدريب النموذج على توليد الرموز باتبال ترتيب DFS المعطى للمنتج دون تسريب صريح للعلامات. وهذا يؤسس ارتباطًا بين ذرات المنتج ورموز المتفاعلات من خلال قيد الترتيب فقط.
3. المفكك (Decoder)
الميزات المدركة للترتيب: نموذج Transformer الأصلي هو نموذج مستقل عن الترتيب (permutation-invariant)، وهو أمر إشكالي لمحاذاة ذرات محددة مع رموز محددة. يضيف UAlign ترميزًا موضعيًا (positional encoding) لميزات عقد المشفر بناءً على رتبة كل ذرة في ترتيب DFS المعطى.
التوليد: تعمل هذه الميزات المدركة للترتيب كـ "مفاتيح" (keys) و"قيم" (values) في طبقات الانتباه المتقاطع (cross-attention)، مما يمكن المفكك من توليد رموز SMILES للمتفاعلات بشكل تلقائي (autoregressively) مع الحفاظ على الارتباط الهيكلي مع المنتج.
4. استراتيجية التدريب
تدريب مرحلتين:
المرحلة الأولى (ترجمة الرسم البياني إلى SMILES): يتم تدريب النموذج لترجمة الرسم البياني للجزيء إلى تمثيل SMILES المقابل له بناءً على ترتيب DFS محدد. هذا يربط توزيعات الوسائط (الرسم البياني وSMIS) ويعلم النموذج كيفية التعامل مع أنماط SMILES غير القياسية.
المرحلة الثانية (التخليق الرجعي): يتم تدريب النموذج على توليد رموز SMIS للمتفاعلات المحافظة على الترتيب بناءً على رسم بيئي للمنتج وترتيب DFS الخاص به.
تعزيز البيانات (Data Augmentation): لمنع الإفراط في التخصيص (overfitting) لأنماط SMILES القياسية وتحسين المتانة، يتم تعزيز بيانات التدريب آنيًا. يتضمن ذلك اختيار ترتيبات DFS عشوائية، وفي بعض الحالات، دمج الجزيئات لتشكيل رسوم بيانية جديدة، مما يضمن تعلم النموذج كيفية التعامل مع مختلف المكونات الرسومية وقيود الترتيب.
المساهمات الرئيسية
مسار جديد من الرسم البياني إلى التسلسل (Graph-to-Sequence): يقدم UAlign نهجًا كاملًا غير قائم على القوالب يستفيد بفعالية من الهياكل الرسومية الجزيئية عبر +EGAT، متجاوزًا قيود نماذج (sequence-to-sequence) البحتة في التقاط المعلومات الطوبولوجية.
محاذاة SMILES غير الخاضعة للإشراف: يقترح البحث آلية بسيطة وفعالة غير خاضعة للإشراف لإقامة ارتباط بين ذرات المنتج ورموز المتفاعلات. يتجنب هذا الحاجة إلى وسم معقد للبيانات أو خطر تسريب العلامات، مع تحقيق أداء يتفوق على أساليب المحاذاة الخاضعة للإشراف.
التدريب على مرحلتين وتعزيز البيانات: تربط هذه الاستراتيجية بفعالية الفجوة في التوزيع بين تمثيلات الرسم البياني وSMILES وتُمكّن النموذج من تعلم أنماط SMILES غير القياسية، مما يحسن بشكل كبير من صحة ودقة التوليد.
النتائج
أُجريت تجارب واسعة النطاق على ثلاث مجموعات بيانات مرجعية: USPTO-50K، و USPTO-FULL، و USPTO-MIT.
الدقة: يتفوق UAlign بشكل كبير على الأساليب الحالية (SOTA) غير القائمة على القوالب وشبه القائمة على القوالب.
في USPTO-50K (فئة التفاعل غير معروفة)، حقق دقة Top-1 و Top-5 و Top-10 بنسبة 53.6% و 84.6% و 90.3% على التوالي، متجاوزًا أفضل نموذج أساسي غير قائم على القوالب بنسبة 3.5% و 4.0% و 4.7% على التوالي.
كما أنه ينافس أو يتفوق على الأساليب القوية القائمة على القوالب (مثل LocalRetro) في عدة إعدادات.
في USPTO-FULL، حقق دقة Top-1 بنسبة 50.4%، متجاوزًا نموذج GTA المتطور بنسبة 3.8%.
الصلاحية (Validity): أظهر النموذج صلاحية SMILES فائقة (99.8% لـ Top-1) مقارنة بالنماذج الأخرى القائمة على SMILES، ويعزى ذلك إلى التدريب على مرحلتين وتعزيز البيانات.
دقة الرحلة العكسية (Round-Trip Accuracy): حقق UAlign أعلى دقة Top-1 للرحلة العكسية (80.9%)، متفوقًا على كل من النماذج الأساسية القائمة على SMILES والرسوم البيانية، مما يشير إلى توليد مسارات معقولة كيميائيًا.
دراسات الحالة: نجح النموذج في التنبؤ بمسارات تخليق متعددة الخطوات لجزيئات معقدة مثل Mitapivat و Pacritinib و Daprodustat، وغالبًا ما طابق أو حسن المسارات المسجلة في الأدبيات العلمية.
الأهمية والادعاءات
يدعي البحث أن UAlign يمثل تقدمًا كبيرًا في التنبؤ بالتخليق الرجعي غير القائم على القوالب من خلال:
سد الفجوة: نجح في الجمع بين الوعي الهيكلي للشبكات العصبية الرسومية والمرونة التوليدية لنماذج Transformer.
الكفاءة: تسمح آلية المحاذاة غير الخاضعة للإشراف للنموذج بإعادة استخدام البنى الفرورية المشتركة بين المتفاعلات والمنتجات دون عبء التوسيم المعقد أو خطر تسريب العلامات.
الأداء: يثبت أن المنهج المصمم جيدًا وغير القائم على القوالب يمكن أن يحقق فعالية تضاهي، أو حتى تتفوق على، الأساليب الراسخة القائمة على القوالب، مع توفير قدرة أفضل على التوسع والتنوع لمجموعات البيانات الكبيرة.
الإمكانات المستقبلية: يضع المؤلفون UAlign كركيزة قوية لعمليات التخطيط المستقبلي للتخليق الرجعي متعدد الخطوات، رغم إقرارهم بالقيود الحالية في التفسير وتنوع التوليد.