IJCAI 2020poster0 citations

Deep Hurdle Networks for Zero-Inflated Multi-Target Regression: Application to Multiple Species Abundance Estimation

Shufeng Kong, Junwen Bai, Jae Hee Lee, Di Chen, Andrew Allyn, Michelle Stuart, Malin Pinsky, Katherine Mills

Abstract

A key problem in computational sustainability is to understand the distribution of species across landscapes over time. This question gives rise to challenging large-scale prediction problems since (i) hundreds of species have to be simultaneously modeled and (ii) the survey data are usually inflated with zeros due to the absence of species for a large number of sites. The problem of tackling both issues simultaneously, which we refer to as the zero-inflated multi-target regression problem, has not been addressed by previous methods in statistics and machine learning. In this paper, we propose a novel deep model for the zero-inflated multi-target regression problem. To this end, we first model the joint distribution of multiple response variables as a multivariate probit model and then couple the positive outcomes with a multivariate log-normal distribution. By penalizing the difference between the two distributions’ covariance matrices, a link between both distributions is established. The whole model is cast as an end-to-end learning framework and we provide an efficient learning algorithm for our model that can be fully implemented on GPUs. We show that our model outperforms the existing state-of-the-art baselines on two challenging real-world species distribution datasets concerning bird and fish populations.

Machine Learning Applications: Applications of Supervised LearningMachine Learning: Multi-instanceMulti-labelMulti-view learningMachine Learning: Deep LearningMachine Learning: Big dataScalability
BibTeX
@inproceedings{ijcai2020p603,
  title     = {Deep Hurdle Networks for Zero-Inflated Multi-Target Regression: Application to Multiple Species Abundance Estimation},
  author    = {Kong, Shufeng and Bai, Junwen and Lee, Jae Hee and Chen, Di and Allyn, Andrew and Stuart, Michelle and Pinsky, Malin and Mills, Katherine and Gomes, Carla},
  booktitle = {Proceedings of the Twenty-Ninth International Joint Conference on
               Artificial Intelligence, {IJCAI-20}},
  publisher = {International Joint Conferences on Artificial Intelligence Organization},
  editor    = {Christian Bessiere},
  pages     = {4375--4381},
  year      = {2020},
  month     = {7},
  note      = {Special track on AI for CompSust and Human well-being},
  doi       = {10.24963/ijcai.2020/603},
  url       = {https://doi.org/10.24963/ijcai.2020/603},
}