Papers
Topics
Authors
Recent
Search
2000 character limit reached

Multi-Kernel Decomposed LSTM Overview

Updated 7 July 2026
  • 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 1×11\times1 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 KK 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 nn-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 WxxtW_x * x_t or Whht1W_h * h_{t-1}; instead, it uses a set of convolutions indexed by kernel size kK={k1,,kn}k \in K=\{k_1,\dots,k_n\}, with channel allocations CkC_k satisfying kCk=C\sum_k C_k=C (Agethen et al., 2019). In the compression setting, the dense weight matrix WRM×NW \in \mathbb{R}^{M\times N} is reshaped into a tensor and represented by a rank-KK 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 KK0 and KK1. In "Kernel-Based Approaches for Sequence Modeling: Connections to Neural Methods" (Liang et al., 2019), it denotes RKHS kernels KK2. 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: KK3 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 KK4,

KK5

KK6

where KK7 and KK8. An optional KK9 convolution then mixes these channels: nn0 The gates become

nn1

nn2

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 nn3, each branch having its own nn4 and nn5. The branch outputs may be combined either by interleaving or by a nn6 “mix” convolution before gate-wise splitting. In practice, the paper uses nn7 with nn8 (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 nn9 denote optical-flow feature maps at time WxxtW_x * x_t0. For each kernel size WxxtW_x * x_t1, the model learns a mask convolution WxxtW_x * x_t2, producing

WxxtW_x * x_t3

This mask is applied pixelwise to every channel of the input: WxxtW_x * x_t4 The resulting masked input enters the kernel-specific branch through

WxxtW_x * x_t5

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 WxxtW_x * x_t6 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 WxxtW_x * x_t7 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 WxxtW_x * x_t8 be one of the two-dimensional input-to-gate or hidden-to-gate weight matrices in an LSTM. Assuming

WxxtW_x * x_t9

the matrix is reshaped to a Whht1W_h * h_{t-1}0-mode tensor

Whht1W_h * h_{t-1}1

A rank-Whht1W_h * h_{t-1}2 Kronecker–Tensor factorization writes

Whht1W_h * h_{t-1}3

where Whht1W_h * h_{t-1}4 and Whht1W_h * h_{t-1}5, with Whht1W_h * h_{t-1}6 the mode-wise Kronecker product (Wang et al., 2020).

Each factor is then expressed in small CP form: Whht1W_h * h_{t-1}7

Whht1W_h * h_{t-1}8

with Whht1W_h * h_{t-1}9 and kK={k1,,kn}k \in K=\{k_1,\dots,k_n\}0. Theorem 3.1 then shows that kK={k1,,kn}k \in K=\{k_1,\dots,k_n\}1 admits a single “sparse-partitioned” CP: kK={k1,,kn}k \in K=\{k_1,\dots,k_n\}2 where

kK={k1,,kn}k \in K=\{k_1,\dots,k_n\}3

and each kK={k1,,kn}k \in K=\{k_1,\dots,k_n\}4 has size kK={k1,,kn}k \in K=\{k_1,\dots,k_n\}5, with

kK={k1,,kn}k \in K=\{k_1,\dots,k_n\}6

This is the KCP format (Wang et al., 2020).

In an LSTM, one typically concatenates all four gates into a single large matrix kK={k1,,kn}k \in K=\{k_1,\dots,k_n\}7, so that the same KCP factorization is shared by kK={k1,,kn}k \in K=\{k_1,\dots,k_n\}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 kK={k1,,kn}k \in K=\{k_1,\dots,k_n\}9, and decomposed in the tensor-factorization sense.

For gate computation, if CkC_k0, CkC_k1, and CkC_k2, then for each gate CkC_k3,

CkC_k4

The usual LSTM nonlinearities then apply: CkC_k5 (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

CkC_k6

which the paper characterizes as infeasible (Wang et al., 2020). Two fast multiplication algorithms are therefore introduced.

Algorithm 1, “Strict Fast Multiplication,” precomputes

CkC_k7

forms each CkC_k8 by concatenating CkC_k9, and successively reshapes and contracts the running tensor with kCk=C\sum_k C_k=C0, alternating left- and right-multiplications to keep intermediate sizes small. The stated complexity is

kCk=C\sum_k C_k=C1

(Theorem 4.3) (Wang et al., 2020).

Algorithm 2, “Relaxed Fast Multiplication,” does not build the full big kCk=C\sum_k C_k=C2. Instead, at each mode kCk=C\sum_k C_k=C3 and for each kCk=C\sum_k C_k=C4, it applies kCk=C\sum_k C_k=C5 and sums over kCk=C\sum_k C_k=C6 whenever an even number of modes has been processed, as justified by Theorem 4.2. Its stated complexity is

kCk=C\sum_k C_k=C7

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 kCk=C\sum_k C_k=C8, kCk=C\sum_k C_k=C9, and WRM×NW \in \mathbb{R}^{M\times N}0 denoting TT/BT/TR/HT rank, the summarized asymptotics are as follows.

Format Param WRM×NW \in \mathbb{R}^{M\times N}1 FLOPs per gate
Ori. WRM×NW \in \mathbb{R}^{M\times N}2 WRM×NW \in \mathbb{R}^{M\times N}3
TT WRM×NW \in \mathbb{R}^{M\times N}4 WRM×NW \in \mathbb{R}^{M\times N}5
BT WRM×NW \in \mathbb{R}^{M\times N}6 WRM×NW \in \mathbb{R}^{M\times N}7
TR WRM×NW \in \mathbb{R}^{M\times N}8 WRM×NW \in \mathbb{R}^{M\times N}9
HT KK0 KK1
KCP KK2 KK3

The paper further states that space is proportional to KK4, hence linear in KK5, and that the FLOPs are comparable to TT, while TT yields far poorer compression (Wang et al., 2020). It also reports that, as KK6 grows, both the parameter count and FLOPs of KCP remain lower than TT, BT, TR, and HT for typical choices of KK7 or KK8.

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 KK9, the baselines are single-kernel ConvLSTM KK00 top-1 and KK01. A simple multi-kernel variant with KK02 reaches KK03, and MK plus stacked KK04 mix reaches KK05. On Sports-1M with 20 random sports classes, sampled at KK06 fps with KK07 frames per clip, the baselines are KK08 for KK09 and KK10 for KK11, while MK KK12 reaches KK13. In I3D-based experiments on a 17-class subset of UCF-101, the RGB-only I3D baseline is KK14, ConvLSTM KK15 gives KK16, ConvLSTM KK17 gives KK18, and Inception-like MK gives KK19. On full UCF-101 with end-to-end fine-tuning, ConvLSTM KK20 gives KK21, KK22 gives KK23, simple MK gives KK24, MK with flow-attention gives KK25, Inception MK gives KK26, and end-to-end fine-tuning gives KK27 (Agethen et al., 2019).

The ablation summary is correspondingly precise: mixing KK28 and KK29 always beats either alone by approximately KK30–KK31; the KK32 mixing layer adds approximately KK33 on UCF-101/VGG; flow-based attention adds approximately KK34; and end-to-end fine-tuning with MK and KK35 bottlenecks yields a large KK36 (Agethen et al., 2019). The same source reports that MK adds approximately KK37–KK38 more time per iteration than the larger single-kernel baseline, and that MK KK39 requires approximately KK40M weights versus KK41M for KK42 alone and KK43M for KK44 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 KK45 and KK46 on UCF11/Youtube, and KK47, KK48, KK49 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 KK50–KK51, while KCP routinely reaches higher compression ratios with fewer FLOPs. Specifically, on UCF11 without sharing, KCP reaches up to KK52kKK53 compression with KK54 MFLOPs versus TT KK55, BT KK56, TR KK57, and HT KK58; with weight-sharing, KCP reaches up to KK59kKK60 versus BT KK61kKK62, TR KK63kKK64, and HT KK65kKK66; on UCF50, KCP reaches up to KK67kKK68 without sharing and KK69kKK70 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 KK71 is more important for expressivity than boosting KK72 and KK73 individually.

The KCP paper also emphasizes parallel potential. Because KCP is a sum over KK74 of independent Kronecker blocks KK75, Algorithm 2 can be split across KK76 threads or processes. In a simple CPU-based timing test, inference time under a multi-process implementation decreases roughly linearly in KK77, 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

KK78

the recurrent kernel-machine derivation yields a memory-cell recursion and, after dynamic gating, an LSTM-like model

KK79

with

KK80

(Liang et al., 2019). Extending this to KK81-gram filters replaces rank-1 filters by length-KK82 convolutions: KK83 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).

Topic to Video (Beta)

No one has generated a video about this topic yet.

Whiteboard

No one has generated a whiteboard explanation for this topic yet.

Follow Topic

Get notified by email when new papers are published related to Multi-Kernel Decomposed LSTM.