arXiv:2505.02095eess.IV2025-05被引 1

用U-Net快速估算脑部电场,1.5GHz下相关系数达0.978,提速超1200倍。

EMulator: Rapid Estimation of Complex-valued Electric Fields using a U-Net Architecture

  • 基于U-Net的回归模型,输入头模与天线位置,输出复数电场
  • 1.5GHz下验证集相关系数0.978,平均计算仅4.4毫秒
  • 支持实时优化多天线参数,适合神经刺激方案设计

脑刺激中的电磁方法普遍依赖对振幅、相位及天线位置等剂量学参数的优化,以控制刺激强度和靶向精度。由于难以获取生物组织内电场分布的在体测量数据,通常采用物理仿真器。但这些仿真器计算成本高、耗时长,难以支撑反复计算以实现优化。为此,我们开发了EMulator——一种基于U-Net架构的回归模型,用于快速稳健地估计复数电场。训练数据来自43个天线在14个分割人脑模型上的仿真结果。训练完成后,输入分割人脑模型与天线位置,即可输出对应电场。在包含6名受试者的验证集上,1.5GHz时电场幅度的复相关系数达0.978,平均计算时间仅4.4毫秒,比当前最先进的物理仿真器COMSOL快至少1200倍。该工作证明了从分割人头模型和天线位置出发,实现电场实时计算的可能性,使得通过随机梯度下降优化多个天线的振幅、相位和位置成为可能,因模型几乎处处可导。

原文摘要 · Abstract (English)

A common factor across electromagnetic methodologies of brain stimulation is the optimization of essential dosimetry parameters, like amplitude, phase, and location of one or more transducers, which controls the stimulation strength and targeting precision. Since obtaining in-vivo measurements for the electric field distribution inside the biological tissue is challenging, physics-based simulators are used. However, these simulators are computationally expensive and time-consuming, making repeated calculations of electric fields for optimization purposes computationally prohibitive. To overcome this issue, we developed EMulator, a U-Net architecture-based regression model, for fast and robust complex electric field estimation. We trained EMulator using electric fields generated by 43 antennas placed around 14 segmented human brain models. Once trained, EMulator uses a segmented human brain model with an antenna location as an input and outputs the corresponding electric field. A representative result of our study is that, at 1.5 GHz, on the validation dataset consisting of 6 subjects, we can estimate the electric field with the magnitude of complex correlation coefficient of 0.978. Additionally, we could calculate the electric field with a mean time of 4.4 ms. On average, this is at least x1200 faster than the time required by state-of-the-art physics-based simulator COMSOL. The significance of this work is that it shows the possibility of real-time calculation of the electric field from the segmented human head model and antenna location, making it possible to optimize the amplitude, phase, and location of several different transducers with stochastic gradient descent since our model is almost everywhere differentiable.

电场模拟U-Net脑刺激加速仿真

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