一种基于能量扩散约束的重复压裂产量Transformer预测模型优化方法

本发明提供了一种基于能量扩散约束的重复压裂产量Transformer预测模型优化方法。该方法借助能量约束扩散框架与经典Transformer架构中注意力模块的天然联系，以全局能量的最小化为约束，通过重新排列状态更新方程中的状态传播项，将Transformer架构每次迭代的计算复杂度从O(N2)降低到O(N)。优化后，5000轮运算的时间耗费从8分钟降低至5分半，大幅提高了用Transformer架构筛选重复压裂目标井的工作效率。