Recursive Neural Networks Based on PSO for Image Parsing

and Applied Analysis 3


Introduction
Image parsing is an important step towards understanding an image, which is to perform a full-scene labeling.The task of image parsing consists in labeling every pixel in the image with the category of the object it belongs to.After a perfect image parsing, every region and every object are delineated and tagged [1].Image parsing is frequently used in a wide variety of tasks including parsing scene [2,3], aerial image [4], and facade [5].
During the past decade, the image parsing technique has undergone rapid development.Some methods for this task such as [6] rely on a global descriptor which can do very well for classifying scenes into broad categories.However, these approaches fail to gain a deeper understanding of the objects in the scene.Many other methods rely on CRFs [7], MRFs [8], or other types of graphical models [9,10] to ensure the consistency of the labeling and to account for context.Also, there are many approaches for image annotation and semantic segmentation of objects into regions [11].Note that most of the graphical-based methods rely on a presegmentation into superpixels or other segment candidates and extract features and categories from individual segments and from various combinations of neighboring segments.The graphical model inference pulls out the most consistent set of segments which covers the image [1].Recently, these ideas have been combined to provide more detailed scene understanding [12][13][14][15].
It is well known that many graphical methods are based on neural networks.The main reason is that neural networks have promising potential for tasks of classification, associative memory, parallel computation, and solving optimization problems [16].In 2011, Socher et al. proposed a RNNbased parsing algorithm that aggregates segments in a greedy strategy using a trained scoring function [17].It recursively merges pairs of segments into supersegments in a semantically and structurally coherent way.The main contribution of the approach is that the feature vector of the combination of two segments is computed from the feature vectors of the individual segments through a trainable function.Experimental results on Stanford background dataset revealed that RNN-based method outperforms state-of-the-art approaches in segmentation, annotation, and scene classification.That being said, it is worth noting that the objective function is nondifferentiable due to the hinge loss.This could cause problems since one of the principles of L-BFGS, which is employed as the training algorithm in RNN, is that the objective should be differentiable.
Since Particle Swarm Optimization (PSO) [18] has proven to be an efficient and powerful problem-solving strategy, we use a novel nonlinear PSO [19] to tune the weights of RNN.The main idea is to use particle swarm for searching good

Image Parsing Based on Recursive Neural Networks
The main idea behind recursive neural networks for image parsing lies in that images are oversegmented into small regions and each segment has a vision feature.These features are then mapped into a "semantic" space using a recursive neural network.Figure 1 outlines the approach for RNNbased image parsing method.Note that the RNN computes (i) a score that is higher when neighboring regions should be merged into a larger region, (ii) a new semantic feature representation for this larger region, and (iii) its class label.
After regions with the same object label are merged, neighboring objects are merged to form the full scene image.These merging decisions implicitly define a tree structure in which each node has associated with the RNN outputs (i)-(iii), and higher nodes represent increasingly larger elements of the image.Details of the algorithm are given from Sections 2.1 to 2.3.

Input Representation of Scene Images. Firstly, an image
is oversegmented into superpixels (also called segments) using the algorithm from [20].Secondly, for each segment, compute 119 features via [10].These features include color and texture features, boosted pixel classifier scores (trained on the labeled training data), and appearance and shape features.Thirdly, a simple neural network layer has been used to map these features into the "semantic" -dimensional space in which the RNN operates, given as follows.
Let   be the features described previously for each segment, where  = 1, . . .,  segs and  segs denotes the number of segments in an image.Then the representation is given as where  sem ∈  ×119 is the matrix of parameters we want to learn,  sem is the bias, and  is applied element wise and can be any sigmoid-like function.In [17], the original sigmoid is function () = 1/(1 +  − ) (Figure 2).

Greedy Structure Predicting.
Since there are more than exponentially many possible parse trees and no efficient dynamic programming algorithms for RNN setting, therefore, Socher recommended a greedy strategy.The algorithm finds the pairs of neighboring segments and adds their activations to a set of potential child node pairs.Then the network computes the potential parent representation for these possible child nodes: With this representation, a local score can be determined by using a simple inner product with a row vector  score ∈  1× : As illustrated in Figure 3, the recursive neural network is different from the original RNN in that it predicts a score for being a correct merging decision.The process repeats until all pairs are merged and only one parent activation is left, as shown in Figure 1.The final score that we need for structure prediction is simply the sum of all the local decisions: where  are all the parameters needed to compute a score  with an RNN, ŷ is a parse for input   , and (ŷ) is the set of nonterminal nodes.

Category Classifiers in the Tree.
The main advantage of the algorithm is that each node of the tree built by the RNN has associated with it a distributed feature representation.To predict class labels, a simple softmax layer is added to each RNN parent node, as shown later: When minimizing the cross-entropy error of this softmax layer, the error will backpropagate and influence the RNN parameters.

Nonlinear Particle Swarm Optimization for Training FNN
As for traditional RNN-based method, the objective  of ( 5) is not differentiable due to the hinge loss.For training RNN, Socher used L-BFGS over the complete training data to minimize the objective, where the iteration of the swarm relates to the update of the parameters of RNN.That being said, it is worth noting that the basic principle of L-BFGS is that the objective function should be differentiable.Since the objective function for RNN is nondifferentiable, L-BFGS could cause problems for computing the weights of RNN.To solve this problem, a novel nonlinear PSO (NPSO) has been used to tune the parameters of RNN.

Nonlinear Particle Swarm Optimization.
As a populationbased evolutionary algorithm, PSO is initialized with a population of candidate solutions.The activities of the population are guided by some behavior rules.For example, let   () = ( 1 (),  2 (), . . .,   ()) (  () ∈ [−  max ,   max ]) be the location of the th particle in the th generation, where   max is the boundary of the th search space for a given problem and  = 1, . . ., .The location of the best fitness achieved so far by the th particle is denoted as   () and the index of the global best fitness by the whole population as   ().The velocity of th particle is   () = (V 1 (), V 2 (), . . ., V  ()), where V  is in [−V  max , V  max ] and V  max is the maximal speed of th dimension.The velocity and position update equations of the th particle are given as follows: where  = 1, . . .,  and  = 1, . . ., . ,  1 ,  2 ≥ 0.  is the inertia weight,  1 and  2 denote the acceleration coefficients, and  1 and  2 are random numbers, generated uniformly in the range [0, 1].Note that a suitable value for the inertia weight provides a balance between the global and local exploration abilities of the swarm.Based on the concept of decrease strategy, our nonlinear inertia weight strategy [19] chooses a lower value of  during the early iterations and maintains higher value of  than linear model [21].This strategy enables particles to search the solution space more aggressively to look for "better areas", thus will avoid local optimum effectively.
The proposed update scheme of () is given as follows: where iter max is the maximum number of iterations,  denotes the iteration generation, and  > 1 is the nonlinear modulation index.Figure 4 illustrates the variations of nonlinear inertia weight for different values of .Note that  = 1 is equal to the linear model.In [19], we showed that a choice of  within [2-3] is normally satisfactory.

Encoding Strategy and Fitness Evaluation.
Let  = ( sem ; ;  score ;  label ) be the set of RNN parameters; then each particle can be the expressed as the combination of all parameters, as shown later: During the iteration, each particle relates to a combination of weights of neural networks.The goal is to minimize a fitness function, given as where   () = (RNN(,   ,  * )) + Δ(  ,   ,  * ) − max   ∈(  ,  ) ((RNN(,   ,   ))) and  * denote the parse tree generated by the greedy strategy according to parameter .Minimizing this objective means minimize the error between the parsing results, which is generated by the best particle and the labeled training images (ground truth).

Summary of PSO-Based Training Algorithm.
Input includes a set of labeled images, the size of the hidden layer , the value of penalization term for incorrect parsing decisions , the regularization parameter , the population of particles , the values of nonlinear parameter  and the number of iterations iter max .
Output includes the set of model parameters  = ( sem , ,  score , and  label ), each with respect to weights of a recursive neural network.
(1) Randomly initialize  particles and randomize the positions and velocities for entire population.Record the global best location   of the population and the local best locations   of the th particle according to (9), where  = 1, 2, . . ., .
(2) For each iteration, evaluate the fitness value of the th particle through (9).If ((  )) < ((  )), set   =   as the so far best position of the th particle.If ((  )) < ((  )), set   =   as the so far best position of the population.
(3) Calculate the inertia weight through (7).Update the position and velocity of particles according to (6).
(4) Repeat Step 2 and Step 3 until maximum number of generation.
(5) Compute the weights of RNN according to the best particle.

Experimental Results and Discussion
4.1.Description of the Experiments.In this section, PSObased RNN method is compared with traditional RNN [17], pixel CRF [10], region-based energy [10], simultaneous MRF [8], and superpixel MRF [8], by using images from Stanford background dataset.All the experiments have been conducted on a computer with Intel sixteen-core processor 2.67 GHz processor and 32 GB RAM.As for RNN, Socher recommends that the size of the hidden layer  = 100, the penalization term for incorrect parsing decisions  = 0.05, and the regularization parameter  = 0.001.As for the particle swarm optimization, we set

Scene Annotation.
The first experiment aims at evaluating the accuracy of scene annotation on the Stanford background dataset.Like [17], we run fivefold cross-validation and report pixel level accuracy in Table 1.Note that the traditional RNN model influences the leaf embeddings through backpropagation, while we use PSO to tune the weights of RNN.
As for traditional RNN model, we label the superpixels by their most likely class based on the multinomial distribution from the softmax layer at the leaf nodes.One can see that in Table 1, our approach outperforms previous methods that report results on this data, which means that the PSO-based RNN constructs a promising strategy for scene annotation.Some typical parsing results are illustrated in Figure 5. [17], the Stanford background dataset can be roughly categorized into three scene types: city, countryside, and sea side.Therefore, like traditional RNN, we trained SVM that using the average over all nodes' activations in the tree as features.That means the entire parse tree and the learned feature representations of the RNN are taken into account.As a result, the accuracy has been promoted to 88.4%, which is better than traditional RNN (88.1%) and Gist descriptors (84%) [6].If only the top node of the scene parse tree is considered, we will get 72%.The results reveal that it does lose some information that is captured by averaging all tree nodes.

Conclusions
In this paper, we have proposed an image parsing algorithm that is based on PSO and Recursive Neural Networks (RNNs).The algorithm is an incremental version of RNN.The basic idea is to solve the problem of nondifferentiable objective function of traditional training algorithm such as L-BFGS.Hence, PSO has been employed as an optimization tool to tune the weights of RNN.The experimental results reveal that the proposed algorithm has better performance than stateof-the-art methods on Stanford background dataset.That being said, the iteration of swarms dramatically increases the runtime of the training process.Our future work may focus on reducing the time complexity of the algorithm.

Figure 1 :
Figure 1: Hierarchical architecture of image parsing based on recursive neural network.

Figure 2 :Figure 3 :
Figure 2: Illustration of the RNN training inputs: (a) a training image (red and blue are differently labeled regions).(b) An adjacency matrix of image segments.(c) A set of correct trees which is oblivious to the order in which segments with the same label are merged.

Figure 5 :
Figure 5: Typical results of multiclass image segmentation and pixel-wise labeling with PSO-based recursive neural networks.