Source-linked AI summary

D3D: Distilled 3D Networks for Video Action Recognition

Jonathan C. Stroud, David A. Ross, Chen Sun, Jia Deng, Rahul Sukthankar

arXiv:1812.08249v2cs.CV

TL;DR

Whether action-recognition-trained spatial streams in 3D CNNs naturally learn motion representations remains uncertain, despite gains from temporal streams. D3D distills temporal-stream outputs into the spatial stream and achieves accuracy on par with two-stream approaches using one RGB-only model without optical-flow inference.

  • Problem

    It remains unclear whether spatial streams in action-recognition-trained 3D CNNs naturally learn motion representations, despite gains from adding temporal streams.

    Source: §1 · pp. 1–2
  • Method

    D3D distills the temporal stream’s outputs into the spatial stream, yielding a single RGB-only model for inference.

    Source: §1 · p. 2
  • Results

    D3D achieves accuracy on par with two-stream approaches using one stream and no optical-flow computation at inference.

    Source: §1 · p. 2, §6 · p. 9
  • Takeaways & Limitations

    D3D transfers to other action-recognition datasets without further distillation.

    Source: §6 · p. 9

    Kinetics videos may be deleted over time, shrinking the dataset and causing results to differ across experiment dates.

    Source: §5.1 · p. 5

Abstract

from arXiv · show

State-of-the-art methods for video action recognition commonly use an ensemble of two networks: the spatial stream, which takes RGB frames as input, and the temporal stream, which takes optical flow as input. In recent work, both of these streams consist of 3D Convolutional Neural Networks, which apply spatiotemporal filters to the video clip before performing classification. Conceptually, the temporal filters should allow the spatial stream to learn motion representations, making the temporal stream redundant. However, we still see significant benefits in action recognition performance by including an entirely separate temporal stream, indicating that the spatial stream is "missing" some of the signal captured by the temporal stream. In this work, we first investigate whether motion representations are indeed missing in the spatial stream of 3D CNNs. Second, we demonstrate that these motion representations can be improved by distillation, by tuning the spatial stream to predict the outputs of the temporal stream, effectively combining both models into a single stream. Finally, we show that our Distilled 3D Network (D3D) achieves performance on par with two-stream approaches, using only a single model and with no need to compute optical flow.

1. Introduction

The introduction argues that 3D CNN spatial streams may not naturally learn the motion cues their temporal filters could capture, and proposes distillation to combine spatial and temporal streams in one RGB-only model.

1. Introduction

  • Motion can distinguish actions that look similar in a frame, yet ensembling RGB and optical-flow 3D CNNs raises HMDB-51 accuracy by 6.6% [3], motivating the question of what motion the spatial stream misses.The introduction notes that temporal filters should capture movement, but separate temporal streams still bring accuracy gains.Source: §1 · pp. 1–2
  • Prior work designed a 3D CNN specifically to learn motion representations; this study finds action-recognition 3D CNNs can learn them on optical-flow prediction with minimal structural changes.The authors hypothesize temporal filters can capture motion when optimized to do so.Source: §1 · p. 2
  • Action-recognition-pretrained 3D CNNs with fixed filters underperform fully fine-tuned models on optical-flow prediction, indicating room to improve their motion representations.The comparison tests whether motion representations arise naturally during action-recognition training.Source: §1 · p. 2
  • D3D distills the temporal stream’s outputs into the spatial stream, which alone is used at inference and improves optical-flow prediction, suggesting better motion representations.The method compresses the two-stream architecture into a single model.Source: §1 · p. 2
  • D3D outperforms single-stream and strong RGB-only baselines, matches two-stream accuracy, transfers from Kinetics to HMDB-51 and UCF-101, and needs no inference-time optical flow.The paper reports competitive results on Kinetics, UCF-101, HMDB-51, and AVA.Source: §1 · p. 2

2. Related Work

Related work spans framewise 2D methods and video-level 3D CNNs, often augmented with separate motion streams or specialized motion modules. This paper positions optical flow as a source of supervision and distillation as a way to transfer temporal-stream motion representations into the spatial stream.

2D CNNs

  • Framewise 2D CNNs pool predictions over time [23] [4] [22], while Two-Stream Networks add an optical-flow temporal stream [23]; LSTMs and CRFs are alternatives, especially when video data is limited.Naïve average pooling ignores temporal dynamics, and training 3D CNNs requires large video datasets.Source: §2 · p. 2

3D CNNs

  • Two-stream I3D, S3D-G, and R(2+1)D yield absolute Kinetics improvements of 3.1%, 2.5%, and 1.1%, respectively [3] [35]; C3D with IDT improves UCF-101 accuracy by 5.2%.These benefits, alongside findings on reversed inputs and single-frame C3D, suggest 3D CNNs do not sufficiently model motion [35].Source: §2 · pp. 2–3

Why Optical Flow?

  • Optical flow may help through texture and color invariance: action recognition performance is poorly correlated with flow accuracy except near motion boundaries and small displacements [21].Other motion representations can replace optical flow, suggesting it is not necessary for action recognition [5] [38] [7] [21]; it can also provide additional supervision, as in ActionFlowNet.Source: §2 · p. 3

Incorporating Motion in 3D CNNs

  • Motion Feature Networks, Optical Flow-Guided Features, and Representation Flow add modules that explicitly compute motion [16] [27] [19], while Hidden Two-Stream and TVNet learn motion representations end-to-end [5] [38].These approaches use specialized layers such as correlations or cost volumes, so they do not establish whether vanilla 3D CNNs can learn motion representations [12] [26].Source: §2 · p. 3

Distillation

  • Distillation transfers a teacher network’s outputs to a student [10] and can transfer knowledge across tasks or modalities [20] [17].Unlike Motion Vector CNNs, which distill temporal-stream knowledge into a separate motion stream, this work distills it into the spatial stream without computing motion vectors or evaluating a temporal-stream 3D CNN at test time.Source: §2 · p. 3

3. Evaluating 3D CNN Motion Representations

The section tests whether 3D CNN features encode motion by measuring how well they can reconstruct optical flow, and evaluates both naturally learned and specifically optimized representations.

3. Evaluating 3D CNN Motion Representations

  • Two-stream gains motivate testing whether action-recognition 3D CNNs learn motion naturally; a decoder’s ability to predict optical flow from hidden features measures that capacity.The authors hypothesize that RGB-based 3D CNNs can learn motion representations, but do not learn them naturally during action-recognition training.Source: §3 · p. 3

3.1. Predicting Optical Flow

  • S3DG [35] features feed Simple, Spatial, or PWC decoders that use no temporal filters, isolating motion information already present in the features.The PWC decoder follows PWC-Net without cost volume or warping [26]; its three-channel (mag, sin θ, cos θ) output is trained against TV-L1 flow, which is used in two-stream approaches [3] [21], with angular-channel loss weighted by magnitude.Source: §3.1 · pp. 3–4

3.2. Evaluating Motion Representations

  • Motion representations are evaluated by endpoint error against estimated TV-L1 flow, with decoder-only training testing naturally learned features and end-to-end fine-tuning testing task-specific optimization.The decoder-only setting leaves the 3D CNN fixed; the second setting fine-tunes the CNN and decoder together.Source: §3.2 · p. 4, §3.1 · p. 4

4. Distilled 3D Networks

D3D distills a fixed temporal stream’s outputs into the spatial stream by jointly optimizing action classification and distillation losses. At inference, it uses only RGB input, without computing optical flow or retaining a separate temporal stream.

4. Distilled 3D Networks

  • D3D trains the spatial stream to match a fixed temporal teacher’s pre-softmax outputs while minimizing action-classification and distillation losses.The losses are cross-entropy and mean squared error, respectively; λ = 1 works well in many cases.Source: §4 · p. 4
  • At inference, D3D discards the temporal stream and optical-flow computation, relying on RGB input while achieving accuracy on par with two-stream methods.The approach adds no computational overhead to the spatial stream [19] [34] [27] [16].Source: §4 · pp. 4–5
  • D3D and its teacher use S3D-G, selected for comparable accuracy at lower computational cost than I3D and Non-local I3D [3] [34].Source: §4 · p. 5

4.1. Alternatives to Distillation

  • Flow-as-supervision optimizes optical-flow prediction and action classification jointly, but does not generally improve action recognition.The authors suggest background-dominated flow loss as a possible reason, since accurate flow correlates with action classification near motion boundaries [21]; distillation instead matches temporal-stream outputs. The approach is similar to ActionFlowNet.Source: §4.1 · p. 5
  • Learned flow passed into a temporal stream outperforms that stream using TV-L1 flow, but underperforms the spatial stream.This end-to-end alternative resembles TVNet and Hidden Two-Stream approaches [38] [5].Source: §4.1 · p. 5

5. Experiments

Across several datasets, D3D improves on single-stream baselines and reaches or exceeds two-stream performance without requiring optical-flow computation. Distillation improves motion representations, and experiments test transfer and alternative distillation designs.

5.2. Predicting Optical Flow

  • A PWC decoder can approximately estimate optical flow from S3D-G spatial features, but a simple decoder cannot recover motion signal, indicating those features do not readily expose motion.The decoder comparison brackets performance against an all-zero flow model and a decoder trained on temporal-stream activations.Source: Table 1 · p. 5, §5.2 · p. 6

5.1. Datasets

  • Kinetics results reflect dataset decay: experiments used the October 2018 video availability, with 226K of 247K Kinetics-400 training examples and 369K of 393K Kinetics-600 examples remaining.The paper marks results on the original Kinetics dataset with an asterisk; HMDB-51 and UCF-101 results use test split 1, while AVA results use the v2.1 validation set.Source: §5.1 · p. 5

5.2. Predicting Optical Flow

  • D3D features yield lower optical-flow prediction error than S3D-G across layers, especially early layers; fine-tuning either network narrows the performance gap.Both networks capture coarse motion but miss fine details; D3D estimates show slightly more accurate motion boundaries. Fine-tuning for flow also improves estimates, supporting the finding that RGB-trained 3D CNNs have limited natural motion signal, which modified objectives can enhance.Source: Figure 4 · p. 6, §5.2 · p. 6, Figure 5 · p. 6

5.3. Distillation on Kinetics

  • On Kinetics-400, D3D improves top-1 accuracy by 1.9% over S3D-G; on Kinetics-600, it improves single-model performance by 1.3%, and D3D+S3D-G outperforms two-stream S3D-G without optical flow.The student is trained by distilling from a temporal-stream teacher with the distillation loss added to the action loss; only D3D is evaluated at inference.Source: §5.3 · pp. 6–7, Table 2 · p. 7, Table 3 · p. 7

5.4. Transfer to UCF101, HMDB51

  • On UCF-101 and HMDB-51, fine-tuned D3D outperforms the reported baselines and matches Hidden Two-Stream when pretrained on Kinetics-400, at over 2x lower cost.Fine-tuning uses action loss only, without distillation, avoiding a temporal stream during fine-tuning as well as inference.Source: Table 4 · p. 7, §5.4 · pp. 7–8

5.5. Transfer to AVA

  • On AVA, D3D improves frame-mAP by 1% over S3D-G using 50 RGB frames and no optical flow, and outperforms the I3D-based approach with added JFT-pretrained ResNet features.D3D replaces I3D as the 3D CNN backbone in a Faster R-CNN-style setup with a separate pretrained ResNet region proposal network.Source: Table 5 · p. 8, §5.5 · p. 8

5.6. Ablation study

  • D3D with the proposed approach outperforms the ablation alternatives: flow-loss training slightly improves S3D-G, while intermediate-layer distillation is less effective than output distillation.Both action and distillation losses are useful, and distillation benefits appear only when the temporal stream is the teacher; flow-based alternatives use layer 3A and the PWC decoder.Source: Table 6 · p. 8, §5.6 · pp. 8–9

6. Conclusions

D3D is a single-stream distilled 3D CNN that performs on par with two-stream approaches without requiring optical flow at inference. It transfers to other action-recognition datasets without further distillation and learns motion representations better than non-distilled counterparts.

  • D3D performs on par with two-stream approaches without requiring optical flow during inference.Source: §6 · p. 9
  • D3D transfers to other action-recognition datasets without further distillation.Source: §6 · p. 9
  • 3D CNNs have limited capacity to learn motion representations, and D3D reconstructs them better than non-distilled counterparts.Source: §6 · p. 9

Appendices · A. Predicted Optical Flow Visualizations

The appendix presents optical-flow visualizations comparing S3D-G and D3D, including a finetuned D3D variant.

A. Predicted Optical Flow Visualizations

  • Figure 6 shows S3D-G and D3D optical flow produced using a PWC decoder applied at layer 3A.The figure includes D3D flow with finetuning.Source: Figure 6 · p. 11
  • The visualization rows, from top to bottom, are RGB frames, TV-L1 optical flow, S3D-G flow, D3D flow, and finetuned D3D flow.Source: Figure 6 · p. 11
  • TV-L1 optical flow is downsampled to 28 × 28 px, matching the decoder output resolution used during training.Source: Figure 6 · p. 11

B. Performance on Kinetics-400 Categories

On Kinetics-400, D3D improves accuracy over S3D-G on more classes than it degrades, while the comparison shows only the top and bottom 25 classes.

B. Performance on Kinetics-400 Categories

  • D3D improves accuracy on 203 of 400 Kinetics-400 classes (50.8%) and degrades it on 103 (27.3%), with less than ±.1% difference on the remaining classes.Figure 8 compares per-class accuracy for D3D and S3D-G and displays only the top and bottom 25 classes.Source: Figure 8 · p. 13
Loading 1812.08249v2…