Causal Future Prediction in a Minkowski Space-Time

Where Physics and Machine Learning Meet

A. Vlontzos, H.B. Rocha, D. Rueckert, B. Kainz

Setup

This repo is based of Matthieu et al. Poincare VAE git link

pip install -r -U requirements.txt

Use

To train a model run:

main.py --gpu 0 --parameters

In inference mode the our algorithm will automatically

Populate the function below with your own method of choosing the next point in the latent space. This part of the algorithm depends on the target application, see paper for details. Given a "count all the white pixels" naive method.

compare_images @ utils.py

Blog Post

Check out our blog post for a summary of the paper.

Citation

@article{vlontzos2020causal,
    title={Causal Future Prediction in a Minkowski Space-Time},
    author={Athanasios Vlontzos and Henrique Bergallo Rocha and Daniel Rueckert and Bernhard Kainz},
    year={2020},
    eprint={2008.09154},
    archivePrefix={arXiv},
    primaryClass={cs.CV}
}