\begin{figure}[!h]
    \centering
    \includegraphics[width=\textwidth]{figures/Clustering_v4.pdf}
    \vspace{-15mm}
    \caption{\textbf{t-SNE Visualization on NCT-CRC Test Set.} We visualize the clustered test set samples in the NCT-CRC~\cite{Kather2019-xy} dataset using t-SNE. The test input is rotated before feeding into each model. We compare the visualization between (top) standard ResNet~\cite{He2016-de}, (middle) SRENet, and (bottom) E2CNN~\cite{Weiler2019-yf}  trained with no rotational augmentation. We colorize the samples in the clustered results according to their classification label.}
    \label{fig:cluster}  
    % \vspace{-3mm}
\end{figure}  