Tensor Completion from Regular Sub-Nyquist Samples
Charilaos I. Kanatsoulis, Xiao Fu, Nicholas D. Sidiropoulos, and, Mehmet Ak\c{c}akaya

TL;DR
This paper demonstrates that regular sampling can effectively reconstruct high-dimensional tensor signals, such as images and videos, with sample complexity depending on tensor rank, and applies this to accelerate fMRI imaging.
Contribution
It introduces a novel framework for tensor signal reconstruction from regular samples, emphasizing tensor rank over bandwidth, and applies it to practical fMRI acceleration.
Findings
Reconstruction from regular samples is feasible for tensors.
Sample complexity depends on tensor rank, not bandwidth.
The method significantly accelerates fMRI sampling without loss of accuracy.
Abstract
Signal sampling and reconstruction is a fundamental engineering task at the heart of signal processing. The celebrated Shannon-Nyquist theorem guarantees perfect signal reconstruction from uniform samples, obtained at a rate twice the maximum frequency present in the signal. Unfortunately a large number of signals of interest are far from being band-limited. This motivated research on reconstruction from sub-Nyquist samples, which mainly hinges on the use of random / incoherent sampling procedures. However, uniform or regular sampling is more appealing in practice and from the system design point of view, as it is far simpler to implement, and often necessary due to system constraints. In this work, we study regular sampling and reconstruction of three- or higher-dimensional signals (tensors). We show that reconstructing a tensor signal from regular samples is feasible. Under the…
| Algorithm | RETSINA | k-t Focuss | k-t SLR | IDFT |
| NRE | 0.124 | 0.339 | 1.41 | 0.8156 |
| 0.081 | 0.286 | 0.073 | 0.7376 | |
| runtime | 12min | 25.6min (48sec/coil) | 480min (15min/coil) | 14sec |
| 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10 | |
|---|---|---|---|---|---|---|---|---|---|
| 0.16 | 0.17 | 0.18 | 0.19 | 0.20 | 0.19 | 0.19 | 0.19 | 0.21 | |
| 0.18 | 0.19 | 0.20 | 0.19 | 0.20 | 0.22 | 0.21 | 0.20 | 0.22 | |
| 0.20 | 0.20 | 0.20 | 0.22 | 0.21 | 0.21 | 0.23 | 0.22 | 0.23 |
Peer Reviews
No public reviews on file for this paper yet. If you reviewed it on a platform where reviews are public (OpenReview, ICLR, NeurIPS, ICML), you can paste yours below so the community can read it here.
Videos
No videos yet. Explain this paper in a talk, walkthrough, or lecture? Add one.
Tensor Completion from Regular Sub-Nyquist Samples
Charilaos I. Kanatsoulis, Xiao Fu,
Nicholas D. Sidiropoulos, and Mehmet Akçakaya
C.I. Kanatsoulis and M. Akçakaya are with the Department of ECE, University of Minnesota, Minneapolis, MN 55455, USA email: [email protected]; [email protected]. X. Fu is with the School of EECS, Oregon State University, Corvallis, OR 97331, USA email: [email protected]. N. D. Sidiropoulos is with the Department of ECE, University of Virginia, Charlottesville, VA 22904, USA email: [email protected].
Abstract
Signal sampling and reconstruction is a fundamental engineering task at the heart of signal processing. The celebrated Shannon-Nyquist theorem guarantees perfect signal reconstruction from uniform samples, obtained at a rate twice the maximum frequency present in the signal. Unfortunately a large number of signals of interest are far from being band-limited. This motivated research on reconstruction from sub-Nyquist samples, which mainly hinges on the use of random / incoherent sampling procedures. However, uniform or regular sampling is more appealing in practice and from the system design point of view, as it is far simpler to implement, and often necessary due to system constraints. In this work, we study regular sampling and reconstruction of three- or higher-dimensional signals (tensors). We show that reconstructing a tensor signal from regular samples is feasible. Under the proposed framework, the sample complexity is determined by the tensor rank—rather than the signal bandwidth. This result offers new perspectives for designing practical regular sampling patterns and systems for signals that are naturally tensors, e.g., images and video. For a concrete application, we show that functional magnetic resonance imaging (fMRI) acceleration is a tensor sampling problem, and design practical sampling schemes and an algorithmic framework to handle it. Numerical results show that our tensor sampling strategy accelerates the fMRI sampling process significantly without sacrificing reconstruction accuracy.
Index Terms:
sampling, reconstruction, tensor completion, MRI acceleration, functional MRI
I Introduction
Sampling and reconstruction of signals is a fundamental problem in signal processing. In the first half of the 20th century, Whittaker, Nyquist, Kotelnikov and Shannon [1, 2, 3, 4] laid the foundation of sampling theory. It guarantees perfect reconstruction of a signal from uniformly spaced samples, if sampling is performed at a rate at least twice the maximum frequency present in the signal. The Shannon-Nyquist theorem applies to both continuous and discrete signals. It capitalizes on the band-limitedness property and is the first, and one of the very few results, that allow perfect reconstruction of a signal under a uniform, or more generally, regular sampling process. The challenge is that applying Shannon-Nyquist sampling to wideband signals requires very high sampling rates—which can be quite costly in practice.
In the early 2000’s compressive sensing (CS) [5, 6, 7] emerged as an alternative, enabling reconstruction from a set of measurements, sampled or compressed below the Nyquist rate. CS works under two basic premises: the signal of interest must have a sparse representation in a known transform domain; and the sampling pattern should be ‘incoherent’. Under these assumptions, tractable algorithms are shown to recover the signal of interest. Compared to the Shannon-Nyquist sampling theorem, CS leverages signal sparsity, rather than bandlimitedness. This result is significant, since some wideband signals of practical interest, are sparse in certain domains [8, 9]. On the downside, CS entails higher reconstrunction complexity than sinc function interpolation, and relies on incoherent/random sampling – thus losing the simplicity of regular/uniform sampling. A few exceptions exist, e.g., [10, 11], but the results are probabilistic and/or usually quite restrictive in practice.
Following the ideas of CS, low-rank matrix completion (LRMC) techniques were proposed for reconstructing matrix signals from a set of samples [12, 13]. This line of research utilizes the rank of the matrix as complexity measure for sampling and has attracted significant attention, since it is related to a number of important applications such as recommender systems [14]. However, similar to CS, LRMC is based on incoherent sampling. Furthermore the reconstruction guarantees in both CS and LRMC are probabilistic, contrary to the Shannon-Nyquist theorem which deterministically guarantees signal reconstruction.
Our work is motivated by the following question. Is there a sub-Nyquist sampling mechanism that works under regular sampling for certain signals of interest? This research question is very intriguing: regular sampling is efficient, friendly to implementation and often mandatory, and sub-Nyquist sampling is desired since numerous real-world signals are far from being bandlimitted.
In this work, we offer an affirmative to the above research question for a large variety of multidimensional signals. The proposed approach guarantees recoverability under realistic conditions, both generic and deterministic. Specifically, we focus on the problem of sampling tensor signals, i.e., signals whose entries are indexed by three or more coordinates [15, 16]. Tensor signals naturally arise in a large number of areas such as machine learning and data analytics [17], signal processing and communications [18], image processing and remote sensing [19, 20, 21], medical imaging [22], genomics [23], chemometrics [24], just to name a few. Hence, considering sampling and reconstruction of tensor signals is of broad interest. The problem is challenging, since various tensor signals are neither bandlimited, sparse, nor low-rank matrices (via ‘unfolding’)—and thus existing sampling techniques are not always applicable.
The reconstruction of sampled tensor signals, known in the literature as tensor completion, has been studied in machine learning and computer vision [25, 26]. The majority of existing works [26, 27, 28, 29, 30] focus on the algorithmic aspect of tensor completion. The few that provide recovery guarantees [31, 32] are based on random sampling schemes and/or LRMC ideas, which are not tailored to the tensor specifics. The work that is closest to ours is [33], which offers reconstruction conditions when the tensor ‘fibers’ are sampled. However, the conditions are somehow restrictive, since the rank is constrained to be lower than the fiber dimension, and a variety of other interesting types of regular tensor sampling have not been considered.
Contributions: In this work, we study the task of sampling and reconstruction of signals that are tensors—or tensor sampling in short. We propose a tensor sampling framework that is flexible and easy to implement. Generic as well as deterministic theoretical conditions are derived, under which reconstruction is guaranteed. Similar to matrix completion, the sample complexity for tensor signal reconstruction is mainly affected by the canonical polyadic rank and the tensor size—instead of signal bandwidth or sparsity. Unlike CS and LRMC, the proposed approach does not require incoherent sampling. Therefore, regular, equispaced and highly structured sampling strategies can be adopted—which has a much broader spectrum of applications in practice.
Our second major contribution lies in designing accelerated acquisition schemes for functional magnetic resonance imaging (fMRI) utilizing the proposed tensor sampling principles. Note that traditional fMRI acquisition is considered an “agonizingly slow” scanning process, which strongly motivates exploring appropriate sampling techniques for acceleration. However, due to hardware limitations, random or incoherent sampling strategies are considered impractical for this task [34]. Nevertheless, the proposed tensor sampling framework fits this task very well as fMRI signals are naturally tensors. Extensive simulations using synthetically generated data show that the proposed tensor sampling schemes are promising. More importantly, experiments using real fMRI data demonstrate remarkable acceleration compared to traditional fMRI scanning approaches, without sacrificing reconstruction accuracy.
An early version of part of this work appears in conference form in Proc. ICASSP 2019 [35]. In this journal version, we consider additional sampling schemes, include deterministic reconstruction conditions along with thorough model analysis and proofs. We also design multi-slice fMRI acceleration schemes and conduct detailed experiments.
II Tensor Algebra Preliminaries
In this work we heavily use tensor algebra. To facilitate the upcoming discussion we briefly present some preliminary tensor algebra concepts. The reader is referred to [15, 16] for further details.
A third-order tensor is a three-way array indexed by with elements . It consists of three modes: columns , rows , fibers ; and three types of slabs: horizontal , vertical and frontal – see Fig. 1, 2, respectively. A rank-one tensor is the outer product of three vectors, , denoted as , where is the outer product operator. Any tensor can be realized as a sum of three way outer products (rank one tensors), i.e.
[TABLE]
The above expression is known as the polyadic decomposition (PD) of a third-order tensor. If denotes the minimum number of outer products needed to synthesize , then is called tensor rank or CP rank and the decomposition is known as canonical polyadic decomposition (CPD) or parallel factor analysis (PARAFAC) [36]. The CPD elementwise representation can be written as where are called the low rank factors of the tensor. A third-order tensor can be fully characterized by its latent factors, thus we adopt the notation \underline{\bm{X}}=\left\llbracket{\bm{A}},{\bm{B}},{\bm{C}}\right\rrbracket to represent the tensor.
A striking property of tensors is that the CPD is essentially unique under mild conditions. A generic result on the uniqueness of the CPD is as follows.
Theorem 1
[37]** Let \underline{\bm{X}}=\left\llbracket{\bm{A}},{\bm{B}},{\bm{C}}\right\rrbracket with , , and . Assume that , and are drawn from some joint absolutely continuous distribution. Also assume without loss of generality. If , then the decomposition of in terms of , and is essentially unique, almost surely.
Here, essential uniqueness means that if also satisfy , then , , and , where is a permutation matrix and is a full rank diagonal matrix such that .
As far as deterministic identifiability is concerned, we have:
Theorem 2
[38]** Let \underline{\bm{X}}=\left\llbracket{\bm{A}},{\bm{B}},{\bm{C}}\right\rrbracket with , , and . The decomposition \underline{\bm{X}}=\left\llbracket{\bm{A}},{\bm{B}},{\bm{C}}\right\rrbracket is essentially unique with CP rank if .
Here denotes the Kruskal rank of a matrix, i.e., the largest integer such that any columns of are linearly independent.
A tensor can be represented in a matrix form using the matricization operation. There are three common ways to matricize (or unfold) a third-order tensor, by stacking columns, rows, or fibers of the tensor to form a matrix. For example the following operation stacks the fibers of tensor as rows of matrix :
[TABLE]
where ‘’ is the vectorization operator. One can see that:
[TABLE]
where denotes the Khatri-Rao (column-wise Kronecker) product. The superscript denotes that the unfolding is performed on the third mode of the tensor, i.e. fibers are stacked together.
Finally we will need the mode product operation in tensor analytics. The mode product operator multiplies a matrix to a tensor in a single mode. A third order tensor has three modes (rows, columns, fibers), thus three different mode products are defined. A joint mode-1, mode-2, and mode-3 product of a third-order tensor is represented by the following notation:
[TABLE]
where “” denotes the operation that multiplies each column of with , “” denotes multiplying each row of with , and “” denotes multiplying each fiber of with . The mode product is reflected in the polyadic decomposition of the tensor, i.e., the outcome of (3) results in a tensor with polyadic decomposition:
[TABLE]
The above decomposition is essentially unique under some conditions—this point will turn out to be crucial in the upcoming discussion.
III Tensor Sampling Mechanisms
The core of this work discusses the sampling and reconstruction of third-order tensors. The main claim is fundamental: roughly speaking, any third-order tensor that does not have very high rank can always be recovered from a sufficient number of regular samples. The sampling is not constrained to follow a randomized or incoherent process. On the contrary, we focus on regular and highly structured schemes. Various regular sampling strategies are considered. They involve sampling whole slabs in different modes (slab sampling), certain fibers in a single or multiple modes (fiber sampling) and entries in a systematic manner (entry sampling). Exposition and development use third-order tensors, but all the techniques can be naturally extended to higher-order tensors in a conceptually straightforward way. Similar to the case of matrices, even if a tensor is high-rank in the strict mathematical sense, it can often be approximated using low rank, in which case it can be approximately recovered using the proposed sampling and reconstruction schemes, as we will see.
III-A General Strategy and Insight
Let us consider the following general form of tensor sampling:
[TABLE]
where is a down-sampling operator with . Our goal is to study under what conditions and sampling strategies, recovering from is possible. This is an inverse problem like in CS [5, 6, 7] and LRMC [12, 13]. However, unlike in [5, 6, 7, 12, 13, 27], we do not consider random/incoherent down-sampling operators but highly structured ones—which model a plethora of engineering applications, are easier for practical system implementation and computationally more efficient.
Our work rests upon two basic ideas. The first utilizes the uniqueness property of the CPD. Recall that every tensor admits a CPD, and the CPD is essentially unique if the CP rank is not very large (in many cases: not full-rank). The second exploits the relation between a sampled sub-tensor and the original tensor . If we sample rows, columns and fibers and form a sub-tensor , then:
[TABLE]
One key observation is that the above sub-tensor can be decomposed to a sum of rank-one terms of number equal to the rank of the original tensor. Furthermore, the latent factors share certain rows with the original latent factors. Intuitively, if is not huge, there is a good chance that the sub-tensor admits a unique CPD, and part of the information of , , and can be extracted from the sub-tensor. Hence, by judiciously sampling and constructing sub-tensors, it seems viable to recover the entire , , and , and thus reconstruct . This is the main idea.
Despite this conceptual simplicity, however, fleshing out this task is nontrivial. First, when factoring the sub-tensors, there are always permutation and scaling ambiguities—even if every sub-tensor admits unique CPD, reconstruction of the whole tensor is not guaranteed. Thus the sampling mechanisms need to be carefully designed to address this issue. Second, balancing the sampling ratio with the ability to identify the original tensor is a key consideration and needs attentive thinking and design. In the remaining section, we propose a series of sampling mechanisms that take into consideration both design challenges. The considered sampling schemes are practical and motivated by real engineering applications, particularly in the field of medical imaging.
III-B Slab sampling
First, we study the task of reconstructing a third-order tensor from slab samples, taken from two different modes. Recovering tensor signals from sampled slabs finds applications in fMRI acceleration [39, 40, 41] and image/video inpainting [42, 43]. However, there is no unified characterization for recoverability under regular sampling patterns, to our best knowledge.
Let be the original full tensor, which is not fully accessible or is subject to sampling. Instead we sample/observe a subset of slabs in one mode, e.g., horizontal slabs, and a subset of slabs in a different mode, e.g. frontal slabs, . If and , two separate sampled tensors are formed, i.e., and , which represent the subset of observable horizontal and frontal slabs of respectively. Apparently, can be written as the mode multiplication of tensor with selection matrix , i.e.
[TABLE]
and as a mode multiplication with matrix , i.e.
[TABLE]
The sampling matrices perform slab selection in a single mode of , thus (they are ‘fat’) and also have full row rank. A schematic illustration of the tensor slab sampling model is given in Fig. 3. Note that, are not constrained to be randomly drawn in our framework. On the contrary, the sampling process is allowed to be regular or highly structured, see Fig. 3. Assuming \underline{\bm{X}}=\left\llbracket{\bm{A}},{\bm{B}},{\bm{C}}\right\rrbracket, following (4), (5), the sub-tensors can be expressed in a PD form:
[TABLE]
Using (6) identifiability of from can be established:
Theorem 3
Let be the original tensor signal to recover, with CPD \underline{\bm{X}}=\left\llbracket{\bm{A}},{\bm{B}},{\bm{C}}\right\rrbracket of rank . Assume that , and are drawn from a joint absolutely continuous distribution over , and that satisfy the equations in (6). Then, recovers the ground-truth almost surely if or ; where denotes the closest power of which is greater than .
The proof is presented in Appendix A. The intuition is that if or admit a unique CPD, under Theorem 1, the factors or respectively can be identified. Then or are recovered from the other tensor, where or have been left uncompressed. Note that in slab sampling only one sub-tensor is required to admit a unique CPD. The reason is that identifying the latent factors of one sub-tensor, directly estimates two original latent factors. Then, the remaining factor can be obtained via solving a linear system of equations. Furthermore, permutation and scaling ambiguities are automatically resolved, since sample common rows of . Overall reconstruction is performed as . Deterministic conditions can also be derived, and this discussion is postponed to the following section.
The previous analysis can be easily extended to the case where slab sampling is performed in all 3 modes of the tensor.
III-C Fiber sampling
Next, we consider the reconstruction of tensor from a subset of fibers, sampled along a single mode of the tensor. Fiber sampling is also of interest to a number of applications in chemometrics [29, 44] nuclear magnetic resonance spectroscopy [45] and fMRI acceleration (see Sec. V). To make the discussion concrete, consider the scenario illustrated in Fig. 4, where fiber patterns appear in the tensor sampling scheme.
A pattern will be defined as a subset of rows and columns for which every point belongs to the pattern. In the illustrated scenario, each pattern (blue, d=1; red, d=2; green, d=3) samples fibers defined by the following subset of rows and columns: . Rearranging the order of the columns results in the model shown in Fig. 5.
In the general case, the proposed fiber sampling framework entails each pattern forming a third-order tensor, i.e., and that samples are taken from every row and column of the tensor. The latter is a necessary condition for every factorization based completion approach, since completely unobserved slabs are impossible to recover. Furthermore, each pattern is required to sample from a common row or column with at least one more, thus creating an overlapping chain between patterns. The reason is that for pairwise mutually exclusive patterns, there exists a non-trivial scaling ambiguity, which cannot be determined. Formally the necessary sampling rules, for the proposed fiber sampling framework, are expressed as:
[TABLE]
where . The rules in 7 handle a plethora of sampling schemes. Specifically, each pattern is allowed to be equispaced, regular, random etc. This shows that reconstruction from regular samples is indeed doable. The sampling in Fig. 4, 5, for example, is regular and each pattern consists of equispaced rows and deterministically spaced columns.
Following similar analysis as in slab sampling, let be the sampled subtensor, formed by pattern . Also let be the row and column selection matrices determining the pattern. Then is written as follows.
[TABLE]
Using the equation in (III-C) we can establish generic identifiability of fiber sampling as:
Theorem 4
Let be the original tensor signal, fiber sampled according to (7), with CPD \underline{\bm{X}}=\left\llbracket{\bm{A}},{\bm{B}},{\bm{C}}\right\rrbracket of rank . Assume that , and are drawn from a joint absolutely continuous distribution over , and that satisfy the equations in (III-C). Then, recovers the ground-truth almost surely if ; where denotes the closest power of which is greater than .
The proof is relegated to Appendix B. In contrast to the previous case of slab sampling, where identifiability of one sampled tensor is enough, fiber sampling requires all ’s to admit a unique CPD model—otherwise certain rows of would be impossible to identify. The claim is simple and intuitive: The number of samples required to recover a fiber sampled tensor are proportional to the rank of the tensor.
Remark 1
Theorem 4 studies general tensors where factor is not required to have full column rank, and thus can be easily handled. Fiber sampling and recovery of tensors with having full column rank, is extensively studied in [33]. Compared to our work, the sampling strategy therein has to follow rules (7b), (7c), whereas (7a) can be relaxed. On the other hand, the results of this paper are tailored to cases where the sampling process exhibits some regularity and is allowed. Note that being full column rank, which is mandatory in [33], is a quite restrictive condition and prohibitive for several applications, e.g., fMRI acceleration as we will see next.
III-D Entry sampling
So far we have discussed slab, fiber sampling of third-order tensors and provided conditions under which reconstruction is guaranteed. In this subsection, we move a step further and study the more general problem of tensor reconstruction from a subset of entries, sampled in a regular fashion along the tensor. Entry sampling is another important sampling mechanism, which along with fiber sampling will prove very useful in accelerating the fMRI scan acquisition (see Sec. V).
We are interested in cases where the sampling process can be viewed as a series of patterns. A pattern is defined, similarly to fiber sampling, as a subset of rows columns and fibers , for which every point belongs to the pattern. For example consider the scenario illustrated in Fig. 6.
The number of patterns is and . In general, the proposed framework requires samples to be taken from all rows, columns and fibers of the tensor, thus and each pattern should include rows, columns and fibers at minimum. Furthermore, the patterns need to overlap like a domino, i.e. any pattern should sample from some common rows, columns or fibers with another. Furthermore, the overlap between two patterns is required to involve 2 elements in one mode and 1 element in a different mode. For example, a pair of patterns should sample 2 common rows and 1 common column. The latter is a necessary condition resulting from the inherent permutation and scaling ambiguity of the CPD. Formally the rules of entry sampling are:
[TABLE]
where . Following similar analysis as in single mode fiber sampling, let be the sampled subtensor representation of pattern . Also let be the row, column and fiber selection matrices determining pattern . Then is written as:
[TABLE]
The model in (III-D) is identifiable, under generic conditions presented in the following theorem.
Theorem 5
Let be the original tensor signal, sampled according to (9), with CPD \underline{\bm{X}}=\left\llbracket{\bm{A}},{\bm{B}},{\bm{C}}\right\rrbracket of rank . Assume that , and are drawn from a joint absolutely continuous distribution over , and that satisfy the equations in (III-D). Then, recovers the ground-truth almost surely if ; where denotes the closest power of which is greater than .
The proof is presented in Appendix B. Similar to fiber sampling, reconstruction of a tensor from entries, sampled as described in (9), is guaranteed, if all the sub-sampled tensors formed by the emerging patterns admit a unique CPD.
IV Deterministic Identifiability and insights
The sampling mechanisms, discussed so far, can be realized as separate, yet coupled, sub-sampled versions of the original third-order tensor . Recovery of , under various sampling mechanisms, was established by applying generic identifiability results on the CPD of the sub-tensors. However, reconstruction of the original tensor is also guaranteed under purely deterministic conditions. In the case of slab sampling we have the following theorem.
Theorem 6
Let be the original tensor signal to recover, with CPD \underline{\bm{X}}=\left\llbracket{\bm{A}},{\bm{B}},{\bm{C}}\right\rrbracket of rank . Assume that satisfy the equations in (6). Then, recovers the ground-truth if and has full column rank, or if and has full column rank.
When fiber or entry sampling is employed, we have:
Theorem 7
Let be the original tensor signal, fiber or entry sampled according to (7) or (9) respectively. Also let \underline{\bm{X}}=\left\llbracket{\bm{A}},{\bm{B}},{\bm{C}}\right\rrbracket denote the rank- CPD of . Assume that have no repeated entries and satisfy the equations in (III-C), (III-D), according to the sampling mechanism. Then, recovers the ground-truth if , where for fiber sampling.
Proof of both theorems is presented in Appendix C.
The implication of Theorems 3- 7 is significant and intuitive. Recovery of is based on two basic principles: identifiability of the factors of the sub-sampled tensors and ability to reconcile for the permutation and scaling ambiguities. The first is a property of the both the signal of interest and the sampling mechanism. In particular the rank of the tensor signal, along with Kruskal or generic conditions on the factors determine the number of samples required for full tensor recovery. Note that the number and structure of samples varies according to the applied sampling mechanism. Hence, there is a clear correlation between the rank of the tensor and the number of samples needed—higher ranks require higher number of samples. The second principle is a necessary property of the sampling mechanism. Although slab sampling automatically handles permutation and scaling ambiguities, fiber and entry sampling schemes have to be carefully designed to satisfy (7) or (9) and eliminate permutation and scaling mismatches.
In a nutshell, one can learn the rows of factors from the sub-tensors, up to column permutation and scaling and resolve the mismatches using common information between the sub-tensors. Then reconstruction of the original tensor is attained as \underline{\bm{X}}=\left\llbracket{\bm{A}},{\bm{B}},{\bm{C}}\right\rrbracket.
To build some further intuition on the theoretical conditions, consider the following example. Let be the tensor with CP rank which is subject to sampling. First we sample the equispaced horizontal slabs and equispaced frontal slabs. Following Theorem 3, reconstruction of is guaranteed if we sample at least horizontal slabs and frontal slabs and vise versa. This results in sampling ratio . Next we sample fibers of the tensor in a regular fashion, similarly to Fig. 4. According to theorem 4, fibers are sufficient to recover the original tensor, which gives sampling ratio . Finally, entries are sampled in a regular fashion, as shown in Fig. 6. The total number of entries required to reconstruct the original tensor is , according to theorem 5 , which results in sampling ratio . Note that for smaller rank, e.g, the total number of samples can be significantly reduced, giving sampling ratio , , .
V Application to parallel fMRI acceleration
Interestingly, the previously described sampling mechanisms find application in accelerating fMRI scan acquisition. fMRI is used to measure brain activity associated with changes in blood oxygen levels. MRI acquisitions typically use a set of coils (sensors), that in parallel collect a series of frames focusing on different parts of the brain. In fMRI, the three-dimensional (3D) volume covering the whole brain is typically acquired using multiple two-dimensional (2D) slices. These are discrete-space signals, sampled along a particular trajectory in the k-space, which is a 2-D frequency domain (), for each brain slice. Therefore an fMRI scan, can be represented by a five-way array with coil, , slice and time (frame) dimensions.
Acquiring high spatial resolution fMRI is challenging due to time restrictions. On the one hand, the -space sampling has to follow the Shannon-Nyquist theorem to avoid artifacts, when inverse Fourier transform is used for reconstruction. On the other hand sampling at a Nyquist rate leads to prolonged scan acquisition time for each frame, which is prohibitive for high temporal resolution, required in fMRI and neuroscience research. The objective is therefore two-fold: accelerate the scanning process and capture fast brain activity changes. Since the scan acquisition time is proportional to the number of -space samples, ongoing efforts focus on sampling part of the -space ( frequencies) of each slice and/or measuring the -space of only a subset of slices. The majority of work is mainly proposed for MRI scans. Classic methods use learning and calibration type techniques [46, 47, 48], while others employ the CS framework [47, 49, 50] or LRMC [50, 51, 52, 53] to perform the reconstruction.
While MRI offers significant freedom in designing the -space sampling trajectories for each frame, fMRI acquisition is more restrictive. Specifically, fMRI is performed using a special fast imaging acquisition, called echo planar imaging, that is practically only used with equispaced sub-sampling patterns due to restrictions associated with magnetic field inhomogenities and Eddy currents [34]. In simple words, the frequencies sampled for each frame have to be equispaced and all coils need to measure the same frequencies. For example the sampling scheme illustrated in Fig. 7 is typical in fMRI and performs 3-fold acceleration.
In general, a fully sampled scan is acquired first, which is beneficial for calibration purposes. Then -fold acceleration is achieved by sampling of equispaced frequencies. The frequencies to sample for each frame can be the same for the whole procedure or can be circularly shifted as in Fig. 7. For the benefit of our method we propose to circularly shift between equispaced set of frequencies in order to capture the temporal behavior of the brain accurately.
Sampling the dimension is one way to accelerate the fMRI scanning process. Another idea that is being used is to observe the -space of only a subset of slices at each time slot. In this work we propose to combine these two ideas to further reduce scanning time. Specifically at each time instance sub-sampled -space measurements are acquired for only a subset of slices, instead of the complete set. To design a sampling mechanism that fits our tensor models and fMRI constraints, we first need to acquire a fully sampled scan for every slice. Then for each frame of equispaced frequencies is observed for of the brain slices, so that at the -th frame we have measured every frequency for every slice. This results in -fold acceleration. Fig. 8 illustrates an fMRI acceleration technique, where .
Note that the two aforementioned sampling procedures can be tricky for classic techniques. On the one hand, calibration-based techniques such as GRAPPA [46] are linear and suffer from noise amplification at high acceleration rates. On the other hand, CS and LRMC schemes have difficulties in operating with regular samples, since their success rests upon incoherent sampling.
On the contrary, the proposed tensor sampling and reconstruction framework is exactly designed to handle these highly structured and constrained sampling schemes used in fMRI acquisitions. In particular, the single-slice fMRI acceleration task, as illustrated in Fig. 7, can be cast as a tensor fiber sampling mechanism, analyzed in subsection III-C. As mentioned earlier the raw fMRI scan is originally a five-way array and thus each slice is a four-way array. Although the previous analysis could be easily extended to tensors of order higher than three, we choose to work with third-order ones. Specifically the -space is processed in a single dimension (mode) by concatenating and . The reason is that the relation between and is often hard to be captured by a multilinear tensor model. As a result one fMRI slice is modeled as a third-order tensor , where with representing the number of frequencies in space respectively, represents the total number of frames (time slots) and the number of coils. Following the analysis of subsection III-C we have the following result.
Proposition 1
Let be the a single-slice fMRI tensor with rank , modeled as previously explained. Under the assumptions of Theorem 4, -fold acceleration can be achieved if .
Similarly, the proposed multi-slice fMRI acceleration scheme, which performs joint -space and slice sampling, is cast as an entry tensor sampling procedure, introduced in III-D. To do so, the -space is considered as a single mode, as before, and we also concatenate coils and slices in one dimension. The resulting third-order tensor has the -space in the first mode, i.e, , represents the total number of frames and the third mode includes the concatenation of coils and slices, i.e., with being the number of slices and coils respectively. Following the analysis of subsection III-D we have:
Proposition 2
Let be the a multi-slice fMRI tensor with rank , modeled as previously explained. Under the assumptions of Theorem 5, -fold acceleration can be achieved if .
We should mention that tensor approaches have also been proposed in medical imaging [54, 55, 56, 57, 58]. The work in [54], for example, uses a tensor model to approach the MRI sampling and reconstruction problem in an on-line fashion. However, [54] works under different sampling schemes, which are not regular and appropriate for fMRI, and reconstruction guarantees are not discussed. Moreover, a tensor model is also used in [55], in the context of MRI denoising, which is different from the fMRI or MRI acceleration problem. Finally the works in [56, 57, 58] adopt a Tucker model [15], and also require auxiliary acquisition data to estimate the bases in non-spatial modes prior to reconstruction, which are not available in most fMRI acquisitions.
VI General Algorithmic framework for Tensor Sampling
Previously we studied sampling and reconstruction of third-order tensors under different, identifiable sampling mechanisms. In the current section, the algorithmic component of our approach is discussed. In general, reconstruction of a tensor from a subset of entries falls under the framework of tensor completion. A plethora of algorithms have been proposed, e.g. [29, 44]. The idea is to use the CPD factors, computed from the incomplete tensor, and reconstruct the original one. Popular methods approach the problem as a system of non-linear equations and handle it using descent direction approaches, such as gradient descent, alternating optimization, or the Gauss-Newton method.
Existing tensor completion works could be employed to approach the recovery task of a regularly sampled tensor. However, the unique characteristics of our models would be ignored. To put it in context, the special structure of regular sampling allows tensor completion by computing the factors of complete tensors. Note that CPD computation of a complete tensor is a considerably easier task than that of an incomplete one. Several polynomial time algebraic algorithms [59, 60] have been shown to retrieve the original factors,under certain conditions, or effectively initialize optimization approaches with significant success.
We propose a three step approach to tackle the completion task, which follows the insights of Theorems 3-7. The first step solves the CPD of the sub-sampled tensors independently, the second reconciles for permutation/scaling ambiguities and gets an initial estimate of the factors, and the third solves the coupled CPD problem. Detailed analysis follows.
VI-A Step 1: Computing the CPD of sub-tensors
First, the CPD of the sub-sampled tensors is computed. This step is guided from the requirements of each sampling mechanism. To be more precise, CPD of is computed if the reconstruction conditions require to admit an identifiable CPD. The slab sampling model, for instance, requires only or to admit unique CPD. Therefore the CPD of only one sub-tensor is needed. On the other hand when fiber or entry sampling is considered, the CPD computation of every sub-tensor is performed, following the previous identifiability analysis.
VI-B Step 2: Initializing the factors
After computing the CPD of the sub-tensors, step 2 computes an initial estimate of the factors, after resolving possible permutation and scaling mismatches. We distinguish between 2 different cases:
Case 1, slab sampling: As mentioned earlier slab sampling automatically reconciles for permutation and scaling ambiguities. Furthermore, two of the factors have been already computed from step 1 (e.g., ,\bm{B}$$\leftarrowCPD()). What remains to be obtained is the third factor (e.g., ), which is revealed by the other sub-tensor (), via solving a linear system of equations:
[TABLE]
Case 2, fiber and entry sampling: Contrary to slab sampling, the permutation and scaling ambiguity is an important issue when fiber or entry sampling is applied. To be more precise, let \underline{\bm{Y}}_{d}=\left\llbracket{\bm{A}}_{d},{\bm{B}}_{d},{\bm{C}}_{d}\right\rrbracket,~{}d\in\{1,\dots D\}, be the sub-tensors formed after fiber or entry sampling. Then:
[TABLE]
where are permutation matrices and are full rank diagonal matrices such that , . Clearly, in order to synthesize from and reconstruct , the permutation and scaling mismatch should be resolved, i.e., , for every .
To overcome this issue, the common information between sub-tensors is utilized. In simple words, (7) or (9) require (or , or ) to share some common rows, i.e., . Now let:
[TABLE]
and normalize , such that the share a common row, e.g.:
[TABLE]
where , and is the diagonal matrix of row vector . Then:
[TABLE]
and we can solve for using the Hungarian algorithm [61]. This procedure resolves the permutation mismatch between the factors, i.e., . Note that in case of fiber sampling .
To reconcile for the scaling ambiguity we require extra information coming from the factors that were not involved in permutation match, i.e., or in our example. The necessary rules (7c), (9c) enforce that there is at least one common row between or . Then the scaling mismatch can be solved by the following set of equations:
[TABLE]
and represent the common rows between and respectively. Next, an initial estimate of the factors is extracted by reading out the appropriate rows from the sub-tensor factors to synthesize i.e.,
[TABLE]
VI-C Step 3: Coupled CPD
Finally, are jointly computed as a classic tensor factorization problem with missing entries, which is equivalent to the following coupled CPD estimator:
[TABLE]
In slab sampling, are identity matrices and in fiber sampling is always the identity. There are several ways to handle the above non-convex problem. We choose to employ the tensorlab [62] toolbox, which uses a Gauss Newton approach to solve this nonlinear least squares (NLS) problem. After obtaining the estimates of and , can be reconstructed by:
.
VI-D REgular Tensor Sampling and INterpolation Algorithm (RETSINA)
As mentioned earlier the accelerated fMRI acquisition can be cast as a tensor sampling and reconstruction task. Therefore it falls under the class of problems that the previously described framework can handle. However we choose to follow a different initialization approach tailored to the specific application. We design two algorithms, one for single-slice fMRI and one for multi-slice fMRI.
Single slice acceleration: The REgular Tensor Sampling and INterpolation Algorithm (RETSINA) is presented in Algorithm 1. We follow a 3 step procedure. In step 1 (initialization), for -fold acceleration we sum every vertical slabs (frames), where the missing -space measurements are considered zeros, and obtain a tensor without missing entries. Then we compute the CPD of and get a rough estimate of factors. In step 2 (refinement), we compute the CPD of (initialized by step 1) and the CPD of with known . Finally, in step 3, we compute the final factors by solving (15) with tensorlab’s Gauss-Newton algorithm. Compared to the previously presented general framework, RETSINA empirically yields enhanced reconstruction accuracy and reduces the operational time.
Multi slice acceleration: The Multi-Slice RETSINA (MS-RETSINA) is presented in Algorithm 2. Compared to RETSINA the initialization step has been modified to this specific case and the refinement step is skipped, since it has been observed that it does not improve the overall performance.
VII Simulations
In this section, we showcase the effectiveness of the proposed tensor sampling framework using numerical experiments. The experiments involve synthetically generated data as well as fMRI scans in the -space. All simulations are performed in Matlab on a Linux server with 3.6GHz cores and 32GB RAM, except part C which is performed on a Linux server with 2GHz cores and 128GB RAM.
VII-A Synthetic Experiments
First synthetically generated experiments are conducted to examine the validity of our claims and the performance of the proposed framework. In particular, a tensor is generated as \underline{\bm{X}}=\left\llbracket{\bm{A}},{\bm{B}},{\bm{C}}\right\rrbracket. The elements of the factor matrices are drawn from an independent identically distributed (i.i.d.) zero mean, unit variance Gaussian distribution. We regularly sample according to the previously presented sampling mechanisms, i.e,. slab sampling, fiber sampling, and entry sampling, as shown in Figs. 3- 6. Specifically, for slab sampling we sample equispaced frontal and horizontal slabs. Regarding fiber sampling, each tensor is a set of fibers, defined by equispaced rows and columns of . Note that one vertical slab is fully observed to reconcile for permutation and scaling ambiguities. Equivalently entry sampling is designed to observe different sets of equispaced entries plus a fully sampled vertical slab.
For the experiments, we define the sampling ratio, , and vary it from to . We also alter the tensor rank from to . To evaluate the performance of tensor reconstruction, we measure the normalized reconstruction error, i.e.
[TABLE]
Fig. 9 presents the results for the three sampling schemes.
As expected the reconstruction accuracy is deteriorating as the rank increases or the sampling ratio decreases. For reasonably small ranks and high number of samples the reconstruction is perfect.
VII-B Accelerated parallel fMRI
Next, the tensor sampling and reconstruction framework is tested in a real and important problem, that of parallel fMRI acceleration. First, we test the performance of the proposed RETSINA with fMRI scans, fully sampled in the -space, obtained from the Center for Magnetic Resonance Research (CMRR) at the University of Minnesota.
The single slice raw scan is originally a fourth-order tensor of size and we unfold it as a third-order tensor . We apply 3-fold acceleration by observing of the frequencies, as shown in Fig. 7. We choose and run step 1, 2, 3 for 50, 2 and 5 iterations respectively. The baseline algorithms used for comparison are k-t Focuss [50], which is a CS type algorithm, k-t SLR [52] which combines ideas from both LRMC and CS, and the zero padding inverse discrete Fourier transform (IDFT). The performance of IDFT is an indicator on how difficult the reconstruction is. Note that k-t SLR directly reconstructs the fMRI signal in the absolute -time-coil space. To be more precise it reconstructs signal , where denotes the inverse Fourier transform from the to the space and is the absolute value. Thus we also measure the NRE of signal , denoted as , for fair comparisons. For both k-t Focuss and k-t SLR the publicly available code was used. Note that k-t Focuss and k-t SLR are single coil algorithms in their original implementation, thus we treated each coil separately. k-t SLR requires parameter tuning and so we used a validation step to tune effectively.
The results are presented in Table I, which includes the NRE in -space and absolute space as well as runtime. The proposed RETSINA achieves highest reconstruction quality in the -space and works comparably well (but markedly faster) with k-t SLR in reconstructing the signal magnitude in the space. This is expected, since RETSINA reconstructs both the magnitude and phase (which is very important in images) in the -space, whereas k-t SLR reconstructs the magnitude in the space. In terms of runtime RETSINA works faster than k-t SLR and k-t Focuss, but slower than IDFT. However IDFT exhibits very poor reconstruction performance. It is worth noting, that k-t Focuss and k-t SLR qualify for parallel implementation, which could potentially speed up operation time. This would be held, however, at the cost of computational resources.
Fig. 10 shows the reconstructed fMRI scans at different time frames produced by RETSINA along with the fully sampled data. The quality of the reconstruction is significantly high, rendering the proposed RETSINA a good alternative for fMRI acceleration.
Finally, in Fig. 11 we illustrate the reconstruction performance at a single frame for the competing algorithms. IDFT gives an illustration of the downsampled image, RETSINA works the best and k-t SLR work comparably well, although being slightly off in contrast.
VII-C Accelerated multi-slice parallel fMRI
Finally, the proposed framework is tested in the task of accelarated multi-slice fMRI acquisition. Recall that acceleration is performed at 2 levels, since at each time slot we measure the sampled -space of only a subset of slices. The multi-slice fMRI raw scan is a fifth-order tensor of size , where the number of slices is . We unfold it as a third-order tensor and observe of the frequencies and of slices, which leads to -fold acceleration, as illustrated in Fig. 8. The tensor rank used in MS-RETSINA is . Table II shows the performance of the proposed MS-RETSINA in terms of NRE for various values of and . An illustrative example of the reconstruction performance is presented in Fig. 12.
VIII Conclusion
In this paper we studied the sampling and reconstruction of tensors under various schemes. Compared to CS, LRMC, as well as other tensor works, we provide concrete conditions, deterministic and generic, under which exact reconstruction of the tensor was proven to be attainable from regular or even equispaced samples. Furthermore, we cast the fMRI acceleration task as regular tensor sampling process and provided an efficient algorithmic framework to approach the problem. Simulations with synthetic data as well as fMRI scans in the -space show the validity and effectiveness of our approach.
Appendix A Proof of theorem 3
First, we adjust lemma 1 from [19] for complex numbers and selection matrices, which is essential for the proofs.
Lemma 1
[19]** Let , where the elements of are drawn from an absolutely continuous joint distribution with respect to the Lebesgue measure in and is a row selection matrix with full row rank. Then the joint distribution of the elements in is absolutely continuous with respect to the Lebesgue measure in
It follows that are drawn from non-singular absolutely continuous distributions. Then Theorem 1 determines the conditions under which the factors of or can be identified. Let’s consider the case where is identifiable. Under the conditions of Theorem 3, \underline{\bm{Y}}_{1}=\left\llbracket{\bm{A}}_{1},{\bm{B}}_{1},{\bm{C}}_{1}\right\rrbracket is essentially unique and from (4) holds that:
[TABLE]
where is a permutation matrix and is a full rank diagonal matrix such that . Recall that admits a (possibly non-unique) PD \underline{\bm{Y}}_{2}=\left\llbracket\bm{A},{\bm{P}}_{2}\bm{B},{\bm{C}}\right\rrbracket. Matricizing and plugging in and leads to:
[TABLE]
Since , then has full column rank almost surely [63], and can be identified from (17). Therefore \hat{\underline{\bm{Y}}}=\left\llbracket{\bm{A}}_{2},{\bm{B}}_{1},{\bm{C}}_{1}\right\rrbracket reconstructs signal .
The proof shares insights with that of Theorem 2 [19]. The basic difference lies in the fact that are full row rank selection matrices instead of full rank dense matrices.
Appendix B Proof of theorems 4,5
To begin, we use Lemma 1 and observe that are drawn from non-singular absolutely continuous distributions. Note that have full row rank by construction. Then we use Theorem 1 to claim identifiability of the factors of each sub-tensor . Under the conditions of Theorems 4, 5, can be identified, which corresponds to identifying all the rows of , up to column permutation and scaling. The caveat is that the rows of the factors are subject to column permutation and scaling mismatch, since they are obtained by the CPD of independent sub-sampled tensors. For example, let \underline{\bm{Y}}_{d}=\left\llbracket{\bm{A}}_{d},{\bm{B}}_{d},{\bm{C}}_{d}\right\rrbracket and \underline{\bm{Y}}_{d^{\prime}}=\left\llbracket{\bm{A}}_{d^{\prime}},{\bm{B}}_{d^{\prime}},{\bm{C}}_{d^{\prime}}\right\rrbracket. Then, from equations (11), it becomes clear that in order to obtain from and complete , the permutation and scaling mismatch should be resolved, i.e., , for every . To do so the following lemma is being used:
Lemma 2
Assume the entries of are jointly drawn from an absolutely continuous distribution over . Then almost surely.
Proof:
The proof is similar to the proof of Corollary 1 in [63] and uses the fact that is a non-trivial analytic function of the entries of and thus almost surely. ∎
Finally, to resolve the mismatch, we utilize the rules in (7), (9). Particularly, when the original tensor is fiber sampled up to column permutation and scaling. Then, column permutation can be fixed to be the same for all s, since the entries of are not equal almost surely. In order to reconcile for scaling mismatch, (7c), guarantees that there exist at least one row of or that is identified (up to permutation and scaling) from 2 different sub-sampled tensors . This is sufficient to resolve the CPD scaling mismatch between every - couple, due to Lemma 1. The entry sampling mechanism, differs to the fiber sampling one, in the fact that can only be partially identified from each sub-sampled version . Following same principles as before, permutation and scaling mismatch on the CPD of different s is resolved by (9c), (9d) along with Lemma 1.
Appendix C Proof of Theorems 6,7
The proof is similar to that of Theorem 3, 4, 5. The main difference lies in the fact that Theorem 2 is now employed, to establish identifiability on the CPD of each sub-sampled tensor and therefore recoverability of the original tensor. Furthermore, permutation and scaling alignment is performed using the rows of the latent factors which are common among the sub-tensors. This is accomplished, since factors with repeated entries are not allowed.
The reference list from the paper itself. Each links out to its DOI / PubMed record.
- 1[1] E. T. Whittaker, “On the functions which are represented by the expansions of the interpolation-theory,” Proceedings of the Royal Society of Edinburgh , vol. 35, pp. 181–194, 1915.
- 2[2] H. Nyquist, “Certain topics in telegraph transmission theory,” Transactions of the American Institute of Electrical Engineers , vol. 47, no. 2, pp. 617–644, 1928.
- 3[3] V. A. Kotelnikov, “On the transmission capacity of the ‘ether’and of cables in electrical communications,” in Proceedings of the first All-Union Conference on the technological reconstruction of the communications sector and the development of low-current engineering. Moscow . Citeseer, 1933.
- 4[4] C. E. Shannon, “Communication in the presence of noise,” Proceedings of the IRE , vol. 37, no. 1, pp. 10–21, 1949.
- 5[5] E. J. Candès, J. Romberg, and T. Tao, “Robust uncertainty principles: Exact signal reconstruction from highly incomplete frequency information,” IEEE Transactions on information theory , vol. 52, no. 2, pp. 489–509, 2006.
- 6[6] D. L. Donoho, “Compressed sensing,” IEEE Transactions on information theory , vol. 52, no. 4, pp. 1289–1306, 2006.
- 7[7] E. J. Candès and M. B. Wakin, “An introduction to compressive sampling,” IEEE signal processing magazine , vol. 25, no. 2, pp. 21–30, 2008.
- 8[8] S. Mallat, A wavelet tour of signal processing . Elsevier, 1999.
