Three referees indicate accept, one indicates that the paper is marginally below threshold. I agree with reviewers 1, 2 and 4 that the presented approach is insightful and useful to NeurIPS applications, and support an accept after reading the rebuttal. However, when revising the paper, please take into account reviewers' concerns about improving quantitative comparisons with other similar methods as well as providing further discussion. Please consider adding experimental support with more complex data to the the main paper or Supplementary Materials.