O BJECT DYNAMICS DISTILLATION FOR SCENE DECOMPOSITION AND REPRESENTATION
Anonymous authors
Paper under double-blind review
A BSTRACT
The ability to perceive scenes in terms of abstract entities is crucial for us to achieve higher-level intelligence.
Recently, several methods have been proposed to learn object-centric representations of scenes with multiple objects, but most of them focus on static images.
In this paper, we work on object dynamics and propose Object Dynamics Distillation Network (ODDN), a framework that distills explicit object dynamic representations (e.g., velocity) from raw video input.
On this basis, we further build a relation module that calculates object-pair interactions and applies it to the corresponding dynamic representations of objects.
We verify our approach on tasks of video events reasoning and video prediction, which are two important evaluations for video understanding.
The results show that visual representations of ODDN perform better in answering reasoning questions around physical events in a video compared to representaions of the previous scene representation methods.
And, ODDN could generate reasonable future frames given two input frames, considering occlusion and objects collision.
In addition, using the object dynamic clues allows the model to obtain better scene decomposition quality in terms of segmentation and reconstruction.
INTRODUCTION
[Figure] Humans learn to decompose the scene into independent objects to understand the environment at a very early age.
This cognitive pattern promotes the generation of common sense of physics, for example, the speed of the object will change after a collision.
However, the neural networks seem to lack the ability to acquire a compositional understanding of the world in terms of structured objects, which is crucial for generalizing and higher-level tasks such as planning and reasoning(Greff et al., 2020).
To tackle this problem, many scene decomposition approaches have been proposed to segment an image with ground-truth segmentation labels as supervision(Ronneberger et al., 2015; Long et al., 2015; Badrinarayanan et al., 2017; Chen et al., 2017). However, the acquisition of semantic segmentation labels is costly.
Besides, these methods do not provide structured representations of segmented areas.
Recently, several unsupervised methods have become popular in learning decomposition and object-centric representations of scenes with multiple objects(Hsieh et al., 2018; Van Steenkiste et al., 2018; Burgess et al., 2019;
Locatello et al., 2020; Greff et al., 2017; Eslami et al., 2016; Kosiorek et al., 2018), and some of those methods learn disentangled features which represent crucial properties (e.g., ’color’, ’size’) of object in the same format.
The work of (van Steenkiste et al., 2019) has proved that disentangled representations do in fact lead to better downstream task performance.
Beyond static scenes (images), people understand dynamic physical events and predict how the future might unfold(Battaglia et al., 2013).
Even infants have the intuition that temporarily occluded objects remain coherent wholes and follow spatially contiguous paths(Baillargeon, 1987). This com- mon sense of physics is inseparable from the understanding of object dynamics.
However, existing scene decomposition and representation works focus on images, which means that when interpreting videos, those approaches process one frame at a time, and do not encode objects’ dynamic properties.
We argue that representing object dynamics is signiﬁcant for tasks as video reasoning or predicting the future.
In this work, we aim to distill and disentangle object dynamics in an unsupervised manner and present ODDN (Object Dynamics Distillation Network).
ODDN is built on a spatial mixtures framework(Greff et al., 2019), a prime example of VAE(Kingma & Welling, 2013) based architecture of decomposing scenes into object-centric latent representations.
Formally, the model clusters and decodes pixels into latent representations through an iterative amortized variational inference procedure then reconstructs the latent vectors into input images.
In this way, scenes are segmented and represented as object-centric disentangled features.
Based ob this, our key insight is that the object dynamics hide in variations between objects’ static latent representations of different frames, thus, we design our model to distill object dynamics from their representations of the current and previous frames.
With the distilled object dynamics as part of the representation, we further build a relation module to model object interactions.
We evaluate ODDN on different downstream tasks as shown in Figure 1.
Results show that representations with object dynamics perform better in reasoning task than representations with static properties only, and the relation module endows ODDN with the capability of predicting the future.
In addition, we notice that incorporating object dynamics and relations into the basic scene decomposition framework beneﬁts the segmentation and reconstruction quality. We highlight contributions of our approach:
• Distillation of disentangled object dynamics from raw video input in an unsupervised manner. • State-of-the-art performance on CLEVRER(Yi et al., 2019) which focus on video understanding and causal reasoning. • Capability of future frames prediction in 3D scenes including physical events (occlusion, collision). Better scene decomposition quality for spatial mixture architecture in terms of segmentation and reconstruction.
RELATED WORK
Object representation of static scenes.
Representation learning is an important topic in deep learning.
Several recent lines of work making breakthroughs in this area learns representations without explicit supervision.
NEM(Greff et al., 2017) builds a spatial mixture model based on EM algorithm to discover a compositional object representation but fails to cope with colored or 3D data.
AIR(Eslami et al., 2016) achieves scene interpretation by using a recurrent neural network that attends to one element at a time and disentangles positions and appearances explicitly in latent variables.
AIR can decompose 3D scenes, however, it generates partial images separately and adds them together to form the ﬁnal image, which leads to its restriction: they can’t be applied to scenes with occluded objects.
MoNet(Burgess et al., 2019) uses multiple attention steps to tackle complex 3D scene, which is the ﬁrst being capable of discovering object masks and completing partially occluded objects.
IODINE(Greff et al., 2019) does the same thing as MoNet by using iterative variational inference to reﬁne the inferred latent representation in each encoding step.
Object representation of dynamic scenes.
Methods mentioned above are designed to decompose static scenes, hence they do not encode object dynamics in the latent space.
R-NEM(Van Steenkiste et al., 2018) applies RNN to NEM to encode object dynamics implicitly, and models relations between objects to learn physical interaction which enables the model to predict future frames for video input.
Similarly, SQAIR(Kosiorek et al., 2018) extends AIR to image sequence to discover and track objects throughout the video frames. Both R-NEM and SQAIR bear the drawbacks of their predecessors.
ViMON(Weis et al., 2020) is a video-extension of MoNet by building temporal connections between latent representations with GRU(Cho et al., 2014).
ViMON learns a linear transformation to predict next time-step image, encouraging dynamics encoded.
PROVIDE(Zablotskaia et al., 2020)
Inferencemodel
Match&Distill
Generativemodel
Embedding 𝑒𝑚𝑏 !
" attention effect attention effect 𝑒𝑓𝑓 !"
𝑒𝑓𝑓 #" Σ 𝑒𝑓𝑓 $%& 𝑧 ’"()’
DynamicUpdate 𝑧 ’"()’‘
Embedding 𝑒𝑚𝑏 #" 𝑧 ’"()’ 𝑧 ’"$+, 𝑧 ’#()’ 𝑧 ’#$+, 𝑧 ’!
()’ 𝑧 ’!
$+, 𝑧 ’"()’ 𝑧 ’"$+, 𝑧 ’"$+,
LinearPredicter 𝑧 ’-" " $+,
Inferred [Figure]
Inferred
[Figure]
Distilled Dynamics
Predicted
[Figure]
applies 2D-LSTM(Graves et al., 2007) to IODINE and is capable of simulating future video frames. Although ViMON and PROVIDE encode object dynamics, they do not model interactions between objects, thus, they can’t handle datasets with object collisions.
Besides, object dynamics of both methods are spread in latent space implicitly, which lacks interpretability. It has been proved that disentangled representations allow one to learn models with better performance and fewer samples for downstream reasoning tasks.
Therefore, in this work, we aim to represent object dynamics explicitly. Instead of using RNNs, we match objects’ latent representations between the current and previous frames and distillate their dynamics by applying a single-layer transformer encoder.
METHOD
Our goal is to decompose dynamic scenes into object-centric representations purely based on visual observations.
And the representation contains interpretable object dynamics. Recent works decompose static scenes by randomly assigning objects to a ﬁxed number of slots, which share weights to obtain latent vectors with a common format.
However, these works do not encode object dynamics for objects of sequential input. Our key idea is that building upon these successful multi-object scenes decomposing and representation approaches, object dynamics could be distilled from the corresponding latent representations of two consecutive frames.
3. O BJECT DYNAMICS DISTILLATION NETWORK
We illustrate the framework of ODDN in Figure 2.
Formally, ODDN takes as input a video, then the inference model processes all frames x 1: N independently to obtain K object latent vectors which encode object static properties.
For frames x 2: N , we distill objects’ dynamics from their latent vectors of the current and the previous frames.
The relation module takes the object latent vectors and the object dynamics of frames x 2: N − 1 as input, calculates the interaction between objects and predicts object latent representations of frames x 3: N .
The input image x ∈ R D could be reconstructed by K latent vectors z k ∈ R M , each of which corresponds to one object or background area after training, and each latent vector is also called a slot.
Firstly, latent z k is decoded into pixel-wise mean µ ik and pixel-wise mask logit ˆ m ik .
Then, the K mask logits are normalized with a softmax function to ensure that the sum of K mask logits for each pixel is 1.
At last, µ and m parameterize the ﬁnal spatial Gaussian mixture distribution:
[Equation]
where σ is the same and ﬁxed for all i and k . The input image can be reconstructed as ˆ x = (cid:80)
Kk =1 m k µ k .
The inference model of ODDN follows previous work (Greff et al., 2019) by using amortized variational inference to infer and reﬁne posterior
q λ ( z | x ) iteratively where λ = { µ z , σ z } .
The main reason to use such an iterative process is that the standard variational inference framework leads to the multimodal problem, which means that any pixel could be capdtured by any slot.
IODINE updates posterior λ as follows:
[Equation]
where the posterior parameters λ start with an arbitrary guess, and the ﬁrst latent representations zare sampled from q λ .
In each reﬁnement step, input x together with a set of variables a , which are computed from z t (means µ , masks m , mean gradient ∇ µ L , mask gradient ∇ m L ) are fed into a reﬁnement network fφ to obtain the additive updates for the posterior.
ODDN predicts future objects state in latent space given ﬁrst two frames:
[Equation]
where z dyn is the distilled object dynamic representation and z sta refers to the object latent representation output by the inference model.
The next-frame latent vectors are predicted with a learnable transformation
W pred , details will be introduced in section 3.2 and section 3.3.
The whole network is trained end-to-end and the ﬁnal loss consists of prediction loss L p and recon- struction loss L r , in detail:
[Equation]
where T is the maximum reﬁnement step number during the inference stage. We measure and optimize the KL divergence between the variational posterior and the prior in L r .
3. DYNAMIC DISTILLATION
In this section, we describe how ODDN derive object dynamics from sequential object latent representations.
For existing scene decomposition and representation approaches, object-to-slot assignments usually switch unpredictably in a video.
Thus, we draw support from the Transformer to match object latent representations in different frames correctly and further distill object dynamics.
Revisit attention mechanism of Transformer.
The key component of the Transformer is the attention module which could be mathematically described as:
[Equation]
where q ∈ R d ; K consists of | K | elements ( k 1 , k 2 , · · · , k
| K | ) with k i ∈ R d ; V consists of | V | elements
This attention function maps the input query q and a set of key ( k ) -value
( v ) pairs to an updated representation, W q , W k , W v are learnable parameters.
Besides, α i ≥ 0 , (cid:80) | v | i =1 α i = 1 , and usually | K | =
| V | .
Model architecture.
The inference model decomposes the input video into N ∗ K latent vectors, where K is the slot number for each frame and N is the number of frames of the video.
We put these sequence consisting latent vectors { z 11 , · · · , z 1 K , · · · , z N 1 , · · · , z NK } into one single Transformer encoder layer to match the corresponding slots which usually switch their assignments with objects several times in a video.
After ﬁnding object correspondence in different frames, the Transformer aggregates latent information and distills object dynamics which are encoded into a low-dimensional representation space.
Mathematically, the dynamic distillation process for the k-th slot in the n-th frame could be described as:
[Equation]
where z dyn is a low-dimensional vector transformed by the fully connected feed-forward network (FFN).
This learning process is performed in parallel by pre-deﬁning an attention mask to ensure that latent representations of the current frame only attend to ones one frame before.
3. RELATION MODULE.
ODDN models interactions between object pairs by computing all object-pair effects and leting them act on and update object dynamics. ODDN learns a soft attention coefﬁcient for each effect.
The object dynamics are updated as:
[Equation]
where W emb ﬁrst embeds representations of object-pair, then W att learns the attention coefﬁcient based on the object-pair embedding, and W eff learns the effect z k j has on z k i .
At last, the summed effects output an update value with W dyn .
We draw inspiration from NPE(Chang et al., 2016) and R-NEM(Van Steenkiste et al., 2018), both architectures model object relations.
NPE has access to ground truth representations of scenes and computes effect for each neighborhood object-pair with a neighborhood mask based on ground-truth object position, and ﬁnally, the summed effects act on object velocity.
R-NEM represents objects in an unsupervised fashion and implicitly encodes sequential information with an RNN(Hochreiter & Schmidhuber, 1997).
Instead of using a neighborhood mask, R-NEM learns soft attention for all object-pairs.
Our approach learns to represent object dynamics explicitly.
Thus, we compute all object-pair effects and let it act on object dynamics as NPE, and learn an attention coefﬁcient for each effect as R-NEM.
After distilling and updating object dynamics, we predict next-frame object latents by simply learning a transformation W pred , and W pred ∈ R ( L +4)
× L , ˆ z [-stat +
-]sta ( n +1) · = W pred
[ z dyn (cid:48) n · ; z stan · ]) .
To alleviate error accumulating when implementing multi-step predicting in the inference stage, we use student forcing strategy during training: instead of only using z n to predict ˆ z n +1 , ˆ z n is also used to iteratively roll out a predicted image sequence.
EXPERIMENTS
In this section, we design experiments in the perspective of representation, prediction, and scene decomposition.
In particular, we study how ODDN performs on tasks of video understanding and
[Table] [Table]
reasoning, video prediction, reconstruction, and segmentation.
To quantify segmentation quality, we measure the similarity between ground-truth (instance) segmentations and our predicted object masks using the Adjusted Rand Index(Rand, 1971) (ARI) and The Foreground Adjusted Rand Index (F-ARI) which is a modiﬁcation of the ARI score ignoring background pixels.
Baselines.
We compare our method with IODINE, PROVIDE and SRVP(Franceschi et al., 2020).
IODINE is designed for static images, thus it neither encodes object dynamics nor considers object relations.
PROVIDE is an extension of IODINE to ﬁt sequential inputs.
The major difference between PROVIDE and our approach is that PROVIDE encodes object dynamics implicitly while we distillate object dynamics explicitly.
In detail, PROVIDE uses 2D-LSTM to achieve temporally conditioned and iterative amortized inference for posterior reﬁnement, as a result, the ﬁnal sampled latents in every time-step contain dynamic information which could be used to predict the future.
Besides, PROVIDE does not model object relations, which means when predicting the future, object latents are processed separately.
SRVP is a dynamic latent model for stochastic video prediction which decouples frame synthesis and dynamics based on residual updates of a small latent state.
Unlike ODDN and PROVIDE, SRVP reperesents the scene with a single latent.
4. VIDEO UNDERSTANDING AND REASONING
We study the representation power of ODDN on CLEVRER, which poses different types of questions for each video in CLEVRER dataset.
Those questions could test the model’s understanding of a video comprehensively, including descriptive questions (“how many objects are moving”), explanatory questions(“the reason why certain collision happens”), predictive questions(“what will happen next”), and counterfactual questions(“what would happen if”).
Table 1 shows that ODDN-ALOE outperforms ALOE on all question types even without selfsupervision, especially on predictive and counterfactual questions, and achieves state-of-the-art performance.
Both predictive and counterfactual types of questions are related to predicting how objects will move.
This result proves that representations with object dynamics could help the neural model to do better understanding and reasoning around physical events.
We also do ablation experiments to study how different representations affect the model’s performance, results are shown in Table 2.
We can tell that representations of PROVIDE lead to a signiﬁcant decline compared to IODINE, and the reason will be discussed in Section. On the contrary, representations of ODDN which
G r o und t r u t h O DD N I O D
I N E li n e a r O DD N w / o r e l a t
i o n
S R V P t= t=3 t= t=5 t= t=7 t= t= P R O V I D E encode object dynamics (dim=20) outperform IODINE (dim=16) by a large margin, especially on predictive (91.8 vs 84.5) and counterfactual (80.1 vs 73.4) questions. In addition, we obtain ODDN features without dynamics by reducing 4 dimension dynamics.
Results show that ODDN without dynamics still performs better than IODINE, which means that ODDN not only distillates useful object dynamics but also generates better static representations.
static representations.
“ODDN w/o relation” outperforms “ODDN w/o relation − dyn ” which further shows the signiﬁcance of our distilled object dynamics. However, “DDN w/o relation − dyn ” is no better than IODINE (dim=16), which proves that the relation module is the main reason for the quality increase of the object static representation.
4. PREDICTION
ODDN achieves predicting future frames in an iterative fashion.
Unlike PROVIDE, ODDN models object dynamics and interactions explicitly.
Results could be observed in Figure 3, where the blue and brown cylinder collides. ODDN could output reasonable predictions and handle objects collision.
PROVIDE could only predict object’s future state independently, and the reason is that PROVIDE does not cope with objects interaction, thus, objects blend when they suppose to collide and change their dynamics.
ODDN also generates more stable predictions comparing to PROVIDE which fails to capture the object’s correct properties (shape, color) after a few frames. “IODINE linear” and SRVP generate identical pictures, which means that both methods are incapable of learning object-level dynamics. “ODDN w/o relation” learn some dynamics, but the generated results are not reasonable, which may be caused by the training noise: predicting the collision without modeling object interactions. We calculated the average MSE of each of the predicted 6 frames over the entire test set, the quantitative results are shown in Figure 7.
4. DECOMPOSITION
Reconstruction.
We show how different architectures perform in terms of reconstruction quality in Table 3.
Comparing to IODINE, PROVIDE brings some negative effects, as shown in Figure 6. [Figure] Both PROVIDE ODDN generate better seg-
[Figure] [Figure]
Model temporal [Table] [Table] PROVIDE √
[Table]
[Table]
[Table]
ODDN √
[Table]
[Table]
[Table]
ODDN* √ 0.
[Table]
PROVIDE occasionally fails to capture the right object shape.
Instead, we notice that ODDN performs better than IODINE.
Speciﬁcally, the reconstruction of close objects is more precise.
We believe that this beneﬁts from the relation module which relies on object surface details to process objects interaction (whether collision occurs and how object dynamics update).
The reconstruction performance is positively correlated to object representation quality which has been veriﬁed in Section.
Segmentation.
ODDN and baseline models decode object latents into pixel values and segmentation masks.
The major contribution of PROVIDE is that it models temporal dependencies between latents across frames to obtain better segmentation results.
Table 3 indicates that PROVIDE hurts the model’s reconstruction capability.
Instead, we show that the future frame prediction task in our architecture also improves the segmentation performance and achieves good reconstruction quality at the same time.
The ARI further improves when increasing the steps of future object states to predict.
We believe that the long-range prediction supervision encourages ODDN to attend more on foreground objects which may change its position and view over time.
This temporal consistency is similar to PROVIDE, which could be the main reason for both models to achieve better segmentation performance than IODINE.
Results are shown in Figure 4. [Figure] P R O V
I D E
S R V P t= t= t= t= t= t= t= t=
4. R EAL BLOCK TOWERS
Setup.
We further evaluate ODDN on the real block tower video dataset(Lerer et al., 2016).
The block tower dataset has 493 real-world videos and each video contains a block tower which may or may not be falling.
We train models on 393 randomly-selected videos and evaluate them on the rest 100 videos.
We study how the models perform in reconstruction and predicting the future frames with the physical concept of “falling”.
Results.
We show reconstruction results by ODDN and baseline models of images with block towers of different inclination in Figure 5.
We found that all models can reconstruct images of stable block towers precisely.
However, for blocks with complex spatial position states (e.g., falling with interactions), IODINE and PROVIDE often fail to restore accurate block properties.
ODDN performs the best and is able to retain the shape and inclination angle of the object.
In Figure 8, we present prediction results of ODDN, PROVIDE and SRVP.
SRVP predicts reasonable object dynamics but is not accurate, and the generated images are not clear.
The blocks predicted by
PROVIDE tend to be blurred and fused over time.
ODDN models object interactions, thus predicted blocks do not blend easily.
D ISCUSSION AND FUTURE WORK
We presented a unsupervised learning framework ODDN, a novel approach to decompose temporal scene of multiple objects with dynamics and interactions.
ODDN distills disentangled object dynamics which are signiﬁcant in downstream video reasoning task.
ODDN also models object interactions which endows the model with the capability of predicting future frames.
The incorporating of object dynamics and interactions beneﬁts the model’s decomposing ability, reﬂecting in better segmentation mask and reconstruction quality.
ODDN also has limitations.
The prediction quality decrease over time because of the error accumulation.
ODDN can only generate deterministic predictions for dynamics obeying regular patterns.
Besides, ODDN needs two frames to ﬁrstly derive object dynamics then predict the future.
However, in the block tower case, humans can tell whether the tower will fall or stay stable with one frame by commonsense.
We leave this for future work.
A DETAINS ON METHOD AND IMPLEMENTATION A. A LGORITHM
Here we detail our algorithm with pseudocode:
Algorithm
Input: video x 1 . . .
x
, hyperparamters K , T
Input: trainable parameters
, σ
Initialization: h k k ←−
Function InferReconstruct ( x ) : for t = 1 to T do z ( t )
k k ∼ q λ ( z ( t ) k | x ) µ ( t ) k , ˆ m ( t ) k
k ←− Decoder θ ( z ( t ) k ) m ( t ) ←− softmax k ( ˆ m
( t ) k ) p
( x | z ( t ) )
←− (cid:80) k m ( t ) k N ( x ; µ ( t ) k , σ 2 )
L ( t ) r ( x ) ←− D KL ( q λ ( z ( t ) | x ) || p ( z )) − βlog ( p ( x | z ( t ) ))
a k k ←− inputs( x, z ( t ) k , λ ( t ) k ) λ ( t +1) k , h ( t + 1) k
←− f φ ( a k , h ( t ) k ) end for for n = 1 to N − 2 do z n · , L r ( x n ) ←− InferReconstruct ( x n ) z ( n +1) · , L r ( x n +1 ) ←−
InferReconstruct ( x n +1 ) dyn ( n +1) ·
←− Transformer ρ ( z n · , z ( n +1) · ) dyn (cid:48) ( n +1) ·
←− Relation δ ( dyn ( n +1) · , z ( n +1) · ) ˆ z ( n +2)
k
k ←− Predictor σ ( dyn (cid:48) ( n +1) k , z ( n +1) k )
L p ( x ( n +2) )
←− − log ( p ( x
[Equation]
n
[Equation]
n +
[Equation]
A. DATASET
CLEVRER is a synthetic video dataset of moving and colliding objects.
Each video containsframes at resolution 480 × 320.
We pre-train ODDN on the entire CLEVRER training set, in order to promote convergence of our Relation Module, we extract images every 4 frames and ensure that at least one collision event is included forming CLEVRER-collision and we ﬁnetune ODDN on CLEVRER-collision.
For testing, we use the validation set which has ground truth masks, we sample 1k sub-clips containing 6 objects, each sub-clip consists of 10 frames.
A. HYPERAMETERS
We Generally follow the setting of PROVIDE.
We initialize the parameters of the posterior λ by sampling from U ( − 0 .
In experiments in prediction tasks, we use a latent dimensionality dim(z) = 64 and downscals the image into 64 × 64 after a center-crop preprocess as IODINE and
PROVIDE, such that dim( λ ) = 128.
And in experiments in video resoning task, we use a latent dimensionality dim(z)
{+= 
+}
as ALOE which makes dim( λ ) = 32, and downscals the image into 64 × 96 without crop.
The variance of the likelihood is set to σ = 0.3 in all experiments.
We keep the default number of iterative reﬁnements at R = 5, and use K = 7 slots for both training and testing.
Furthermore, we set β = 100.0 for all experiments. A. TRAINING We train our models on 8 GeForce RTX 3090 GPUs, which takes approximately two days per model. We use ADAM for all experiments, with a learning rate of 0.0003 and default values for all remaining parameters.
During training, we gradually increase the number of frames per video, as we have found this to make the optimisation more stable. We train models with sequences of length 8 and the batch size is 12.
B MODEL COMPUTATIONAL COMPLEXITY
[Table]
C M ORE VISUALIZATION RESULTS C. C OMPONENT VISUALIZATION
Attention.
We visualize the attention coefﬁcients of our relation module, results are shown in
Figure 10.
The cyan cylinder mostly attends on the brown cylinder and the attention score achieves the maximum when the collision happens, while other objects which do not get involved in the interaction process barely gain attention.
We also draw an attention map for the Transformer in
Figure 11(b), which shows that the attention model could match latents in different frames precisely.
Disentanglement.
Disentanglement is an important property that describes how well representations separate features into distinct factors.
In Figure 11(c), we show that different dimensions of the learned object dynamics corresponding to different moving directions. [Figure] Interaction attention coefficients of
[Equation]
[Figure] [Figure]