絡(luò)模型實(shí)戰(zhàn)指南:MPS/MPDO 與神經(jīng)網(wǎng)絡(luò)量子態(tài)全面對(duì)比)
NetKet 張量網(wǎng)絡(luò)模型實(shí)戰(zhàn)指南MPS/MPDO 與神經(jīng)網(wǎng)絡(luò)量子態(tài)全面對(duì)比【免費(fèi)下載鏈接】netketMachine learning algorithms for many-body quantum systems項(xiàng)目地址: https://gitcode.com/gh_mirrors/ne/netketNetKet 是一個(gè)專為量子多體系統(tǒng)設(shè)計(jì)的機(jī)器學(xué)習(xí)庫(kù)它把張量網(wǎng)絡(luò)模型如 MPS、MPDO與神經(jīng)網(wǎng)絡(luò)量子態(tài)統(tǒng)一在同一個(gè) JAX 生態(tài)中讓你可以用幾行代碼完成變分蒙特卡洛求解。本文將圍繞「NetKet 張量網(wǎng)絡(luò)模型」這一核心關(guān)鍵詞帶你對(duì)比 MPS/MPDO 與神經(jīng)網(wǎng)絡(luò)量子態(tài)的差異并給出可落地的實(shí)戰(zhàn)選擇建議新手也能快速上手。什么是 NetKet量子多體系統(tǒng)的機(jī)器學(xué)習(xí)利器NetKet 是面向量子多體物理的機(jī)器學(xué)習(xí)工具箱其核心理念是用可微分的參數(shù)化量子態(tài)逼近基態(tài)、熱態(tài)或穩(wěn)態(tài)。它天然支持 JAX 自動(dòng)微分、GPU/TPU 加速與分布式計(jì)算并且把 Hilbert 空間、哈密頓量、采樣器、變分驅(qū)動(dòng)器和各類 Ansatz 模型包括張量網(wǎng)絡(luò)模型無(wú)縫整合。對(duì)新手來(lái)說(shuō)NetKet 最友好的地方在于你不需要手寫(xiě)復(fù)雜的梯度公式只需定義模型與算子交給 VMC 驅(qū)動(dòng)即可自動(dòng)迭代優(yōu)化。張量網(wǎng)絡(luò)模型核心MPS 與 MPDO 是什么在 NetKet 中張量網(wǎng)絡(luò)模型集中在netket/models/tensor_networks/目錄下共提供四類實(shí)現(xiàn)全部基于 Flax 模塊構(gòu)建可直接與MCState或MCMixedState配合使用。MPS矩陣乘積態(tài)一維量子態(tài)的經(jīng)典選擇MPS 是處理一維強(qiáng)關(guān)聯(lián)量子系統(tǒng)最經(jīng)典的表示。NetKet 提供兩種邊界條件實(shí)現(xiàn)MPSPeriodic周期邊界波函數(shù)形式為矩陣乘積的跡適合環(huán)形系統(tǒng)MPSOpen開(kāi)放邊界首尾張量為向量適合鏈狀系統(tǒng)核心超參數(shù)只有兩個(gè)bond_dim鍵維決定張量連接的大小param_dtype決定參數(shù)是實(shí)數(shù)還是復(fù)數(shù)。源碼位于 mps.py內(nèi)部通過(guò)jax.lax.scan高效收縮張量鏈并支持checkpoint選項(xiàng)以節(jié)省顯存。MPDO矩陣乘積密度算符處理混合態(tài)與耗散系統(tǒng)MPDO 是 MPS 向密度矩陣的推廣采用純化purification表示適合描述開(kāi)放量子系統(tǒng)的混合態(tài)與穩(wěn)態(tài)。同樣分為MPDOPeriodic與MPDOOpen兩種實(shí)現(xiàn)位于 mpdo.py。相比 MPSMPDO 多了一個(gè)kraus_dimKraus 維參數(shù)它控制局域噪聲通道的維度鍵維為 D 時(shí)矩陣尺寸為 D2×D2因此參數(shù)開(kāi)銷更大但能刻畫(huà)退相干與耗散過(guò)程常配合SteadyState驅(qū)動(dòng)器和LocalLiouvillian求解 Lindblad 穩(wěn)態(tài)。NetKet 張量網(wǎng)絡(luò)模型與神經(jīng)網(wǎng)絡(luò)量子態(tài)對(duì)比一張表看懂NetKet 同時(shí)內(nèi)置大量神經(jīng)網(wǎng)絡(luò)量子態(tài)模型如 RBM、Jastrow、MLP、GCNN 等見(jiàn)netket/models/。兩類方法在物理與工程上各有側(cè)重對(duì)比如下對(duì)比維度張量網(wǎng)絡(luò)模型MPS/MPDO神經(jīng)網(wǎng)絡(luò)量子態(tài)核心機(jī)制基于鍵維控制的局部張量收縮全連接/卷積等非線性映射參數(shù)數(shù)量少隨系統(tǒng)線性增長(zhǎng)多但表達(dá)能力更靈活表達(dá)能力一維系統(tǒng)強(qiáng)高維受限高維系統(tǒng)優(yōu)勢(shì)明顯糾纏刻畫(huà)受鍵維限制鍵維越大糾纏越強(qiáng)容量大可描述體積律糾纏物理可解釋性高張量結(jié)構(gòu)對(duì)應(yīng)局域關(guān)聯(lián)較低黑盒特征明顯采樣與計(jì)算收縮開(kāi)銷隨鍵維快速上升前向傳播并行高效混合態(tài)支持MPDO 天然支持需 NDM 等專用結(jié)構(gòu)參數(shù)效率對(duì)比張量網(wǎng)絡(luò)模型更省MPS 的參數(shù)量約為 O(L·d·D2)其中 L 為系統(tǒng)尺寸、d 為局域自由度、D 為鍵維。在鍵維較小時(shí)參數(shù)量遠(yuǎn)低于同等容量的神經(jīng)網(wǎng)絡(luò)。這意味著在小系統(tǒng)上NetKet 張量網(wǎng)絡(luò)模型收斂更快、更不易過(guò)擬合。表達(dá)能力對(duì)比神經(jīng)網(wǎng)絡(luò)更靈活神經(jīng)網(wǎng)絡(luò)量子態(tài)通過(guò)非線性激活與深層結(jié)構(gòu)可以逼近更復(fù)雜的關(guān)聯(lián)模式尤其適合二維、三維或帶阻挫的系統(tǒng)而 MPS 天然擅長(zhǎng)一維系統(tǒng)高維時(shí)若直接應(yīng)用需要特殊的收縮策略。應(yīng)用場(chǎng)景對(duì)比各司其職一維基態(tài)/激發(fā)態(tài)優(yōu)先選 MPS物理直覺(jué)清晰、數(shù)值穩(wěn)定開(kāi)放系統(tǒng)穩(wěn)態(tài)與耗散動(dòng)力學(xué)選 MPDO配合LocalLiouvillian二維及以上大尺度系統(tǒng)選 RBM、GCNN 等神經(jīng)網(wǎng)絡(luò)量子態(tài)需要嚴(yán)格可解釋的張量結(jié)構(gòu)選張量網(wǎng)絡(luò)模型如何選擇NetKet 張量網(wǎng)絡(luò)模型實(shí)戰(zhàn)決策指南給你一份可落地的選擇清單系統(tǒng)維度是 1 維嗎是 → 用MPSPeriodic周期或MPSOpen開(kāi)鏈問(wèn)題是純態(tài)還是混合態(tài)混合態(tài) → 用 MPDO MCMixedState鍵維多大合適從 D2 或 D4 起步逐步增大并觀察能量收斂需要復(fù)參數(shù)嗎含磁通或非厄米場(chǎng)景設(shè)為param_dtypecomplex顯存緊張保持checkpointTrue并用symperiod利用平移對(duì)稱性減少參數(shù)量快速上手在 NetKet 中創(chuàng)建 MPS 模型的實(shí)戰(zhàn)步驟以下代碼演示了一維 Ising 基態(tài)求解的完整流程完整示例可參考test/models/test_mps.pyimport netket as nk g nk.graph.Hypercube(length6, n_dim1, pbcTrue) hi nk.hilbert.Spin(s0.5, Ng.n_nodes) ma nk.models.tensor_networks.MPSPeriodic(hilberthi, bond_dim2) sa nk.sampler.MetropolisLocal(hilberthi, n_chains16) vs nk.vqs.MCState(sa, ma) ha nk.operator.IsingJax(hi, graphg, h1.0) op nk.optimizer.Sgd(learning_rate0.05) nk.driver.VMC(ha, op, variational_statevs).run(100)三步走即可定義 Hilbert 空間與圖 → 構(gòu)建 MPS 模型并包裝進(jìn)MCState→ 用 VMC 驅(qū)動(dòng)迭代。MPDO 的用法類似只需把MCState換成MCMixedState并搭配nk.operator.LocalLiouvillian與nk.SteadyState??偨Y(jié)NetKet 張量網(wǎng)絡(luò)模型MPS/MPDO以少參數(shù)、高可解釋性和數(shù)值穩(wěn)健著稱是一維與開(kāi)放量子系統(tǒng)的首選 Ansatz神經(jīng)網(wǎng)絡(luò)量子態(tài)則在復(fù)雜高維系統(tǒng)中表現(xiàn)更優(yōu)。兩者并非對(duì)立而是互補(bǔ)——理解它們的差異你就能在實(shí)戰(zhàn)中快速選出最合適的工具。建議從 test_mpdo.py 與 test_mps.py 等測(cè)試用例出發(fā)動(dòng)手跑通第一個(gè) NetKet 張量網(wǎng)絡(luò)模型實(shí)驗(yàn)?!久赓M(fèi)下載鏈接】netketMachine learning algorithms for many-body quantum systems項(xiàng)目地址: https://gitcode.com/gh_mirrors/ne/netket創(chuàng)作聲明:本文部分內(nèi)容由AI輔助生成(AIGC),僅供參考