Sigmoid Tensor Decomposition for Compact Representations of Binary Data
Irina Kopnina, Sergey Matveev, Ivan Oseledets
Abstract
Nonlinear low-rank decompositions are receiving growing attention as a way to represent structured data with fewer parameters than conventional linear methods. Existing nonlinear approaches have focused mainly on matrices, which limits their ability to capture multidimensional structure. We propose a nonlinear tensor decomposition based on sigmoid nonlinearization in the Tucker and tensor-train formats. A binary tensor is represented by a low-rank real-valued score tensor, whose parameters are estimated by minimizing the binary cross-entropy loss. The resulting optimization problem is jointly nonconvex, but becomes convex with respect to one factor or score block when the remaining factors are fixed, as in the corresponding matrix case. We evaluate the method on several structured binary tensors and compare it with conventional Tucker and tensor-train decompositions across a range of common ranks. These results indicate that sigmoid nonlinearization is particularly effective in the low-rank regime, where it can provide more accurate binary reconstructions than conventional linear tensor decompositions at comparable model sizes.
Create a lesson
Related papers
LithoSAGE: Spatially Adaptive and Geometrically Equivariant Neural Operators for Computational Lithography
Ke Jin, Zhenli Xu, Zilin You
Boundary-preserving randomized neural network methods for Poisson equations on general two- and three-dimensional domains
Huifang Zhou
A stability order barrier and SSP bound for Runge--Kutta methods with non-negative coefficients: AI-assisted proofs
Sigal Gottlieb, Zachary Grant, Lior Horesh et al.
Hybrid Newton-BFGS high-index saddle dynamics for accelerating saddle-point searches
Jianyuan Yin, Lei Zhang, Shixue Zhang
Randomly shifted lattice rules for SDEs with constant diffusion: error bounds uniform in time discretization and dimension
Du Ouyang, Josef Dick, Ziyang Ye
Asymptotic-Preserving Tensor Neural Networks for 3D High-Field Kinetic Equations with Applications to Uncertainty Quantification
Shi Jin, Tian-ai Zhang