# Codebooks ## K-means codebook A common technique for codebook generation involves utilising K-means clustering on a sample of the image descriptors. In this way descriptors may be mapped to *visual* words which lend themselves to binning and therefore the creation of bag-of-words histograms for the use of classification. In this courseworok 100-thousand random SIFT descriptors of the Caltech_101 dataset are used to build the K-means visual vocabulary. ## Vocabulary size The number of clusters or the number of centroids determines the vocabulary size when creating the codebook with the K-means the method. Each descriptor is mapped to the nearest centroid, and each descriptor belonging to that cluster is mapped to the same *visual word*. This allows similar descriptors to be mapped to the same word, allowing for comparison through bag-of-words techniques. ## Bag-of-words histogram quantisation of descriptor vectors An example histograms for training and testing images is shown on figure \ref{fig:histo_tr}, computed with a vocubulary size of 100. The histograms of the same class appear to have comparable magnitudes for their respective keywords, demonstrating they had a similar number of descriptors which mapped to each of the clusters. The effect of the vocubalary size (as determined by the number of K-means centroids) on the classificaiton accuracy is shown in figure \ref{fig:km_vocsize}. A small vocabulary size tends to misrepresent the information contained in the different patches, resulting in poor classification accuracy. Conversly a large vocabulary size (many K-mean centroids), may display overfitting. In our tests, we observe a plateau after a cluster count of 60 on figure \ref{fig:km_vocsize}. \begin{figure}[H] \begin{center} \includegraphics[width=12em]{fig/kmeans_vocsize.pdf} \includegraphics[width=12em]{fig/time_kmeans.pdf} \caption{Effect of vocabulary size; classification error (left) and time (right)} \label{fig:km_vocsize} \end{center} \end{figure} The time complexity of quantisation with a K-means codebooks is $O(DNK)$, where N is the number of entities to be clustered (descriptors), D is the dimension (of the descriptors) and K is the cluster count [@km-complexity]. As the computation time is high, the tests we use a subsample of descriptors to compute the centroids (a random selection of 100 thousand descriptors). An alternative method we tried is applying PCA to the descriptors vectors to improve time performance. However, the descriptor dimension of 128 is relatiely small and as such we found PCA to be unnecessary. K-means is a process that converges to local optima and heavily depends on the initialization values of the centroids. Initializing K-means is an expensive process, based on sequential attempts of centroids placement. Running for multiple instances significantly affects the computation process, leading to a linear increase in execution time. We did not observe increase in accuracy with K-means estimator size larger than one, and therefore present results accuracy and execution time results with a single K-Mean estimator. \begin{figure}[H] \begin{center} \includegraphics[width=12em]{fig/trainhist.pdf} \includegraphics[width=12em]{fig/testhist.pdf} \caption{Bag-of-words histograms; Training (left), Testing (right)} \label{fig:histo_tr} \end{center} \end{figure} # RF classifier We use a random forest classifier to label images based on the bag-of-words histograms. Random forests are an ensemble of randomly generated decision trees. Random forest classifier performance depends on the ensemble size, tree depth, randomness and weak learner used. ## Hyperparameters tuning Figure \ref{fig:km-tree-param} shows the effect of tree depth and number of trees, when classifying a bag-of-words created by K-means with 100 cluster centers. \begin{figure}[H] \begin{center} \includegraphics[width=12em]{fig/error_depth_kmean100.pdf} \includegraphics[width=12em]{fig/trees_kmean.pdf} \caption{Classification error varying tree depth (left) and forest size (right)} \label{fig:km-tree-param} \end{center} \end{figure} Random forests will select a random number of features on which to apply a weak learner (such as axis aligned split) and then chose the best feature of the sampled ones to perform the split on, based on a given criteria (our results use the *Gini index*). The fewer features that are compared for each split the quicker the trees are built and the more random they are. Therefore the randomness parameter can be considered the number of features used when making splits. We evaluate accuracy given different randomness when using a K-means vocabulary in figure \ref{fig:kmeanrandom}. The results in the figure \ref{fig:kmeanrandom} use a forest size of 100 as we infered that this is the estimatator count for which performance gains tend to plateau (when selecting $\sqrt{n}$ random features). \begin{figure}[H] \begin{center} \includegraphics[width=18em]{fig/new_kmean_random.pdf} \caption{Classification error for different number of random features} \label{fig:kmeanrandom} \end{center} \end{figure} Changing the randomness parameter had no significant effect on execution time. This may be attributed to increased required tree depth to purify the training set. ## Weak Learner comparison In figure \ref{fig:2pt} it is possible to notice an improvement in recognition accuracy by 1%, with the two pixels test, achieving better results than the axis-aligned counterpart. The two-pixels test however brings a slight deacrease in time performance which has been measured to be on average 1 second more. This is due to the complexity added by the two-pixels test, since it adds one dimension to the computation. \begin{figure}[H] \begin{center} \includegraphics[width=18em]{fig/2pixels_kmean.pdf} \caption{K-means classification accuracy changing the type of weak learners} \label{fig:2pt} \end{center} \end{figure} \begin{figure}[H] \begin{center} \includegraphics[width=18em]{fig/e100k256d5_cm.pdf} \caption{Confusion Matrix: K=256, ClassifierForestSize=100, Depth=5} \label{fig:km_cm} \end{center} \end{figure} \begin{figure}[H] \begin{center} \includegraphics[width=10em]{fig/success_km.pdf} \includegraphics[width=10em]{fig/fail_km.pdf} \caption{K-means + RF Classifier: Success (left); Failure (right)} \label{fig:km_succ} \end{center} \end{figure} # RF codebook An alternative to codebook creation via K-means involves using an ensemble of totally random trees. We code each decriptor according to which leaf of each tree in the ensemble it is sorted. This effectively performs and unsupervised transformation of our dataset to a high-dimensional spare representation. The dimension of the vocabulary size is determined by the number of leaves in each random tree and the ensemble size. From comparing execution times of K-means in figure \ref{fig:km_vocsize} and the RF codebook in \ref{fig:p3_voc} we observe considerable speed gains from utilising the RF codebook. This may be attributed to the reduce complexity of RF Codebook creation, which is $O(\sqrt{D} N \log K)$ compared to $O(DNK)$ for K-means. Codebook mapping given a created vocabulary is also quicker than K-means, $O(\log K)$ (assuming a balanced tree) vs $O(KD)$. \begin{figure}[H] \begin{center} \includegraphics[width=18em]{fig/256t1_e200D5_cm.pdf} \caption{Confusion Matrix: CodeBookForestSize=256; ClassifierForestSize=200; Depth=5} \label{fig:p3_cm} \end{center} \end{figure} \begin{figure}[H] \begin{center} \includegraphics[width=10em]{fig/success_3.pdf} \includegraphics[width=10em]{fig/fail_3.pdf} \caption{Part3: Success (left) and Failure (right)} \label{fig:p3_succ} \end{center} \end{figure} \begin{figure}[H] \begin{center} \includegraphics[width=12em]{fig/error_depth_p3.pdf} \includegraphics[width=12em]{fig/trees_p3.pdf} \caption{Classification error varying trees depth (left) and numbers of trees (right)} \label{fig:p3_trees} \end{center} \end{figure} \begin{figure}[H] \begin{center} \includegraphics[width=18em]{fig/p3_rand.pdf} \caption{Effect of randomness parameter on classification error} \label{fig:p3_rand} \end{center} \end{figure} \begin{figure}[H] \begin{center} \includegraphics[width=12em]{fig/p3_vocsize.pdf} \includegraphics[width=12em]{fig/p3_time.pdf} \caption{Effect of vocabulary size; classification error (left) and time (right)} \label{fig:p3_voc} \end{center} \end{figure} \begin{figure}[H] \begin{center} \includegraphics[width=18em]{fig/p3_colormap.pdf} \caption{Varying leaves and estimators: effect on accuracy} \label{fig:p3_colormap} \end{center} \end{figure} # Comparison of methods and conclusions Overall K-means achieves slightly better accuracy that the RF-codebook at the expense of a higher execution time for training **(and testing???)**. As discussed in section I, due to the initialization process for optimal centroids placements, K-means can result unpreferable for large descriptors' sizes (in absence of methods for dimensionality reduction), and in many cases the increase in training time would not justify the minimum increase in classification performance. For Caltech_101 RF-codebook seems to be the most suitable method to perform RF-classification. It is observable that for the particular dataset we are analysing the class *water_lilly* is the one that gets misclassified the most, both in k-means and RF-codebook (refer to figures \ref{fig:km_cm} and \ref{fig:p3_cm}. This means that the features obtained from this class do not guarantee very discriminative splits, hence the first splits in the trees will prioritize features taken from other classes. # References