可发1区的超级创新思路（python 、MATLAB实现）：基于多尺度注意力TCN-KAN与小波变换的时间序列预测模型

一、数学模型与原理

1.1 小波变换多尺度分解

输入功率序列 x(t) 经小波变换分解为近似系数 Aj​ 与细节系数 Dj​：

1.2 多尺度TCN特征提取

对每个尺度子序列 {A3​,D3​,D2​,D1​} 采用独立TCN：

 

 式中 ∗d​ 为扩张率 d=2l 的扩张卷积，Wd​ 为可学习参数。

1.3 多尺度注意力机制

设第 k 个尺度特征为 Hk​∈RT×C，注意力权重计算为：

 

融合特征：

1.4 KAN预测层

基于Kolmogorov-Arnold表示定理构建网络：

其中 ϕq,p​ 为可学习的一维函数（B样条参数化），Φq​ 为全连接层。

---

二、模型架构与代码实现

2.1 核心模块实现

（1）小波分解层（Python）

python

import pywt

class WaveletDecomp(nn.Module):
    def __init__(self, wavelet='db4', levels=3):
        super().__init__()
        self.wavelet = wavelet
        self.levels = levels

    def forward(self, x):
        # x: [B, T]
        coeffs = []
        for i in range(x.size(0)):
            c = pywt.wavedec(x[i].cpu().numpy(), self.wavelet, level=self.levels)