arXiv:2410.14986cs.LGcond-mat.mes-hall2024-10

用深度神经网络加速磁学模拟,速度提升至线性复杂度。

NeuralMAG: Fast and Generalizable Micromagnetic Simulation with Deep Neural Nets

  • 用U形网络分解并累加自旋交互,替代传统复杂计算。
  • 实现O(N)时间复杂度,大规模模拟速度显著提升。
  • 通用性强,可适配不同尺寸形状与材料设置。

微磁学在磁存储设计中应用广泛,其数值模拟依赖于基于朗道-利夫希茨-吉尔伯特(LLG)方程的第一性原理计算。然而,传统方法因计算速度慢而受限,尽管快速傅里叶变换(FFT)将复杂度降至O(NlogN),仍难以应对大规模模拟。本文提出NeuralMAG,一种基于深度学习的微磁模拟方法。该方法沿用LLG迭代框架,但通过U形网络(Unet)加速退磁场计算:编码器在多尺度提取自旋聚合特征并学习局部交互,解码器将各尺度交互累积以逼近全局卷积。这种分治累加策略实现O(N)时间复杂度,显著提升大规模模拟效率。与现有神经方法不同,NeuralMAG聚焦核心计算而非端到端任务拟合,具备内在泛化能力。我们仅训练一个模型,在两种微磁任务上评估其在不同样本尺寸、形状和材料参数下的表现,验证了其有效性与通用性。

原文摘要 · Abstract (English)

Micromagnetics has made significant strides, particularly due to its wide-ranging applications in magnetic storage design. Numerical simulation is a cornerstone of micromagnetics research, relying on first-principle rules to compute the dynamic evolution of micromagnetic systems based on the renowned LLG equation, named after Landau, Lifshitz, and Gilbert. However, simulations are often hindered by their slow speed. Although Fast-Fourier transformation (FFT) calculations reduce the computational complexity to O(NlogN), it remains impractical for large-scale simulations. In this paper, we introduce NeuralMAG, a deep learning approach to micromagnetic simulation. Our approach follows the LLG iterative framework but accelerates demagnetizing field computation through the employment of a U-shaped neural network (Unet). The Unet architecture comprises an encoder that extracts aggregated spins at various scales and learns the local interaction at each scale, followed by a decoder that accumulates the local interactions at different scales to approximate the global convolution. This divide-and-accumulate scheme achieves a time complexity of O(N), significantly enhancing the speed and feasibility of large-scale simulations. Unlike existing neural methods, NeuralMAG concentrates on the core computation rather than an end-to-end approximation for a specific task, making it inherently generalizable. To validate the new approach, we trained a single model and evaluated it on two micromagnetics tasks with various sample sizes, shapes, and material settings.

微磁学深度学习加速模拟神经网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。