Scrutinizing and De-Biasing Intuitive Physics with Neural Stethoscopes

Published at BMVC 2019

Authors: Fabian Fuchs, Oliver Groth, Adam Kosiorek, Alex Bewley, Markus Wulfmeier, Andrea Vedaldi, Ingmar Posner


Predicting the stability of block towers is a popular task in the domain of intuitive physics. Previously, work in this area focused on prediction accuracy, a one-dimensional performance measure. We provide a broader analysis of the learned physical understanding of the final model and how the learning process can be guided. To this end, we introduce neural stethoscopes as a general purpose framework for quantifying the degree of importance of specific factors of influence in deep neural networks as well as for actively promoting and suppressing information as appropriate. In doing so, we unify concepts from multitask learning as well as training with auxiliary and adversarial losses. We apply neural stethoscopes to analyse the state-of-the-art neural network for stability prediction. We show that the baseline model is susceptible to being misled by incorrect visual cues. This leads to a performance breakdown to the level of random guessing when training on scenarios where visual cues are inversely correlated with stability. Using stethoscopes to promote meaningful feature extraction increases performance from 51% to 90% prediction accuracy. Conversely, training on an easy dataset where visual cues are positively correlated with stability, the baseline model learns a bias leading to poor performance on a harder dataset. Using an adversarial stethoscope, the network is successfully de-biased, leading to a performance increase from 66% to 88%.

Teaser Figure

Paper

Dataset

We extended the ShapeStacks dataset to specifically tailor it towards scrutinising biases in the physical reasoning of a trained model. The additional label of local stability allows for investigating to what extent a trained model follows visual cues. In the paper Scrutinising and De-Biasing Intuitive Physics with Neural Stethoscopes, we promote a framework combining adversarial and auxiliary training to promote or suppress extraction of features to guide the learning process.

How to use it

Manual

RGB images (7.1 GB):

steth_shapestacks_rgb.tar.gz
steth_shapestacks_rgb.tar.gz.md5

MuJoCo world definitions (optional, 30 MB):

steth_shapestacks_mjcf.tar.gz
steth_shapestacks_mjcf.tar.gz.md5

Image Dataset