週期域上三次非線性薛定諤方程的運算元學習
研究團隊提出了一種幾何條件化的傅立葉神經運算元（FNO），用於求解二維平環面上不同縱橫比下的三次非線性薛定諤方程。該運算元將解的實部和虛部以及縱橫比引數作為輸入，學習一步解運算元。實驗表明，該運算元能捕捉有理環面和無理環面上的不同動力學行為，並重現了Sobolev範數的差異。消融研究顯示，包含縱橫比引數能提高長期預測精度。
近日，arXiv上發表了一項關於運算元學習在週期域上三次非線性薛定諤方程應用的研究。該研究由Emmanuel E. Oguadimma等人完成，提出了一種幾何條件化的傅立葉神經運算元（FNO），專門用於求解二維平環面（flat tori）上具有不同縱橫比的立方散焦非線性薛定諤方程。在二維環面上，方程的解行為強烈依賴於環面的縱橫比，因為縱橫比決定了Fourier共振結構。有理縱橫比和無理縱橫比會導致不同的高頻能量級聯行為，這一現象在以往的文獻中已有記載。為了捕捉這種依賴性，研究者設計了一個幾何條件化的FNO，其輸入包括解的實部和虛部以及縱橫比引數ω²。該網路被訓練來近似一步求解運算元，即從當前時間步的解預測下一個時間步的解。訓練資料透過Fourier偽譜方法生成，使用隨機相位的初始條件。訓練後的模型在未見過的軌跡上進行評估。實驗結果顯示，該學習運算元能夠很好地捕捉兩種環面上的主要解動力學，並重現了Sobolev範數的不同行為：在有理環面上H²範數增長更強，而在無理環面上增長受到更多約束，這與以往文獻中的發現一致。研究者還進行了詳細的消融研究，考察了保留Fourier模式數量、啟用函式、Fourier層深度以及顯式幾何條件化等因素的影響。結果表明，包含ω²引數能顯著提高長期預測的準確性，尤其是在有理幾何情況下。這支援了使用幾何感知神經運算元來學習非線性色散偏微分方程中譜轉移現象的觀點。該研究為神經運算元方法在偏微分方程應用中的幾何依賴性問題提供了重要見解，透過將幾何資訊顯式編碼到網路中，模型能夠適應不同的物理場景，展現出更強的泛化能力。未來的工作可能涉及將該方法擴充套件到更復雜的幾何或更高維度的系統。