Partial Trace Regression and Low-Rank Kraus Decomposition
Partial Trace Regression and Low-Rank Kraus Decomposition Hachem Kadri 1 Stéphane Ayache 1 Riikka Huusari 2 Alain Rakotomamonjy 3 4 Liva Ralaivola 4 Abstract The trace regression model, a direct extension of the well-studied linear regression model, al- lows one to map matrices to real-valued outputs. We here introduce an even more general model, namely the partial-trace regression model, a fam- ily of linear mappings from matrix-valued inputs to matrix-valued outputs; this model subsumes the Figure 1. Illustration of the partial trace operation. The partial trace regression model and thus the linear regres- trace operation applied to m × m-blocks of a qm × qm matrix sion model. Borrowing tools from quantum infor- gives a q × q matrix as an output. mation theory, where partial trace operators have been extensively studied, we propose a framework where tr(·) denotes the trace, B∗ is some unknown ma- for learning partial trace regression models from trix of regression coefficients, and is random noise. This data by taking advantage of the so-called low-rank model has been used beyond mere matrix regression for Kraus representation of completely positive maps. problems such as phase retrieval (Candes et al., 2013), quan- We show the relevance of our framework with tum state tomography (Gross et al., 2010), and matrix com- synthetic and real-world experiments conducted pletion (Klopp, 2011). for both i) matrix-to-matrix regression and ii) pos- ` itive semidefinite matrix completion, two tasks Given a sample S = f(Xi; yi)gi=1, where Xi is a p1 × p2 which can be formulated as partial trace regres- matrix and yi 2 R, and each (Xi; yi) is assumed to be sion problems.
[Show full text]