T-ConvFISTA algorithm

Code of the T-ConvFISTA (TC-FISTA) algorithm.

In addition, we provide a pretrained model on the Electroencephalogram signals (EEG) of the paper.

Related paper:

Tensor Convolutional Dictionary Learning with CP Low-Rank activations. P. Humbert, L. Oudre, N. Vayatis, J, Audiffren. In IEEE Transactions on Signal Processing, 2021.

Container

-- Notebooks --

The reposity contains the code of TC-FISTA and three notebooks:

1. One for synthetic data
2. One to visualize the pretrained model of the EEG application of the paper
3. One to empirically check the validity of the different proofs in the paper

-- Data --

The folder 'data' contains all nescessary files to visualize results on EEG.

One example of learnt atom with its (CP)-Low-Rank activations:

Requirements

This code runs on Python >= 3.5. Set up environment with:

pip install -r requirements.txt