Code for "Multi-Time Attention Networks for Irregularly Sampled Time Series", ICLR 2021.

Last update: Dec 17, 2022

Overview

Multi-Time Attention Networks (mTANs)

This repository contains the PyTorch implementation for the paper Multi-Time Attention Networks for Irregularly Sampled Time Series by Satya Narayan Shukla and Benjamin M. Marlin. This work has been accepted at the International Conference on Learning Representations, 2021.

Requirements

The code requires Python 3.7 or later. The file requirements.txt contains the full list of required Python modules.

pip3 install -r requirements.txt

Training and Evaluation

Interpolation Task on Toy Dataset

python3 tan_interpolation.py --niters 5000 --lr 0.0001 --batch-size 128 --rec-hidden 32 --latent-dim 1 --length 20 --enc mtan_rnn --dec mtan_rnn --n 1000  --gen-hidden 50 --save 1 --k-iwae 5 --std 0.01 --norm --learn-emb --kl --seed 0 --num-ref-points 20 --dataset toy

Interpolation Task on PhysioNet Dataset

python3 tan_interpolation.py --niters 500 --lr 0.001 --batch-size 32 --rec-hidden 64 --latent-dim 16 --quantization 0.016  --enc mtan_rnn --dec mtan_rnn --n 8000  --gen-hidden 50 --save 1 --k-iwae 5 --std 0.01 --norm --learn-emb --kl --seed 0 --num-ref-points 64 --dataset physionet --sample-tp 0.9

Classification Task on PhysioNet Dataset (mTAND-Full)

python3 tan_classification.py --alpha 100 --niters 300 --lr 0.0001 --batch-size 50 --rec-hidden 256 --gen-hidden 50 --latent-dim 20 --enc mtan_rnn --dec mtan_rnn --n 8000 --quantization 0.016 --save 1 --classif --norm --kl --learn-emb --k-iwae 1 --dataset physionet

Classification Task on PhysioNet Dataset (mTAND-Enc)

python3 tanenc_classification.py --niters 200 --lr 0.0001 --batch-size 128 --rec-hidden 128 --enc mtan_enc --n 8000 --quantization 0.016 --save 1 --classif --num-heads 1 --learn-emb --dataset physionet --seed 0

Classification Task on MIMIC-III Dataset (mTAND-Full)

python3 tan_classification.py --alpha 5 --niters 300 --lr 0.0001 --batch-size 128 --rec-hidden 256 --gen-hidden 50 --latent-dim 128 --enc mtan_rnn --dec mtan_rnn   --save 1 --classif --norm --learn-emb --k-iwae 1 --dataset mimiciii

For MIMIC-III Dataset, first you need to have an access to the dataset which can be requested here. We follow the data extraction process described here: https://github.com/mlds-lab/interp-net.

Classification Task on MIMIC-III Dataset (mTAND-Enc)

python3 tanenc_classification.py --niters 200 --lr 0.0001 --batch-size 256 --rec-hidden 256 --enc mtan_enc  --quantization 0.016 --save 1 --classif --num-heads 1 --learn-emb --dataset mimiciii --seed 0

Classification Task on Human Activity Dataset (mTAND-Enc)

python3 tanenc_classification.py --niters 1000 --lr 0.001 --batch-size 256 --rec-hidden 512 --enc mtan_enc_activity  --quantization 0.016 --save 1 --classif --num-heads 1 --learn-emb --dataset activity --seed 0 --classify-pertp

Interpolation Results

Interpolation performance on PhysioNet with varying percent of observed time points:

Classification Results

Classification performance on PhysioNet, MIMIC-III and Human activity dataset, and time per epoch in minutes for all the methods on PhysioNet.

Reference

@inproceedings{
shukla2021multitime,
title={Multi-Time Attention Networks for Irregularly Sampled Time Series},
author={Satya Narayan Shukla and Benjamin Marlin},
booktitle={International Conference on Learning Representations},
year={2021},
url={https://openreview.net/forum?id=4c0J6lwQ4_}
}

Code for "Multi-Time Attention Networks for Irregularly Sampled Time Series", ICLR 2021.

Related tags

Overview

Multi-Time Attention Networks (mTANs)

Requirements

Training and Evaluation

Interpolation Results

Classification Results

Reference

Owner

The Laboratory for Robust and Efficient Machine Learning

Global-Local Attention for Emotion Recognition

Code for our TKDE paper "Understanding WeChat User Preferences and “Wow” Diffusion"

Adabelief-Optimizer - Repository for NeurIPS 2020 Spotlight "AdaBelief Optimizer: Adapting stepsizes by the belief in observed gradients"

Implementation of popular bandit algorithms in batch environments.

根据midi文件演奏“风物之诗琴”的脚本 "Windsong Lyre" auto play

Collapse by Conditioning: Training Class-conditional GANs with Limited Data

RoMa: A lightweight library to deal with 3D rotations in PyTorch.

Pytorch implementation of the paper: "A Unified Framework for Separating Superimposed Images", in CVPR 2020.

Deep Ensemble Learning with Jet-Like architecture

Label-Free Model Evaluation with Semi-Structured Dataset Representations

Resources for the "Evaluating the Factual Consistency of Abstractive Text Summarization" paper

[NeurIPS 2021] “Improving Contrastive Learning on Imbalanced Data via Open-World Sampling”,

Self-supervised Multi-modal Hybrid Fusion Network for Brain Tumor Segmentation

Help you understand Manual and w/ Clutch point while driving.

STEM: An approach to Multi-source Domain Adaptation with Guarantees

Voice control for Garry's Mod

This is a virtual picture dragging application. Users may virtually slide photos across the screen. The distance between the index and middle fingers determines the movement. Smaller distances indicate click and motion, whereas bigger distances indicate only hand movement.

Fewshot-face-translation-GAN - Generative adversarial networks integrating modules from FUNIT and SPADE for face-swapping.

PyTorch implementation of a Real-ESRGAN model trained on custom dataset

Teaching end to end workflow of deep learning