---
title: MIMO Temporal Wavelet Transform
url: https://www.emergentmind.com/topics/multi-input-multi-output-temporal-wavelet-transform
type: topic
---

# MIMO Temporal Wavelet Transform

Searching arXiv for the cited papers to ground the article in published work.
arXiv search query: 2305.16211 OR 2506.22001 OR 2002.09905
A multi-input multi-output temporal wavelet transform is a temporal analysis–synthesis construction in which multiple input signals are decomposed into multiple temporally ordered subband outputs, typically low-pass and high-pass components across dyadic scales, and then either reconstructed or further processed for coding, prediction, or enhancement. In recent arXiv work, this concept appears in learned motion-compensated temporal filtering for video coding, where a group of pictures is mapped to a hierarchy of temporal subbands [2305.16211]; in MIMO speech enhancement, where multichannel recordings are processed with wavelet-based multi-resolution representations while preserving inter-channel spatial cues [2506.22001]; and in spatial-temporal multi-frequency video prediction, where temporal discrete wavelet transforms operate directly on frame sequences and are fused with spatial decomposition and recurrent modeling [2002.09905]. Across these formulations, the defining features are temporal multiresolution, explicit multi-branch outputs, and operator structures that expose cross-input dependencies rather than collapsing them into a single residual stream.

## 1. Conceptual scope and canonical realizations

The most explicit formulation of a temporal wavelet transform as a MIMO system is the learned motion-compensated temporal filtering framework for video coding. There, the recursion over dyadic temporal levels yields “a MIMO mapping from multiple temporal inputs (the whole GOP) to multiple subband outputs: all \(h^{(j,t)}\) and the final \(l^{(J,0)}\),” with temporal low-pass and high-pass branches produced by lifting steps implemented with learned prediction and update operators [2305.16211]. The same work extends the temporal transform to a spatio-temporal transform by applying a learned \(2\)D wavelet transform to each temporal subband image and by supporting both YUV \(4{:}2{:}0\) and RGB \(4{:}4{:}4\).

In WTFormer, the MIMO aspect is not a GOP-to-subband coder but a multichannel enhancement system whose outputs are “enhanced signals for all \(M\) input channels,” with preservation of inter-channel relationships such as ITD, IPD, ILD, and relative transfer functions [2506.22001]. Temporal wavelet processing is embedded inside a time-frequency encoder–decoder through Haar-based wavelet transform convolutions operating on STFT feature maps, so the transform is internal to feature extraction rather than a standalone reconstruction stage.

In the video prediction model based on spatial-temporal multi-frequency analysis, the paper’s implemented pipeline is “multi-input single-output (per step), with multi-output achieved by rollout,” because it predicts the next frame autoregressively from \(m\) past frames [2002.09905]. The same source, however, gives a “proposed extension” in which future temporal subbands for \(T_{\text{out}}\) steps are predicted jointly and reconstructed by inverse transforms. This suggests a broader use of the term: a MIMO temporal wavelet transform may denote either an explicit filter bank with multiple temporal outputs or a temporal decomposition stage that feeds a joint multi-output estimator.

| Realization | Input–output structure | Temporal wavelet mechanism |
|---|---|---|
| Learned MCTF video coding [2305.16211] | whole GOP \(\rightarrow\) all \(h^{(j,t)}\) and final \(l^{(J,0)}\) | motion-compensated lifting |
| WTFormer speech enhancement [2506.22001] | \(M=8\) microphones \(\rightarrow\) enhanced signals for all channels | Haar WTConv inside STFT-domain network |
| Spatial-temporal video prediction [2002.09905] | \(m\) past frames \(\rightarrow\) next frame; MIMO extension proposed | multi-level DWT on temporal dimension |

## 2. Operator-theoretic structure and perfect reconstruction

In the lifting-based video formulation, a single temporal analysis stage acts on two frames \(x_1\) and \(x_2\) under an even–odd split. The forward transform is

\[
h = x_2 - P(W(x_1; v_{2\leftarrow 1}); \theta_P),
\]

\[
l = x_1 + U(W^{-1}(h; v_{2\rightarrow 1}); \theta_U),
\]

or in simplified form,

\[
l = x_1 + U(h; \theta_U).
\]

Here \(W(\cdot; v)\) is a differentiable, bilinear warping operator parameterized by motion vectors estimated by SPyNet and entropy-coded with a hyperprior + dual spatial prior, while \(P\) and \(U\) are learned CNN predictors and updaters applied after motion compensation and denoising. The outputs \(l\) and \(h\) are temporal low-pass and high-pass subbands. The inverse transform reverses step order and signs,

\[
x_1 = l - U(h; \theta_U), \qquad x_2 = h + P(W(x_1; v_{2\leftarrow 1}); \theta_P),
\]

and achieves perfect reconstruction prior to quantization because the lifting steps are triangular with identity diagonals [2305.16211].

The same construction admits a \(2\times 2\) operator-matrix form. With \(P\circ W\) denoting warp-then-predict and \(U\circ W^{-1}\) the update branch, the analysis transform maps \([x_1; x_2]\) to \([l; h]\) through off-diagonal motion-dependent learned operators, making the two-channel filter bank explicitly MIMO rather than a separable scalar transform. For lossless compression, integer rounding of \(P\) and \(U\) outputs yields an integer-to-integer lifting stage in the sense of Calderbank–Daubechies–Sweldens [2305.16211].

A second, more classical formulation appears in the temporal DWT descriptions used for video prediction and as the underlying \(1\)-D case for WTFormer’s separable \(2\)-D wavelet convolutions. If \(a^{(0)}[n]=x[n]\), then level-\(\ell\) analysis is

\[
a^{(\ell)}[n]=\sum_k a^{(\ell-1)}[k]\; h[2n-k], \qquad
d^{(\ell)}[n]=\sum_k a^{(\ell-1)}[k]\; g[2n-k],
\]

with inverse reconstruction by upsampling and synthesis filtering,

\[
x[n]=\sum_k a^{(L)}[k]\; h'[n-2k]+\sum_{\ell=1}^{L}\sum_k d^{(\ell)}[k]\; g'_{\ell}[n-2k].
\]

In the video-prediction paper, Haar is stated as an assumption rather than a specified choice, whereas WTFormer explicitly uses Haar wavelet bases in WTConv [2002.09905; 2506.22001]. The shared filter-bank interpretation is that temporal approximation coefficients aggregate slower motion content while detail coefficients isolate faster temporal changes.

## 3. Multi-level dyadic decomposition and the MIMO mapping

The dyadic recursion in learned MCTF is indexed by level \(j \ge 1\) with stride \(d = 2^{j-1}\). Let \(F_t^{(0)} = f_t\). Then analysis at level \(j\) is defined by

\[
h_t^{(j)} = F_{t+d}^{(j-1)} - P_j(W(F_t^{(j-1)}; v_{t+d\leftarrow t}); \theta_{P,j}),
\]

\[
l_t^{(j)} = F_t^{(j-1)} + U_j(W^{-1}(h_t^{(j)}; v_{t+d\rightarrow t}); \theta_{U,j}),
\]

with \(F_t^{(j)} := l_t^{(j)}\) and recursion on the low-pass branch. For a GOP of \(8\) frames, the coding order is level \(1\), then level \(2\), then level \(3\), and finally the deepest low-pass \(l^{(3,0)}\). The system is open-loop: motion estimation is performed on original frames rather than decoded reconstructions, which avoids drift [2305.16211].

As temporal distance doubles with level, displacement magnitudes increase and prediction becomes harder. To address this, MCTF-DS downsamples frames by \(\times 2\) before motion estimation at levels \(j>1\), codes lower-resolution motion latents, and upsamples the motion vectors before motion compensation and inverse motion compensation. The paper states that this reduces motion bitrate and often improves robustness to large motions with minimal quality loss [2305.16211].

The temporal DWT used in the video-prediction model is also multilevel, but it operates directly on raw frames along the time axis. The paper states that it conducts “multi-level DWT on temporal dimension (DWT-T) on the input sequence,” continues “until the number of low-frequency sub-bands or high-frequency sub-bands equals two,” and shows “three DWT-T as an example.” The resulting temporal subbands are concatenated and passed to a small CNN, the T-WAM module, and then fused with historical information from an LSTM cell [2002.09905]. In that setting, faster motions are primarily represented in higher-frequency temporal subbands, while slower motions accumulate in lower-frequency subbands and the final approximation.

WTFormer uses a different decomposition locus: the time-domain signals from the microphone array are first transformed by STFT, and wavelet transform convolutions with a Haar basis then operate on the \(2\)-D time-frequency feature maps. The paper states that WTConv yields multi-resolution subbands \(LL\), \(LH\), \(HL\), and \(HH\), expands the receptive field, and captures both long-term temporal dependencies and local transients [2506.22001]. A plausible implication is that the MIMO property here is distributed across temporal scale, frequency bins, and channels rather than tied to a single explicit temporal filter bank over the raw signals.

## 4. Extension across space, color, and channels

In learned video coding, temporal analysis is followed by a learned \(2\)D wavelet transform, iWave++, applied in horizontal and vertical dimensions to each temporal subband image. iWave++ uses CNN-based prediction and update filters in lifting, four spatial decomposition levels, scalar quantization with a trainable parameter, and a CNN-based context entropy model with a Gaussian mixture likelihood. The result is an explainable and invertible transform with spatial scalability. For color, the coder supports YUV \(4{:}2{:}0\) and RGB \(4{:}4{:}4\); in YUV \(4{:}2{:}0\), motion vectors are computed on the luma channel and reused for chroma, and subbands for each color channel are coded independently, with no explicit cross-channel coupling in the entropy model [2305.16211].

The spatial-temporal video-prediction model uses a closely related but task-specific spatial construction. Each frame is decomposed by a \(2\)D DWT into \(LL\), \(LH\), \(HL\), and \(HH\) subbands, and the paper states that three S-WAMs are cascaded, each performing a \(2\)D DWT per frame followed by shallow CNNs per subband and residual fusion with RRDB features [2002.09905]. The stated purpose is to enrich structural information, reserve fine details, and preserve anisotropic information: \(LH\) for horizontal, \(HL\) for vertical, and \(HH\) for diagonal detail. Temporal DWT and spatial DWT are then fused in a unified encoder–decoder with CNN and LSTM components.

In WTFormer, cross-channel structure is central. The observed mixture is modeled in the STFT domain as

\[
\mathbf{Y}_{f,t}=\mathbf{S}_{f,t}+\mathbf{N}_{f,t}=H_s S_{f,t}+H_n N_{f,t}, \qquad
\mathbf{Y}_{f,t},\mathbf{S}_{f,t},\mathbf{N}_{f,t}\in\mathbb{C}^M,
\]

and the system estimates early reverberant speech by a learned MIMO complex masking filter. The architecture uses three WTBlocks in the encoder, two cascaded Conformer blocks for time–frequency dependencies, and multidimensional collaborative attention in place of U-Net skip connections. MCA computes attention over spatial, time, and frequency dimensions, averages the three attention outputs, gates them with a sigmoid, and fuses them with encoded features [2506.22001].

The output definition makes the MIMO constraint explicit: enhanced signals are generated for all \(M\) channels, so inter-channel relationships can be preserved end-to-end. The paper defines common spatial cues as

\[
\mathrm{IPD}_{ij}(t,f)=\angle X_i(t,f)-\angle X_j(t,f), \qquad
\mathrm{ILD}_{ij}(t,f)=20\log_{10}\frac{|X_i(t,f)|}{|X_j(t,f)|},
\]

with ITD estimated from time-domain cross-correlations or phase slopes across frequency [2506.22001]. This places the temporal wavelet transform within a broader MIMO framework in which preserving spatial covariance is as important as denoising.

## 5. Optimization, inference control, and learning objectives

The learned MCTF video coder is trained end-to-end with a staged curriculum. Stage \(1\) trains motion estimation and denoising on distortion only with \(d_{\max}=1\); stage \(2\) adds the motion vector rate; stage \(3\) trains all components including iWave++ on the full rate–distortion objective

\[
L_{\text{full}}=\sum_{i=0,1}\left[R_{\text{all},i}+\lambda \cdot D_{\text{MSE}}(f_i,\hat{f}_i)\right];
\]

stages \(4\) and \(5\) add multiple MCTF stages for GOP sizes \(4\) and \(8\) with \(d_{\max}=2\) and \(4\), randomly sampling frame distances per batch, training MCTF stages first and then all modules jointly. The paper reports that attempting to train inverse MCTF with four frames per batch caused instability and degraded RD, whereas two-frame training with multi-stage scheduling stabilized optimization [2305.16211].

The same work introduces content-adaptive inference, MCTF-CA, to mitigate error propagation and ghosting under strong motion and occlusion without retraining. For each \(8\)-frame unit, the system evaluates five options—GOP8 with standard MCTF, GOP8 with downsampling strategy, split into two GOP4 with and without downsampling, and split into four GOP2—and chooses the minimum of

\[
C_8=\sum_{i=0}^{7}\left[R_{\text{all},i}+\lambda \cdot D_{\text{MSE}}(f_i,\hat{f}_i)\right].
\]

The choice is signaled with \(3\) bits per \(8\)-frame unit [2305.16211].

WTFormer uses a different optimization principle: a multi-task loss with learnable uncertainty weights combining denoising and spatial preservation,

\[
\mathcal{L}_{\text{total}}=\frac{10}{2\sigma_1^2}\,\mathcal{L}_{\text{ns}}+\frac{1}{2\sigma_1^2}\,\mathcal{L}_{\text{ps}}+\log(\sigma_1\sigma_2),
\]

where \(\mathcal{L}_{\text{ns}}\) is SI-SNR-based noise-suppression loss and \(\mathcal{L}_{\text{ps}}\) is a MUSIC-based spatial spectrum loss [2506.22001]. For a ULA with spacing \(d=4\) cm and speed of sound \(c=343\) m/s, the MUSIC pseudo-spectrum is

\[
P(\theta)=\frac{1}{\mathbf{a}(\theta)^H \mathbf{E}_n \mathbf{E}_n^H \mathbf{a}(\theta)},
\]

and the spatial loss is the mean-squared error between input and enhanced spectra averaged over \(300\) narrowband frequencies and \(181\) scanned angles. The implementation uses LibriSpeech train-360 split \(90\%/5\%/5\%\), Adam with learning rate \(4\times10^{-4}\) halved on plateau, batch size \(16\), \(80\) epochs, and automatic mixed precision [2506.22001].

In the spatial-temporal video-prediction model, the implemented objective combines image-domain loss and adversarial loss,

\[
\mathcal{L}_{\text{img}}=\mathcal{L}_2(Y,\hat{Y})+\mathcal{L}_{\text{gdl}}(Y,\hat{Y}),
\]

\[
\mathcal{L}_D^A=-\log D([X,Y])-\log(1-D([X,\hat{Y}])), \qquad
\mathcal{L}_G^A=-\log D([X,\hat{Y}]),
\]

\[
\mathcal{L}_G=\lambda_1 \mathcal{L}_{\text{img}}+\lambda_2 \mathcal{L}_G^A,
\]

with \(\lambda_1=1\) and \(\lambda_2\) set to \(0.01\) for KTH and \(0.001\) for BAIR and KITTI/Caltech [2002.09905]. The same source also gives a proposed MIMO extension in which future temporal subbands are predicted jointly and supervised by pixel-domain, perceptual, temporal-consistency, and wavelet-domain losses. Because this part is explicitly presented as a proposed extension, it should be interpreted as a framework-level generalization rather than the paper’s base implementation.

## 6. Empirical behavior, scalability, and limitations

The learned MCTF framework provides intrinsic temporal scalability because the dyadic temporal hierarchy yields a base layer in the low-pass subbands and enhancement layers in the high-pass subbands; spatial DWT adds spatial scalability. On UVG, MCTF-CA achieves average Bjøntegaard-Delta rate savings of \(-21\%\) for GOP4 and \(-9\%\) for GOP8 over HM LD-P; on HEVC Class E, \(-26\%\) for GOP4 and \(-11\%\) for GOP8; and on MCL-JCV, \(-12.6\%\) for GOP4 and near parity at GOP8 with \(-0.41\%\). The paper states that at high rates, MCTF-CA consistently surpasses HM and state-of-the-art learned coders such as DCVC-HEM, while also noting that DCVC-HEM performs better at low rates on some sets [2305.16211]. Complexity remains substantial: for \(1080\)p inputs, DCVC is \(32\) MB and \(1167\) kMAC/px, DCVC-HEM is \(70\) MB and \(1673\) kMAC/px, and MCTF-CA is \(90\) MB and \(3554\) kMAC/px. Most complexity in MCTF-CA is attributed to temporal subband coding by iWave++, with MCTF modules accounting for approximately \(29\%\) of model size and approximately \(12\%\) of kMAC/px for GOP8 [2305.16211].

WTFormer reports PESQ \(3.02\), STOI \(0.92\), eSTOI \(0.84\), and SI-SNR \(10.31\) dB, together with spatial-retention metrics \(\Delta\)ITD \(84.27\,\mu\)s, \(\Delta\)IPD \(0.75\) rad, and \(\Delta\)ILD \(0.73\) dB [2506.22001]. The paper states that these results improve spatial metrics over MIMO-UNet and EaBNet at a much smaller model size of approximately \(0.98\)M parameters. Ablation results attribute notable PESQ gains and ITD retention to WTConv, identify MCA as critical for both PESQ and ITD, and show that removing the MUSIC-based spatial loss degrades ILD and ITD [2506.22001]. The stated limitations are practical: Conformer and MCA introduce latency, causal or streaming variants may be needed for real time, MUSIC-based loss is array-geometry dependent, and highly non-stationary noise may challenge covariance stability [2506.22001].

The spatial-temporal multi-frequency video-prediction model reports, on KTH \(10\rightarrow 20\), PSNR \(29.85\), SSIM \(0.893\), and LPIPS \(11.81\); on KTH \(10\rightarrow 40\), PSNR \(27.56\), SSIM \(0.851\), and LPIPS \(14.13\); on BAIR, PSNR \(21.02\), SSIM \(0.844\), and LPIPS \(9.36\); and FVD \(72.3\) on KTH and \(159.6\) on BAIR [2002.09905]. The paper states that removing S-WAM or T-WAM degrades performance and removing both degrades the most, supporting the claimed role of multi-frequency analysis in fidelity and temporal consistency. At the same time, its native temporal wavelet design remains autoregressive at inference for multi-frame outputs, so the full MIMO interpretation depends on the proposed extension rather than the base predictor [2002.09905].

Taken together, these systems delineate three closely related meanings of multi-input multi-output temporal wavelet transform. In video coding, it is an explicit invertible temporal filter bank with multiple subband outputs and perfect reconstruction before quantization. In MIMO speech enhancement, it is a multiresolution transform embedded in a channel-preserving network whose outputs remain multichannel and spatially coherent. In video prediction, it is a multi-frequency temporal decomposition that can support MIMO forecasting when future subbands are predicted jointly. The common thread is not a single standardized architecture, but a class of multiscale temporal operators that retain structure across inputs instead of reducing temporal interaction to a single residual or single-channel output.

Source: https://www.emergentmind.com/topics/multi-input-multi-output-temporal-wavelet-transform