التمرير الهرمي بين الرسائل لتعلم الرسوم البيانية الجزيئية

نقدم معمارية تبادل الرسائل العصبية الهرمية لتعلم البيانات على الرسوم البيانية الجزيئية. تعتمد نموذجنا على تمثيلين متكاملين للرسم البياني: تمثيل الرسم البياني الجزيئي الأصلي وشجرة الاتصال المرتبطة به، حيث تمثل العقد في الشجرة مجموعات ذات معنى في الرسم البياني الأصلي، مثل الحلقات أو المركبات الجسرية. ثم نواصل تعلم تمثيل الجزيء من خلال تبادل الرسائل داخل كل رسم بياني، مع تبادل الرسائل بين التمثيلين باستخدام تدفق معلومات من المستوى الخشن إلى الدقيق والعكس. تمكننا هذه الطريقة من التغلب على بعض القيود المعروفة في الشبكات العصبية الرسومية التقليدية، مثل اكتشاف الدورات، مع الحفاظ على كفاءة عالية في التدريب. وقد تم التحقق من أداء النموذج على مجموعة بيانات ZINC وعلى مجموعات بيانات مشتقة من مجموعة معايير MoleculeNet.