View source for Synthesizing Programs for Images usingReinforced Adversarial Learning

'''Synthesizing Programs for Images using Reinforced Adversarial Learning: ''' Summary of the ICML 2018 paper 

Paper: [[http://proceedings.mlr.press/v80/ganin18a.html]]
Video: [[https://www.youtube.com/watch?v=iSyvwAwa7vk&feature=youtu.be]]

== Presented by ==

1. Nekoei, Hadi [Quest ID: 20727088]

= Motivation =

Conventional neural generative models have major problems. 

* It is not clear how to inject knowledge about the data into the model. 

* Latent space is not easily interpretative. 

The provided solution in this paper is to generate programs to incorporate tools, e.g. graphics editors, illustration software, CAD. and '''creating more meaningful API(sequence of complex actions vs raw pixels)'''.

= Introduction =

Humans, frequently, use the ability to recover structured representation from raw sensation to understand their environment. Decomposing a picture of a hand-written character into strokes or understanding the layout of a building can be exploited to learn how actually our brain works.  

In the visual domain, inversion of a renderer for the purposes of scene understanding is typically referred to as inverse graphics. However, training vision systems using the inverse graphics approach has remained a challenge. Renderers typically expect as input programs that have sequential semantics, are composed of discrete symbols (e.g., keystrokes in a CAD program), and are long (tens or hundreds of symbols). Additionally, matching rendered images to real data poses an optimization problem as black-box graphics simulators are not differentiable in general.  

To address these problems, a new approach is presented for interpreting and generating images using Deep Reinforced Adversarial Learning in order to solve the need for a large amount of supervision and scalability to larger real-world datasets. In this approach, an adversarially trained agent '''(SPIRAL)''' generates a program which is executed by a graphics engine to generate images, either conditioned on data or unconditionally. The agent is rewarded by fooling a discriminator network and is trained with distributed reinforcement learning without any extra supervision. The discriminator network itself is trained to distinguish between generated and real images.

[[File:Fig1 SPIRAL.PNG | 400px|center]]

== Related Work ==
Related works in this filed is summarized as follows:
* There has been a huge amount of studies on inverting simulators to interpret images (Nair et al., 2008; Paysan et al., 2009; Mansinghka et al., 2013; Loper & Black, 2014; Kulkarni et al., 2015a; Jampani et al., 2015)

* Inferring motor programs for reconstruction of MNIST digits (Nair & Hinton, 2006)

* Visual program induction in the context of hand-written characters on the OMNIGLOT dataset (Lake et al., 2015)

* inferring and learning feed-forward or recurrent procedures for image generation (LeCun et al., 2015; Hinton & Salakhutdinov, 2006; Goodfellow et al., 2014; Ackley et al., 1987; Kingma & Welling, 2013; Oord et al., 2016; Kulkarni et al., 2015b; Eslami et al., 2016; Reed et al., 2017; Gregor et al., 2015).

'''However, all of these methods have limitations such as:''' 

* Scaling to larger real-world datasets

* Requiring hand-crafted parses and supervision in the form of sketches and corresponding images

* Lack the ability to infer structured representations of images

= The SPIRAL Agent =
=== Overview ===
The paper aims to construct a generative model <math>\mathbf{G}</math> to take samples from a distribution <math>p_{d}</math>. The generative model consists of a recurrent network <math>\pi</math> (called policy network or agent) and an external rendering simulator R that accepts a sequence of commands from the agent and maps them into the domain of interest, e.g. R could be a CAD program rendering descriptions of primitives into 3D scenes. 
In order to train policy network <math>\pi</math>, the paper has exploited generative adversarial network. In this framework, the generator tries to fool a discriminator network which is trained to distinguish between real and fake samples. Thus, the distribution generated by <math>\mathbf{G}</math> approaches <math>p_d</math>.

== Objectives ==
The authors give training objective for <math>\mathbf{G}</math> and <math>\mathbf{D}</math> as follows.

'''Discriminator:''' Following (Gulrajani et al., 2017), the objective for <math>\mathbf{D}</math> is defined as: 

\begin{align}
\mathcal{L}_D = -\mathbb{E}_{x\sim p_d}[D(x)] + \mathbb{E}_{x\sim p_g}[D(x)] + R
\end{align}

where <math>\mathbf{R}</math> is a regularization term softly constraining <math>\mathbf{D}</math> to stay in the set of Lipschitz continuous functions (for some fixed Lipschitz constant).

'''Generator:''' To define the objective for <math>\mathbf{G}</math>, a variant of the REINFORCE (Williams, 1992) algorithm, advantage actor-critic (A2C) is employed:


\begin{align}
\mathcal{L}_G = -\sum_{t}\log\pi(a_t|s_t;\theta)[R_t - V^{\pi}(s_t)]
\end{align}


where <math>V^{\pi}</math> is an approximation to the value function which is considered to be independent of theta, and <math>R_{t} = \sum_{t}^{N}r_{t}</math> is a 
1-sample Monte-Carlo estimate of the return. Rewards are set to:

<math>
r_t = \begin{cases}
 0 & \text{for } t < N \\
 D(\mathcal{R}(a_1, a_2, \cdots, a_N)) & \text{for } t = N
\end{cases}
</math>



One interesting aspect of this new formulation is that 
the search can be biased by introducing intermediate rewards
which may depend not only on the output of R but also on
commands used to generate that output.

== Conditional generation: ==
In some cases such as producing a given image <math>x_{target}</math>, conditioning the model on auxiliary inputs is useful. That can be done by feeding <math>x_{target}</math> to both policy and discriminator networks as:
<math>
p_g = R(p_a(a|x_{target}))
</math>

While <math>p_{d}</math> becomes a Dirac-<math>\delta</math> function centered at <math>x_{target}</math>. 
For the first two terms in the objective function for D, they reduce to 
<math>
-D(x_{target}|x_{target})+ \mathbb{E}_{x\sim p_g}[D(x|x_{target})] 
</math>

It can be proven that for this particular setting of <math>p_{g}</math> and <math>p_{d}</math>, the <math>l2</math>-distance is an optimal discriminator. It may be as a poor candidate for the reward signal of the generator, even if it is not the only solution of the objective function for D.

===Traditional GAN generation: ===
Traditional GANs use the following minimax objective function to quantify optimality for relationships between D and G:

[[File:edit7.png| 400px|center]]

Minimizing the Jensen-Shannon divergence between the two distribution often leads to vanishing gradients as the discriminator saturates. We circumvent this issue using the conditional generation function, which is much better behaved.

== Distributed Learning: ==
The training pipeline is outlined in Figure 2b. It is an extension of the recently proposed '''IMPALA''' architecture (Espeholt et al., 2018). For training, three kinds of workers are defined:


* Actors are responsible for generating the training trajectories through interaction between the policy network and the rendering simulator. Each trajectory contains a sequence <math>((\pi_{t}; a_{t}) | 1 \leq t \leq N)</math> as well as all intermediate
renderings produced by R.


* A policy learner receives trajectories from the actors, combines them into a batch and updates <math>\pi</math> by performing '''SGD''' step on <math>\mathcal{L}_G</math> (2). Following common practice (Mnih et al., 2016),  <math>\mathcal{L}_G</math> is augmented with an entropy penalty encouraging exploration.


* In contrast to the base '''IMPALA''' setup, an additional discriminator learner is defined. This worker consumes random examples from <math>p_{d}</math>, as well as generated data (final renders) coming from the actor workers, and optimizes <math>\mathcal{L}_D</math> (1).

[[File:Fig2 SPIRAL Architecture.png | 700px|center]]

'''Note:''' no trajectories are omitted in the policy learner. Instead, the <math>D</math> updates is decoupled from the <math>\pi</math> updates by introducing a replay buffer that serves as a communication layer between the actors and the discriminator learner. That allows the latter to optimize <math>D</math> at a higher rate than the training of the policy network due to the difference in network sizes (<math>\pi</math> is a multi-step RNN, while <math>D</math> is a plain '''CNN'''). Even though sampling from a replay buffer inevitably results in smoothing of <math>p_{g}</math>, this setup is found to work well in practice.

= Experiments=


== Environments ==
Two rendering environment is introduced. For MNIST, OMNIGLOT and CELEBA generation an open-source painting librabry LIMBYPAINT (libmypaint
contributors, 2018).) is used. The agent controls a brush and produces
a sequence of (possibly disjoint) strokes on a canvas
C. The state of the environment is comprised of the contents
of <math>C</math> as well as the current brush location <math>l_{t}</math>. Each action
$a_{t}$ is a tuple of 8 discrete decisions <math>(a1t; a2t; ... ; a8t)</math> (see
Figure 3). The first two components are the control point <math>p_{c}</math>
and the endpoint <math>l_{t+1}</math> of the stroke.

[[File:Fig3_agent_action_space.PNG | 450px|center]]

The next 5
components represent the appearance of the stroke: the
pressure that the agent applies to the brush (10 levels), the
brush size, and the stroke color characterized by a mixture
of red, green and blue (20 bins for each color component).
The last element of at is a binary flag specifying the type
of action: the agent can choose either to produce a stroke
or to jump right to $l_{t+1}$.

In the MUJOCO SCENES experiment, we render images
using a MuJoCo-based environment (Todorov et al., 2012).
At each time step, the agent has to decide on the object
type (4 options), its location on a 16 $\times$ 16 grid, its size
(3 options) and the color (3 color components with 4 bins
each). The resulting tuple is sent to the environment, which
adds an object to the scene according to the specification.

== Datasets ==

=== MNIST ===
For the MNIST dataset, two sets of experiments are conducted:

1- In this experiment, an unconditional agent is trained to model the data distribution. Along with the reward provided by the discriminator, a small negative reward is provided to the agent for each continuous sequence of strokes to encourage the agent to draw a digit in a continuous motion of stroke. Example of such generation is depicted in the Fig 4a. 

2- In the second experiment, an agent is trained to reproduce a given digit. 
Several examples of conditional generated digits are shown in Fig 4b. 

[[File:Fig4a MNIST.png | 450px|center]]

=== OMNIGLOT ===
Now the trained agents are tested in a similar but more challenging setting of handwritten characters. As can be seen in Fig 5a, the unconditional generation has a lower quality compared to digits in the previous dataset. The conditional agents, on the other hand, were able to reach a convincing quality (Fig 5b). Moreover, as OMNIGLOT has lots of different symbols, the model that we created was able to learn a general idea of image production without memorizing the training data. We tested this result by inputting new unseen line drawings to our trained agent. As we concluded, it provided excellent results as shown in Figure 6.  

[[File:Fig5 OMNIGLOT.png | 450px|center]]


For the MNIST dataset, two kinds of rewards, discriminator score and <math>l^{2}-\text{distance}</math> has been compared. Note that the discriminator based approach has a significantly lower training time and lower final <math>l^{2}</math> error.
Following (Sharma et al., 2017),  also a “blind” version of the agent without feeding any intermediate canvas states as an input to <math>\pi</math> is trained. The training curve for this experiment is also reported in Fig 8a. 
(dotted blue line) The results of training agents with discriminator based and <math>l^{2}-\text{distance}</math> approach is shown in Fig 8a as well.

=== CELEBA ===

Since the ''libmypaint'' environment is also capable of producing
complex color paintings, this direction is explored by
training a conditional agent on the CELEBA dataset. In this
experiment, the agent does not receive any intermediate rewards.
In addition to the reconstruction reward (either <math>l^2</math> or
discriminator-based), earth mover’s
distance between the color histograms of the model’s output
and <math>x_{target}</math> is penalized. (Figure 7)

[[File:Fig6 CELEBA.png | 450px|center]]

Although blurry, the model’s reconstruction closely matches
the high-level structure of each image such as the
background color, the position of the face, and the color of
the person’s hair. In some cases, shadows around eyes and
the nose are visible.

=== MUJOCO SCENES ===

For the MUJOCO SCENES dataset, the trained agent is used to construct simple CAD programs that best explain input images. Here only the case of the conditional generation is considered. Like before, the reward function for the generator can be either the <math>l^2</math> score or the discriminator output. In addition, there are not any auxiliary reward signals. This model has the capacity to infer and represent up to 20 objects and their attributes due to its unrolled 20 time steps.

As shown in Figure 8b, the agent trained to directly minimize
<math>l^2</math>  is unable to solve the task and has significantly
higher pixel-wise error. In comparison, the discriminator based
variant solves the task and produces near-perfect reconstructions
on a holdout set (Figure 10).

[[File:Fig8 MUJOCO_SCENES.png | 500px|center]]
For this experiment, the total number of possible execution traces is   <math>M^N</math>, where <math>M = 4·16^2·3·4^3·3 </math> is the total number of attribute settings for a single object and N = 20 is the length of an episode. Then a general-purpose Metropolis-Hastings inference algorithm that samples an execution trace defining attributes for a maximum of 20 primitives was run on a set of 100 images. These attributes are considered as latent variables. During each time step of the inference, the attribute blocks (including presence/absence tags) corresponding to a single object are evenly flipped over the appropriate range. The resulting trace is presented as an output sample by the environment and then the output sample is accepted or rejected using the Metropolis-Hastings update rule, where the Gaussian likelihood is centered on the test image and the fixed diagonal covariance is 0.25. From Figure 9, the MCMC search baseline cannot solve the task even after a lot of evaluation.
[[File:figure9 mcmc.PNG| 500px|center]]

= Discussion =
As in the OMNIGLOT
experiment, the <math>l^2</math>-based agent demonstrates some
improvements over the random policy but gets stuck and as
a result, fails to learn sensible reconstructions (Figure 8b).

[[File:Fig7 Results.png | 500px|center]]


Scaling visual program synthesis to the real world and combinatorial
datasets has been a challenge. It has been shown that it is possible to train an adversarial generative agent employing
black-box rendering simulator. Our results indicate that
using the Wasserstein discriminator’s output as a reward
function with asynchronous reinforcement learning can provide
a scaling path for visual program synthesis. The current
exploration strategy used in the agent is entropy-based but
future work should address this limitation by employing sophisticated
search algorithms for policy improvement. For
instance, Monte Carlo Tree Search can be used, analogous
to AlphaGo Zero (Silver et al., 2017). General-purpose
inference algorithms could also be used for this purpose.

= Future Work =
Future work should explore different parameterizations of
action spaces. For instance, the use of two arbitrary control
points are perhaps not the best way to represent strokes, as it
is hard to deal with straight lines. Actions could also directly parametrize 3D surfaces, planes, and learned texture models
to invert richer visual scenes. On the reward side, using
a joint image-action discriminator similar to BiGAN/ALI
(Donahue et al., 2016; Dumoulin et al., 2016) (in this case,
the policy can be viewed as an encoder, while the renderer becomes
a decoder) could result in a more meaningful learning
signal, since D will be forced to focus on the semantics of
the image.

== Other Resources ==
#Code implementation [https://github.com/carpedm20/SPIRAL-tensorflow]
= References =

# Yaroslav Ganin, Tejas Kulkarni, Igor Babuschkin, S.M. Ali Eslami, Oriol Vinyals, [[https://arxiv.org/abs/1804.01118]].