基于GPT+Adaptor的时间序列多任务统一大模型

人工智能
文中在7种时间序列任务上进行了效果对比,本文提出的时间序列统一大模型取得了超出各个任务业内SOTA模型的效果,例如下面是长周期预测任务,基于GPT2+Adaptor的统一模型取得了最优的效果。

今天跟大家聊一聊大模型时间序列预测的最新工作,来自阿里巴巴达摩院,提出了一种基于adaptor的通用时间序列分析框架,在长周期预测、短周期预测、zero-shot、few-shot、异常检测、时间序列分类、时间序列填充等7项时间序列任务上都取得了显著的效果。

图片

论文标题:One Fits All: Universal Time Series Analysis by Pretrained LM and Specially Designed Adaptors

下载地址:https://arxiv.org/pdf/2311.14782v1.pdf

1、背景

搭建时间序列预测领域的大模型的一个难点在于,比较难获取到像NLP、CV领域中那样海量充足的训练数据。为了解决这个问题,这篇文章提出以NLP或CV领域训练好的大模型为底座,结合Adaptor技术,适配到时间序列中,解决各类时间序列问题。

Adaptor在NLP、CV等领域应用很广泛,尤其是最近大模型应用中,adaptor经常被用来进行大模型的轻量级finetune。Adaptor是一个轻量级网络,通过将其插入到大模型中的一些模块中,然后fix大模型参数,只更新adaptor的参数,就可以实现轻量级的大模型finetune。

图片图片

下面,给大家介绍阿里达摩院这篇工作中,是如何利用adaptor结合预训练的NLP、CV模型搭建统一时间序列模型的。

2、整体结构

本文提出的模型基于Freeze参数的预训练语言模型,结合4种类型的adaptor实现。整体模型结构如下图所示。

图片图片

首先,对于输入时间序列,使用RevIN的方式进行归一化,即减去每个时间序列的均值,除以每个时间序列的方差。然后使用PatchTST中的方式,将时间序列通过滑动窗口切分成多个patch,生成patch embedding。处理好后的时间序列,会输入到一个NLP领域的预训练语言模型中,整个训练过程中,语言模型的原始参数固定不变,只更新模型中新增的4类adaptor参数。

3、Adaptor设计

本文提出了4种类型的adaptor,插入到大模型中的不同位置,以此实现让NLP、CV领域的大模型适配时间序列的目的。这4种adaptor分别是temporal adaptor、channel adaptor、frequency adaptor、anormaly adaptor。

Temporal Adaptor:temporal adaptor是一个时间维度的MLP网络,用来融合时间维度的信息,文中采用bottlenect的结构,先将时间维度或空间维度的高维信息映射到低维,然后再映射回高维空间,以此在提取时序关系的过程中防止过拟合的风险。

Channel Adaptor:channel adaptor的结构和temporal adaptor相似,区别在于在空间维度进行,用来提取多元序列各个变量之间的关系,也采用了bottlenect;

图片图片

Frequency Adaptor:frequency adaptor在频域进行时间序列的信息提取,这部分将时间序列映射到频域,在频域做MLP,然后再映射回时域,以此实现频域这种全局信息的提取。

Anomaly Adapter:这部分主要是实现了一种新的时间序列异常检测方法,这里利用了attention score矩阵,对于正常序列attention score矩阵呈现周期重复的特性,而异常序列则没有,因此文中使用一个高斯核作为anomaly adaptor,用attention的输出结果和其计算KL散度进行时间序列异常检测。

图片图片

此外,各个adaptor对不同数据的影响不同,因此文中使用了一个gate网络,进行有选择性的adaptor使用。

4、实验效果

文中在7种时间序列任务上进行了效果对比,本文提出的时间序列统一大模型取得了超出各个任务业内SOTA模型的效果,例如下面是长周期预测任务,基于GPT2+Adaptor的统一模型取得了最优的效果。

图片图片

责任编辑:武晓燕 来源: 圆圆的算法笔记
相关推荐

2024-04-08 12:19:19

AI数据

2022-09-28 15:34:06

机器学习语音识别Pytorch

2023-11-06 07:27:38

模型NLP领域

2011-04-15 09:14:03

抄袭巨头IT

2023-07-07 12:30:00

模型技术

2023-01-17 09:38:17

模型训练

2023-11-29 07:25:58

2024-04-10 12:42:51

AI模型

2010-02-26 17:47:07

2023-10-17 12:47:26

AI数据

2024-01-11 09:01:32

GPT商店OpenAI人工智能

2023-10-13 15:34:55

时间序列TimesNet

2013-08-13 14:39:29

多任务下载

2014-05-09 12:59:26

iOS移动互联网

2010-06-04 18:59:43

关闭MySQL

2023-06-26 07:51:48

2023-04-03 10:32:56

模型数据集

2017-05-08 14:42:36

2024-02-19 07:47:27

2023-10-14 13:09:53

谷歌模型
点赞
收藏

51CTO技术栈公众号