\begin{figure}
    \centering
    \scalebox{1}{
    \includegraphics[width=\columnwidth]{figures/architecture/main_architecture.png}
    }
    \small
    \caption{Illustration of our model \ourmethod. 
    Numbers are IDs of items and words.
    Note that dashed items and words are not considered. \ourmethod\ employs optimal transport to align and fuse user interest factors from ratings and texts.}
    \label{fig:main_architecture}
    % \vspace{-0.5cm}
\end{figure}

% \begin{wrapfigure}{r}{0.5\textwidth}
%     \centering
%     \includegraphics[width=0.5\textwidth]{figures/architecture/main_architecture.png}
%     \caption{Illustration of our model \ourmethod. Dashed squares and circles are not included in deriving user interests. \looseness=-1}
%     \label{fig:main_architecture}
% \end{wrapfigure}