Skip to content

StatMixedML/ConvTS-Mixer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ConvTSMixer, MlpTSMixer and TsT

Time Series Extension of "Patches Are All You Need"

Implementations of some prototypical time series mixers based on Conv, MLP, and ViT archs. modified for the probabilistic multivariate forecasting use case, where the emission head is currently an "independent same-family" distribution, e.g., diagonal Student-T.

In everything that follows, the inputs are typically 4-Tensors of shape [Batch, Variate-dim, Context-length, 1+Features], and during training, the subsequent prediction window values are given [B, Variate-dim, Pred-length]. The inputs are embedded via 2d-conv to obtain patch embeddings:

Screenshot 2024-10-25 at 09 29 21

ConvTSMixer

Screenshot 2024-10-25 at 09 29 55

MlpTSMixer

Screenshot 2024-10-25 at 09 30 09

TsT (ViT style)

Screenshot 2024-10-25 at 09 30 40

Output head

Screenshot 2024-10-25 at 09 31 12

About

Time Series Extension of "Patches Are All You Need"

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published