GOCPT: Generalized Online Canonical Polyadic Tensor Factorization and Completion

Chaoqi Yang, Cheng Qian, Jimeng Sun

Proceedings of the Thirty-First International Joint Conference on Artificial Intelligence

Main Track. Pages 2348-2354.

https://doi.org/10.24963/ijcai.2022/326

PDF BibTeX Video

Abstract

Low-rank tensor factorization or completion is well-studied and applied in various online settings, such as online tensor factorization (where the temporal mode grows) and online tensor completion (where incomplete slices arrive gradually). However, in many real-world settings, tensors may have more complex evolving patterns: (i) one or more modes can grow; (ii) missing entries may be filled; (iii) existing tensor elements can change. Existing methods cannot support such complex scenarios. To fill the gap, this paper proposes a Generalized Online Canonical Polyadic (CP) Tensor factorization and completion framework (named GOCPT) for this general setting, where we maintain the CP structure of such dynamic tensors during the evolution. We show that existing online tensor factorization and completion setups can be unified under the GOCPT framework. Furthermore, we propose a variant, named GOCPTE, to deal with cases where historical tensor elements are unavailable (e.g., privacy protection), which achieves similar fitness as GOCPT but with much less computational cost. Experimental results demonstrate that our GOCPT can improve fitness by up to 2.8% on the JHU Covid data and 9.2% on a proprietary patient claim dataset over baselines. Our variant GOCPTE shows up to 1.2% and 5.5% fitness improvement on two datasets with about 20% speedup compared to the best model.

Keywords

  • Data Mining: Mining Spatial and/or Temporal Data
  • Constraint Satisfaction and Optimization: Solvers and Tools
  • Data Mining: Mining Data Streams
  • Machine Learning: Online Learning
Welcome Future Conferences AI Journal Division IJCAI Proceedings Travel Grant Program IJCAI Awards