Multi-Kernel Decomposed LSTM Overview
- The paper shows that replacing standard LSTM gates with multiple kernel transformations captures multi-scale motion patterns effectively.
- It leverages both convolutional branching with flow-based attention and tensor-decomposition (KCP) to reduce parameter count and computational load.
- Empirical results on benchmarks like UCF-101 and UCF11 demonstrate enhanced accuracy and significant compression compared to traditional LSTM architectures.
Searching arXiv for the cited papers to ground the article. Multi-Kernel Decomposed LSTM denotes a class of LSTM variants in which the standard monolithic recurrent transformation is replaced by multiple kernels, tensor-decomposed factors, or both. In the literature represented here, the term spans at least two technically distinct constructions. One line augments ConvLSTM with parallel convolutional kernels of different spatial sizes, optionally followed by a mixing layer and a flow-based attention mechanism, yielding a “Network-in-LSTM” organization for video modeling (Agethen et al., 2019). A second line compresses conventional LSTM weight matrices by expressing them in Kronecker–CP form as a sum of Kronecker terms with internal low CP rank, together with fast contraction algorithms for the resulting tensorized gates (Wang et al., 2020). A related theoretical line derives LSTM-like gated memory cells and their -gram, multi-kernel generalizations from recurrent kernel machines, thereby placing multi-kernel sequence models within an RKHS-based framework (Liang et al., 2019).
1. Conceptual scope and terminology
The defining departure from the standard LSTM is the replacement of a single transition operator by a structured collection of operators. In the convolutional setting, each gate no longer uses a single convolution or ; instead, it uses a set of convolutions indexed by kernel size , with channel allocations satisfying (Agethen et al., 2019). In the compression setting, the dense weight matrix is reshaped into a tensor and represented by a rank- Kronecker–Tensor factorization whose factors are further expressed in small CP form, producing what the paper calls the KCP format (Wang et al., 2020).
These constructions share a common structural motif: the recurrent update is decomposed into multiple sub-transformations whose outputs are later concatenated, mixed, or summed. This suggests a useful umbrella interpretation of “multi-kernel decomposed” LSTM as a family of architectures that exchange a single recurrent operator for a structured composition of several lower-level operators.
The term kernel is not uniform across the relevant literature. In "Deep Multi-Kernel Convolutional LSTM Networks and an Attention-Based Mechanism for Videos" (Agethen et al., 2019), it denotes convolutional kernels such as 0 and 1. In "Kernel-Based Approaches for Sequence Modeling: Connections to Neural Methods" (Liang et al., 2019), it denotes RKHS kernels 2. The term decomposition is likewise used in different senses: tensor decomposition for weight compression in KCP-LSTM (Wang et al., 2020), and multi-branch convolutional factorization inside the gate transition in ConvLSTM-MK (Agethen et al., 2019).
2. Multi-kernel ConvLSTM gate construction
In the standard ConvLSTM recalled by Agethen and Hsu, each gate is parameterized by a single convolution: 3 The multi-kernel modification replaces each single-kernel convolution by a set of convolutions of different spatial sizes (Agethen et al., 2019). For each gate 4,
5
6
where 7 and 8. An optional 9 convolution then mixes these channels: 0 The gates become
1
2
followed by the usual cell and hidden-state updates (Agethen et al., 2019).
The architecture is explicitly described as resembling a Network-in-LSTM approach. Operationally, there are parallel branches for each kernel size 3, each branch having its own 4 and 5. The branch outputs may be combined either by interleaving or by a 6 “mix” convolution before gate-wise splitting. In practice, the paper uses 7 with 8 (Agethen et al., 2019).
A central design claim is that the single-kernel ConvLSTM imposes “an unavoidable trade-off between effectiveness and efficiency,” whereas multiple kernels permit simultaneous accommodation of fine-scale and coarse-scale motion patterns (Agethen et al., 2019). The paper’s critical discussion states this directly as multi-scale motion modeling: small kernels capture fine, slow motion; large kernels capture larger displacements.
3. Flow-based attention and branch specialization
The same work introduces an attention-based mechanism designed specifically for the multi-kernel extension. Let 9 denote optical-flow feature maps at time 0. For each kernel size 1, the model learns a mask convolution 2, producing
3
This mask is applied pixelwise to every channel of the input: 4 The resulting masked input enters the kernel-specific branch through
5
after which the outputs are concatenated across kernels (Agethen et al., 2019).
The stated goal is to let each kernel specialize on regions of the input with appropriate motion speed. During back-propagation, the masks 6 specialize so that large kernels “see” large-displacement (fast) regions and small kernels “see” slow-motion regions (Agethen et al., 2019). This is a concrete architectural claim in that paper, not merely an interpretation.
The attention mechanism modulates only the input-to-hidden transition. The paper identifies hidden-to-hidden attention as a future direction, stating that the current attention only modulates input-to-hidden and that the next step is to extend to 7 branches (Agethen et al., 2019). It also notes that optical flow must be computed or predicted, adding overhead and potential error, which makes the attention mechanism technically meaningful but not free.
A common misconception is to treat the multi-kernel attention mechanism as a generic self-attention substitute. The reported mechanism is narrower: it is a set of learned mask convolutions on optical-flow feature maps that gates the input differently for each convolutional kernel size (Agethen et al., 2019).
4. Kronecker–CP decomposition of LSTM weights
A different notion of decomposition appears in the KCP-LSTM formulation. Let 8 be one of the two-dimensional input-to-gate or hidden-to-gate weight matrices in an LSTM. Assuming
9
the matrix is reshaped to a 0-mode tensor
1
A rank-2 Kronecker–Tensor factorization writes
3
where 4 and 5, with 6 the mode-wise Kronecker product (Wang et al., 2020).
Each factor is then expressed in small CP form: 7
8
with 9 and 0. Theorem 3.1 then shows that 1 admits a single “sparse-partitioned” CP: 2 where
3
and each 4 has size 5, with
6
This is the KCP format (Wang et al., 2020).
In an LSTM, one typically concatenates all four gates into a single large matrix 7, so that the same KCP factorization is shared by 8 gates, with optional further weight-sharing across gates (Wang et al., 2020). This is the precise point at which multi-kernel and decomposed structures meet in this line of work: the model is multi-kernel in the sense of multiple Kronecker terms 9, and decomposed in the tensor-factorization sense.
For gate computation, if 0, 1, and 2, then for each gate 3,
4
The usual LSTM nonlinearities then apply: 5 (Wang et al., 2020).
5. Fast multiplication and complexity regimes
A naïve CP contraction for the KCP-formatted tensor is stated to have complexity
6
which the paper characterizes as infeasible (Wang et al., 2020). Two fast multiplication algorithms are therefore introduced.
Algorithm 1, “Strict Fast Multiplication,” precomputes
7
forms each 8 by concatenating 9, and successively reshapes and contracts the running tensor with 0, alternating left- and right-multiplications to keep intermediate sizes small. The stated complexity is
1
(Theorem 4.3) (Wang et al., 2020).
Algorithm 2, “Relaxed Fast Multiplication,” does not build the full big 2. Instead, at each mode 3 and for each 4, it applies 5 and sums over 6 whenever an even number of modes has been processed, as justified by Theorem 4.2. Its stated complexity is
7
which is always less than or equal to the cost of Algorithm 1 (Wang et al., 2020).
The paper also provides a theoretical comparison against TT, BT, TR, and HT. Under 8, 9, and 0 denoting TT/BT/TR/HT rank, the summarized asymptotics are as follows.
| Format | Param 1 | FLOPs per gate |
|---|---|---|
| Ori. | 2 | 3 |
| TT | 4 | 5 |
| BT | 6 | 7 |
| TR | 8 | 9 |
| HT | 0 | 1 |
| KCP | 2 | 3 |
The paper further states that space is proportional to 4, hence linear in 5, and that the FLOPs are comparable to TT, while TT yields far poorer compression (Wang et al., 2020). It also reports that, as 6 grows, both the parameter count and FLOPs of KCP remain lower than TT, BT, TR, and HT for typical choices of 7 or 8.
6. Empirical results, theoretical lineage, and interpretive boundaries
The empirical profile of the multi-kernel ConvLSTM is reported on UCF-101 and Sports-1M, with additional I3D-based experiments (Agethen et al., 2019). On UCF-101 Split 1 with frozen VGG-16 conv5_3 features 9, the baselines are single-kernel ConvLSTM 00 top-1 and 01. A simple multi-kernel variant with 02 reaches 03, and MK plus stacked 04 mix reaches 05. On Sports-1M with 20 random sports classes, sampled at 06 fps with 07 frames per clip, the baselines are 08 for 09 and 10 for 11, while MK 12 reaches 13. In I3D-based experiments on a 17-class subset of UCF-101, the RGB-only I3D baseline is 14, ConvLSTM 15 gives 16, ConvLSTM 17 gives 18, and Inception-like MK gives 19. On full UCF-101 with end-to-end fine-tuning, ConvLSTM 20 gives 21, 22 gives 23, simple MK gives 24, MK with flow-attention gives 25, Inception MK gives 26, and end-to-end fine-tuning gives 27 (Agethen et al., 2019).
The ablation summary is correspondingly precise: mixing 28 and 29 always beats either alone by approximately 30–31; the 32 mixing layer adds approximately 33 on UCF-101/VGG; flow-based attention adds approximately 34; and end-to-end fine-tuning with MK and 35 bottlenecks yields a large 36 (Agethen et al., 2019). The same source reports that MK adds approximately 37–38 more time per iteration than the larger single-kernel baseline, and that MK 39 requires approximately 40M weights versus 41M for 42 alone and 43M for 44 alone.
For KCP-LSTM, experiments are reported on UCF11, Youtube Celebrities Face, and UCF50 (Wang et al., 2020). The comparison includes Original LSTM, TT-LSTM, BT-LSTM, TR-LSTM, HT-LSTM, and KCP-LSTM with rank settings 45 and 46 on UCF11/Youtube, and 47, 48, 49 on UCF50, all with and without four-gate weight-sharing. The key empirical trends are that all tensor formats achieve near-identical final Top-1 and average accuracies within 50–51, while KCP routinely reaches higher compression ratios with fewer FLOPs. Specifically, on UCF11 without sharing, KCP reaches up to 52k53 compression with 54 MFLOPs versus TT 55, BT 56, TR 57, and HT 58; with weight-sharing, KCP reaches up to 59k60 versus BT 61k62, TR 63k64, and HT 65k66; on UCF50, KCP reaches up to 67k68 without sharing and 69k70 with sharing (Wang et al., 2020). Figure 1 is summarized as showing nearly identical convergence across methods, and Figure 2 as showing that raising the KT-rank 71 is more important for expressivity than boosting 72 and 73 individually.
The KCP paper also emphasizes parallel potential. Because KCP is a sum over 74 of independent Kronecker blocks 75, Algorithm 2 can be split across 76 threads or processes. In a simple CPU-based timing test, inference time under a multi-process implementation decreases roughly linearly in 77, up to manageability overhead (Wang et al., 2020).
The theoretical lineage supplied by the kernel-machine formulation is useful for clarifying what is and is not being generalized. Starting from a Mercer kernel
78
the recurrent kernel-machine derivation yields a memory-cell recursion and, after dynamic gating, an LSTM-like model
79
with
80
(Liang et al., 2019). Extending this to 81-gram filters replaces rank-1 filters by length-82 convolutions: 83 with separate filter banks for the cell candidate and each gate. The paper states that CNNs, Gated CNNs, CIFG-LSTMs, and RANs emerge as special cases under specific gate settings (Liang et al., 2019). This suggests that multi-kernel LSTM variants can be interpreted not only as engineering modifications, but also as members of a broader family of gated convolutional sequence models.
Taken together, these results delimit two major interpretations of Multi-Kernel Decomposed LSTM. One interpretation is architectural and spatial: multiple convolutional kernels are embedded inside each recurrent gate, optionally modulated by flow-based attention (Agethen et al., 2019). The other is algebraic and compressive: the gate matrices are decomposed into multiple Kronecker factors with internal CP structure and fast contractions (Wang et al., 2020). The kernel-machine account provides a unifying theoretical backdrop in which dynamic gating and multi-kernel convolution arise from a common sequence-modeling formalism (Liang et al., 2019).