Malzeme biliminde moleküler dinamik simülasyonları, maddelerin makroskobik özelliklerini yöneten atomik ayrıntıları ortaya çıkarmak için yaygın olarak kullanılmaktadır. Yoğunluk fonksiyonu teorisi gibi kuantum mekaniksel yöntemler yüksek doğruluk sunsa da, hesaplama karmaşıklığı atom sayısının küpüyle ölçeklendiğinden simülasyon boyutu ve süresi önemli ölçüde sınırlanmaktadır. Bu engeli aşmak amacıyla geliştirilen makine öğrenmesi potansiyelleri ve özellikle eşdeğişir çizge sınır ağı potansiyelleri, derin öğrenme aracılığıyla veri verimliliğini ve doğruluğu artırmaktadır. Ancak çoklu mesaj iletim katmanlarının alıcı alanını genişletmesi, büyük ölçekli moleküler dinamik paketlerinde tercih edilen uzaysal ayrıştırma paralelleştirme yönteminde veri iletişimini karmaşıklaştırmaktadır. Bu durum, eşdeğişir çizge sınır ağı potansiyellerinin çoklu GPU sistemlerinde ölçeklenebilir biçimde çalıştırılmasını zorlaştırmaktadır.
Bu sorunu çözmek amacıyla araştırmacılar, çizge sınır ağı potansiyelleri ile uyumlu uzaysal ayrıştırma tabanlı etkin bir paralel algoritma geliştirmiş ve NequIP mimarisine dayanan SevenNet yazılım paketini sunmuşlardır. Geliştirilen yaklaşımda işlemciler arası iletişim aralığı orijinal kesme yarıçapı ile sınırlandırılmış; ileri yönde atom konumları ve düğüm özellikleri, ters yönde ise enerji gradyanları iletilerek kuvvet hesaplamaları eksiksiz gerçekleştirilmiştir. SevenNet paketi moleküler dinamik simülasyonları için LAMMPS yazılımıyla entegre edilmiş ve PyTorch TorchScript altyapısıyla derlenmiştir. Algoritmanın başarımı, SiO2 sistemi üzerinde farklı kanal sayıları ve mesaj iletim katmanlarıyla eğitilen 20 model kullanılarak 32 GPU'lu küme üzerinde zayıf ve güçlü ölçekleme testleriyle kapsamlı şekilde değerlendirilmiştir. Ayrıca Materials Project veri kümesiyle eğitilen SevenNet-0 genel amaçlı potansiyeli oluşturulmuş ve 112.000 atomlu amorf Si3N4 sisteminin erime-soğutma simülasyonu gerçekleştirilmiştir.
Yapılan başarım testleri sonuçunda SevenNet, 32 GPU'lu kümede SiO2 sistemleri için yürütülen zayıf ölçekleme senaryolarında %80'in üzerinde paralel verimliliğe ulaşmıştır. Modellerdeki kanal sayısı veya mesaj iletim katmanı sayısı arttıkça iletişim veri boyutu ve sıklığı yükselmesine rağmen paralel verimliliğin korunduğu gözlemlenmiştir. Sabit boyutlu güçlü ölçekleme testlerinde ise 32 kanallı modeller 32 GPU kullanımında 13.1 ile 15.4 arasında hızlanma oranı elde etmiş, ancak GPU kaynaklarının yetersiz kullanımı durumunda küçük atom sayılı sistemlerde performans artışı doyuma ulaşmıştır. Genel amaçlı SevenNet-0 potansiyeli ile yürütülen 112.000 atomlu amorf Si3N4 erime-soğutma simülasyonu 8 GPU üzerinde 0.94 paralel verimlilikle tamamlanmış ve elde edilen yapısal radyal ile açısal dağılım fonksiyonlarının ilk ilkeler kuantum hesaplama referanslarıyla tam uyum sağladığı doğrulanmıştır.
Çalışmada elde edilen bulgular, eşdeğişir çizge sınır ağı potansiyellerinin doğru tasarlanmış ileri ve ters iletişim rutinleri sayesinde uzaysal ayrıştırma mimarisinde yüksek verimle paralelleştirilebileceğini kanıtlamaktadır. Yazar tartışmasında, mesaj iletimli cizge ağlarının genişleyen alıcı alanlarına rağmen iletişim mesafesini kesme yarıçapında tutmanın gereksiz hesaplama yükünü önlediğini ve derin modellerde serial yük artışının nispi iletişim maliyetini düşürdüğünü belirtmiştir. Literatürde daha önce sınırlı atom sayılarıyla kısıtlanan eşdeğişir potansiyel simülasyonları, SevenNet sayesinde yüz binlerce atom içeren karmaşık malzeme sistemlerine genişletilebilmektedir. Bu durum, Türk araştırmacıların büyük ölçekli amorf malzeme tasarımı, faz değişimi ve aygıt düzeyindeki moleküler dinamik araştırmalarında kuantum kesinliğinde veri üretmelerine ve mevcut LAMMPS altyapılarına doğrudan entegre edebilecekleri güçlü bir araç elde etmelerine olanak tanımaktadır.
Kuramsal ve kavramsal çerçeve
Çalışma, atomik grafiklerde eşdeğişir temsil ilkesine dayanan NequIP çizge sınır ağı mimarisini ve paralel hesaplamada atomik alt alan paylaşımını sağlayan uzaysal ayrıştırma (spatial decomposition) kuramsal çerçevesini esas almaktadır.
Makalenin ele aldığı literatür
Makale, geleneksel atomlar arası potansiyeller (Behler-Parrinello, GAP), değişmez cizge sınır ağları (SchNet, PhysNet, M3GNet) ve eşdeğişir çizge sınır ağı modelleri (NequIP, Allegro, MACE) ile büyük ölçekli moleküler dinamik simülasyon yazılımı LAMMPS literatürünü ele almaktadır.
Araştırma Tasarımı
- Evren veya inceleme alanı
- M3GNet veri kümesi ve SiO2 ile Si3N4 bileşiklerinin atomik yapısal konfigürasyonları
- Örneklem veya araştırma materyali
- 1600 adet α-quartz SiO2 DFT yörünge konfigürasyonu ve Materials Project veri tabanından türetilen 89 elementi kapsayan M3GNet veri kümesi
- Veri toplama süreci
- VASP paketiyle PBE GGA fonksiyoneli kullanılarak yürütülen DFT moleküler dinamik simülasyonlarından alınan veri örnekleri
- Veri analiz yöntemi
- PyTorch ve e3nn kütüphaneleriyle eğitilen modellerin TorchScript ile LAMMPS simülasyon motoruna bağlanması ve uzaysal ayrıştırma katmanları arasında ileri/ters iletişim adımlarının yürütülmesi
Çizge Sınır Ağı Atomlar Arası Potansiyeli (GNN-IP)Uzaysal Ayrıştırma ParalelleştirmesiE(3)-Eşvaryant TemsilZayıf ve Güçlü Ölçekleme VerimliliğiErime-Soğutma Simülasyonu
01İletişim aralığını kesme yarıçapı ile sınırlayan paralel uzaysal ayrıştırma algoritması, zayıf ölçekleme senaryolarında 32 GPU kullanıldığında dahi %80'in üzerinde paralel verimlilik sağlamaktadır(Park et al., 2024).
02Sabit boyutlu güçlü ölçekleme testlerinde, 32 GPU kullanıldığında 32 kanallı modellerde 13.1 ile 15.4 kat hızlanma elde edilmekte; ancak kanal sayısının düşmesi GPU kapasitesinin yetersiz kullanımına yol açarak performans doyumunu öne çekmektedir(Park et al., 2024).
0389 elementi kapsayan M3GNet veri kümesi üzerinde eğitilen genel amaçlı SevenNet-0 modeli, test kümesinde enerji için 24 meV/atom, kuvvet için 0.067 eV/A ve gerilim için 0.65 GPa ortalama mutlak hata değerlerine ulaşmıştır(Park et al., 2024).
04Pre-trained SevenNet-0 potansiyeli kullanılarak gerçekleştirilen 112.000 atomlu amorf Si3N4 erime-soğutma simülasyonu, 8 GPU üzerinde 12.7 saatte tamamlanmış ve %94 zayıf ölçekleme paralel verimliliği sergilemiştir(Park et al., 2024).