新聞中心

NVIDIA ALCHEMI:AI 驅動材料發現的工作流程與評估路徑 NEWS DETAIL

當前位置:首頁 > 新聞中心
資訊分類 · 新聞中心 作者 · 中科新遠內容團隊 審核人 · 中科新遠技術內容組 發佈時間 · 2025-01-20 更新時間 · 2026-07-22 來源 · 原頁面資料;原廠資訊待核驗
NVIDIA ALCHEMI:AI 驅動材料發現的工作流程與評估路徑

若材料研發團隊需要在進入實驗室前,先從大量候選結構中找出較值得驗證的材料,NVIDIA ALCHEMI 所描述的路徑是:以 AI 支援假設生成、候選空間定義、屬性預測與實驗驗證,並在幾何鬆弛等高重複計算環節採用 NVIDIA NIM 與機器學習原子間相互作用勢能模型(MLIP)進行加速。這不是取代實驗或 DFT 的單一步驟方案;其價值在於把候選生成、快速篩選、高保真模擬與實驗決策串接為可評估的工作流程。

材料發現情境與架構路徑

新材料發現通常包含四個相互依賴的階段:假設生成、解空間定義、屬性預測及實驗驗證。來源指出,化學資訊大型語言模型可協助彙整文獻與表述假設;既有化學資料庫可用於界定研究子空間;生成式 AI 可依目標屬性提出候選結構。其後,分子表徵模型、MLIP、圖神經網路(GNN)與 DFT 可在不同精度和計算成本間分工。

建議的架構不是把所有候選直接送進昂貴模擬,而是分層處理:先以資料庫搜尋或生成模型建立候選集,再以屬性模型及 MLIP 篩選,對優先候選執行 DFT 驗證,最後安排合成與測試。對生成式模型提出的結構而言,幾何鬆弛尤其重要,因為候選結構未必處於穩定狀態;系統需要反覆計算原子力與能量,調整原子位置,再依鬆弛後能量與目標屬性排序。

批次幾何鬆弛的實作方式

來源將 NVIDIA Batched Geometry Relaxation NIM 描述為公開 API 的容器,聚焦於在 Atomic Simulation Environment(ASE)中,以 MACE-MP-0 或 AIMNet2 執行潛在電池材料相關的幾何鬆弛。用戶端以 ase.Atoms 表示材料或分子,序列化為 JSON 後,透過 POST 請求送至 NIM API;鬆弛完成的結果再轉回 ase.Atoms。範例使用 Docker 啟動容器、指定 GPU,並將本機連接埠 8003 對映至 API 端點 /v1/infer

其核心考量是批次化。來源指出,逐一處理系統可能造成 GPU 使用率不足,也可能帶來 CPU 與 GPU 間資料移動的額外損失。透過 NVIDIA Warp 編寫可在執行期編譯為 GPU 核心程式碼的 Python 函式,可平行執行多個能量最小化工作。這種方法特別適合候選數量多、每個候選均需數千次鬆弛步驟的篩選作業。

部署與驗證檢查點

  1. 先定義材料類型、目標屬性、可接受誤差與最終實驗判定標準,避免僅以推論速度決定模型選擇。
  2. 確認團隊具備 Python、ASE、Docker 與 GPU 容器部署能力,並在完整 SKU/BOM 中確認實際 GPU、儲存、網路與執行環境配置。
  3. 依研究對象選擇模型評估路徑:來源範例將 MACE-MP-0 用於週期性無機晶體,AIMNet2 用於小型有機分子;模型適用範圍與訓練資料相符性仍須逐案確認。
  4. 以代表性候選集比較 MLIP 結果、DFT 結果與實驗量測,檢查能量排序、結構收斂、失敗率及批次大小變動後的結果一致性。
  5. 將 NIM API 納入既有資料追蹤流程,保留輸入結構、模型版本、參數、收斂條件與輸出結果,讓後續實驗能回溯篩選依據。

效能數據的解讀與風險

來源列出的效能結果具有明確測試條件,不能直接外推至所有材料工作負載。在單張 NVIDIA H100 80 GB GPU、MACE-MP-0、ASE FIRE 最佳化器,以及 2,048 個每晶胞含 20 至 40 個原子的週期性無機晶體條件下,來源記錄未使用 NIM 為 874 秒;啟用 NIM、批次大小為 1 時為 36 秒;批次大小為 128 時為 9 秒。對 GDB-17 中 851 個平均約 20 原子的小型有機分子,使用 AIMNet2 與 ASE FIRE 的來源數據則為 678 秒、12 秒與批次大小 64 時的 0.9 秒。

這些數據反映特定模型、系統規模、批次設定與硬體條件下的結果。來源亦指出,MACE-MP-0 與 AIMNet2 的加速差異源自週期性晶體與小分子等建模系統差異,而非僅由部署模型決定。實際專案應以自身材料組成、原子數、收斂門檻、模型精度需求及資料傳輸路徑重做測試;模型容器位址、授權條件、支援版本與部署要求,應查閱有日期的 NVIDIA 官方產品文件後確認。

常見問題

MLIP 能否完全取代 DFT 與實驗驗證?

不能。來源將 MLIP 定位為在準確性與計算成本間取得平衡的代理模型,可用於快速預測能量與原子力,以及支援大規模篩選。對需要高保真確認的候選,仍應以 DFT 模擬驗證,並以實驗確認可合成性、穩定性與實際性能。

何時適合採用批次幾何鬆弛?

當研究流程同時處理大量候選,且每個候選需多次能量最小化時,批次執行更值得評估。若候選數量少、單一系統過大、模型不適合目標化學空間,或資料準備與傳輸成本主導總耗時,實際收益可能不同,應以專案資料測量。

結論

NVIDIA ALCHEMI 所呈現的是以 AI 擴充材料研發決策的方案路徑:生成與篩選候選、以 MLIP 加速幾何鬆弛、以 DFT 和實驗建立驗證閉環。採購或導入決策應優先確認模型對目標材料的適用性、容器與 API 的正式文件、GPU 部署條件,以及在自有候選集上的準確度與端到端吞吐量,而非僅以來源中的特定基準加速比作為結論。

圍繞「NVIDIA ALCHEMI:AI 驅動材料發現的工作流程與評估路徑」繼續瞭解 NVIDIA 產品與網路方案