TTML: tensor trains for general supervised machine learning
From MaRDI portal
Publication:6393166
arXiv2203.04352MaRDI QIDQ6393166FDOQ6393166
Authors: Bart Vandereycken, Rik Voorhaar 
Publication date: 8 March 2022
Abstract: This work proposes a novel general-purpose estimator for supervised machine learning (ML) based on tensor trains (TT). The estimator uses TTs to parametrize discretized functions, which are then optimized using Riemannian gradient descent under the form of a tensor completion problem. Since this optimization is sensitive to initialization, it turns out that the use of other ML estimators for initialization is crucial. This results in a competitive, fast ML estimator with lower memory usage than many other ML estimators, like the ones used for the initialization.
Has companion code repository: https://github.com/rikvoorhaar/ttml
This page was built for publication: TTML: tensor trains for general supervised machine learning
Report a bug (only for logged in users!)Click here to report a bug for this page (MaRDI item Q6393166)
