Streaming Bayesian Deep Tensor Factorization
Shikai Fang, Zheng Wang, Zhimeng Pan, Ji Liu, Shandian Zhe
Abstract
Despite the success of existing tensor factorization methods, most of them conduct a multilinear decomposition, and rarely exploit powerful modeling frameworks, like deep neural networks, to capture a variety of complicated interactions in data. More important, for highly expressive, deep factorization, we lack an effective approach to handle streaming data, which are ubiquitous in real-world applications. To address these issues, we propose SBDT, a Streaming Bayesian Deep Tensor factorization method. We first use Bayesian neural networks (NNs) to build a deep tensor factorization model. We assign a spikeand-slab prior over each NN weight to encourage sparsity and to prevent overfitting. We then use the multivariate delta method and moment matching to approximate the posterior of the NN output and calculate the running model evidence, based on which we develop an efficient streaming posterior inference algorithm in the assumed-densityfiltering and expectation propagation framework. Our algorithm provides responsive incremental updates for the posterior of the latent factors and NN weights upon receiving newly observed tensor entries, and meanwhile identify and inhibit redundant/useless weights. We show the advantages of our approach in four real-world applications.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 47829b1d-c3b6-4530-867d-c9b458613765Cited by top-tier papers12
- Bayesian Continuous-Time Tucker DecompositionShikai Fang, Akil Narayan, Robert M. Kirby, Shandian ZheICML 2022 · 20 citations
- Streaming Factor Trajectory Learning for Temporal Tensor DecompositionShikai Fang, Xin Yu, Shibo Li, Zheng Wang et al.NeurIPS 2023 · 12 citations
- Dynamic Tensor Decomposition via Neural Diffusion-Reaction ProcessesZheng Wang, Shikai Fang, Shibo Li, Shandian ZheNeurIPS 2023 · 12 citations
- Nonparametric Factor Trajectory Learning for Dynamic Tensor DecompositionZheng Wang, Shandian ZheICML 2022 · 8 citations
- Functional Bayesian Tucker Decomposition for Continuous-indexed Tensor DataShikai Fang, Xin Yu, Zheng Wang, Shibo Li et al.ICLR 2024 · 8 citations
Related papers
- Factor Augmented Tensor-on-Tensor Neural NetworksGuanhao Zhou, Yuefeng Han, Xiufan YuAAAI 2025 · 7 citations
- Undirected Probabilistic Model for Tensor DecompositionZerui Tao, Toshihisa Tanaka, Qibin ZhaoNeurIPS 2023 · 8 citations
- Efficient Variational Inference for Sparse Deep Learning with Theoretical GuaranteeJincheng Bai, Qifan Song, Guang ChengNeurIPS 2020 · 55 citations
- Nonparametric Decomposition of Sparse TensorsConor Tillinghast, Shandian ZheICML 2021 · 10 citations
- HLRTF: Hierarchical Low-Rank Tensor Factorization for Inverse Problems in Multi-Dimensional ImagingYi-Si Luo, Xile Zhao, Deyu Meng, Tai-Xiang JiangCVPR 2022 · 45 citations
