波函数演化交给Transformer:概率归一写进结构,长时预测比Crank–Nicolson快22倍
一维无限深势阱中的波函数演化改由Transformer直接预测,概率归一不再靠损失项惩罚,而是作为硬约束写进网络输出。此前无约束的多层感知机预测总概率落在1.039418到1.045530之间,偏离1;硬约束模型在基态、第一激发态、高斯波包和叠加态四类初值上把最大概率偏差压到2.98×10⁻⁷,平均保真度0.999949。在相同空间分辨率下,单次前向推理耗时约0.0006至0.001秒,预测195步时相对顺序推进的Crank–Nicolson取得22.35倍加速。