الوصف
يمثل DimeNet و DimeNet++ تقدمًا كبيرًا في مجال الشبكات العصبية الجزيئية. نموذج DimeNet الأصلي، الذي تم تقديمه في "Directional Message Passing for Molecular Graphs" (ICLR 2020)، يستفيد من المعلومات الاتجاهية لتحسين تمرير الرسائل بين الذرات، مما يؤدي إلى تمثيلات أكثر دقة لهياكل وخصائص الجزيئات. يعزز هذا النهج قدرة النموذج على التقاط التفاعلات المعقدة داخل الجزيئات.
DimeNet++، النموذج الخلف الذي تم تقديمه في "Fast and Uncertainty-Aware Directional Message Passing for Non-Equilibrium Molecules" (NeurIPS-W 2020)، يبني على أساس DimeNet من خلال تقديم تحسينات كبيرة في السرعة والدقة. يُقال إنه أسرع بما يصل إلى 8 مرات وأكثر دقة بنسبة 10٪ من DimeNet الأصلي، مما يجعله خيارًا عمليًا أكثر لمحاكاة الجزيئات على نطاق واسع وتطبيقات اكتشاف الأدوية. يعمل التنفيذ المقدم في مستودع GitHub هذا كمرجع لكلا النموذجين.
يتضمن المستودع المكونات الأساسية لتشغيل هذه النماذج وتجربتها. يتميز بدفاتر ملاحظات للتدريب (`train.ipynb`) والتنبؤ (`predict.ipynb`)، مما يسمح للمستخدمين بتطبيق النماذج بسهولة على مجموعات بياناتهم. للمستخدمين الذين يحتاجون إلى تدريب قائم على المجموعات، يتوفر أيضًا برنامج نصي (`train_seml.py`) مدمج مع Sacred و SEML. علاوة على ذلك، يتم توفير نماذج مدربة مسبقًا لكل من DimeNet++ ونموذج أولي لـ TensorFlow 2.0 في مجلد `pretrained`، مما يسهل التجريب والنشر بشكل أسرع.
التفاصيل المعمارية الرئيسية وملاحظات التنفيذ متاحة داخل المستودع، بما في ذلك استراتيجيات التهيئة المحددة لطبقات الإخراج ومناقشات حول المشكلات أو الأخطاء المحتملة. تم تنفيذ المشروع باستخدام Python مع TensorFlow كإطار عمل رئيسي للتعلم العميق، مدعومًا بمكتبات مثل NumPy و SciPy و SymPy و TensorFlow Addons. يسلط المشروع الضوء أيضًا على تطبيقات أخرى في JAX & Haiku و PyTorch Geometric، مما يشير إلى مجتمع بحثي نابض بالحياة حول هذه النماذج. النماذج ذات صلة بشكل خاص بالباحثين والمطورين في الكيمياء الحاسوبية وعلوم المواد والمعلوماتية الكيميائية الذين يسعون إلى الاستفادة من التعلم العميق لتنبؤ خصائص الجزيئات وتحليلها.
أبرز ملامح DimeNet و DimeNet++
تمرير الرسائل الاتجاهي للرسوم البيانية الجزيئية
بنية نموذج DimeNet
بنية نموذج DimeNet++
تنفيذ مرجعي
دفتر ملاحظات تدريب متضمن
دفتر ملاحظات تنبؤ متضمن
برنامج نصي لتدريب المجموعات متاح
نماذج مدربة مسبقًا متوفرة
نموذج أولي لـ TensorFlow 2.0 متاح
التركيز على تنبؤات الطاقة والقوة (عبر توصية GemNet)
تنبؤات واعية بالشك (DimeNet++)
البدء مع DimeNet و DimeNet++
الوصول إلى النموذج: استنساخ مستودع GitHub.
إعداد البيئة: تثبيت حزم Python المطلوبة (TensorFlow، NumPy، SciPy، SymPy، TensorFlow Addons).
التكامل عبر API: استخدم دفاتر الملاحظات أو البرامج النصية المقدمة للتدريب والتنبؤ.
تجربة النماذج: تحميل النماذج المدربة مسبقًا لتكرار أسرع.
التكيف مع تدريب المجموعات: استخدم البرنامج النصي `train_seml.py` للتدريب الموزع.
إنشاء تنبؤات: استخدم دفتر ملاحظات `predict.ipynb` مع النماذج المدربة.
حالات استخدام DimeNet و DimeNet++
- تنبؤ خصائص الجزيئات
- اكتشاف الأدوية
- علوم المواد
- أبحاث المعلوماتية الكيميائية
- الكيمياء الحاسوبية






