implementation of https://arxiv.org/abs/1807.02011 input -> untrained model -> output input -> trained model -> output heat map