% \begin{figure*}[htbp]
%     \centering
%     % First Subfigure: Logit Difference Trajectories
%     \begin{subfigure}[b]{1.0\linewidth}
%         \centering
%         \includegraphics[width=\linewidth]{figures/logit_lens/baseline/Qwen_Qwen3_4B_Instruct_2507_logit_diff_trajectories.png}
%         \caption{Logit Difference Trajectories (Separated by Quadrant)}
%         \label{fig:logit_diff}
%     \end{subfigure}
%     
%     \vspace{0.5cm} % Add spacing between plots
% 
%     % Second Subfigure: Probability Trajectories
%     \begin{subfigure}[b]{1.0\linewidth}
%         \centering
%         \includegraphics[width=\linewidth]{figures/logit_lens/baseline/Qwen_Qwen3_4B_Instruct_2507_prob_trajectories.png}
%         \caption{Probability Trajectories (Valid vs. Invalid)}
%         \label{fig:prob_traj}
%     \end{subfigure}
% 
%     \vspace{0.5cm} % Add spacing between plots
% 
%     % Third Subfigure: Quadrant Comparison
%     \begin{subfigure}[b]{1.0\linewidth}
%         \centering
%         \includegraphics[width=\linewidth]{figures/logit_lens/baseline/Qwen_Qwen3_4B_Instruct_2507_quadrant_comparison.png}
%         \caption{Aggregate Comparison of Logit Differences}
%         \label{fig:quadrant_comp}
%     \end{subfigure}
% 
%     \caption{Layer-wise activation steering analysis for Qwen/Qwen3-4B-Instruct-2507. 
%     (a) shows the logit difference (valid - invalid) trajectories split by truth/plausibility quadrants. 
%     (b) displays the probability trajectories for valid and invalid tokens. 
%     (c) provides a direct comparison of the logit difference means and standard deviations across all four quadrants.}
%     \label{fig:qwen_analysis}
% \end{figure*}

\begin{figure*}[t!] % [t!] encourages placement at the top of the page
    \centering
    
    \begin{subfigure}[b]{0.48\linewidth}
        \centering
        \includegraphics[width=\linewidth]{figures/logit_lens/baseline/Qwen_Qwen3_4B_Instruct_2507_logit_diff_trajectories.png}
        \caption{Logit Difference Trajectories}
        \label{fig:logit_diff}
    \end{subfigure}
    \hfill 
    \begin{subfigure}[b]{0.48\linewidth}
        \centering
        \includegraphics[width=\linewidth]{figures/logit_lens/baseline/Qwen_Qwen3_4B_Instruct_2507_prob_trajectories.png}
        \caption{Probability Trajectories}
        \label{fig:prob_traj}
    \end{subfigure}
    
    \vspace{0.5cm}

    \begin{subfigure}[b]{0.9\linewidth}
        \centering
        \includegraphics[width=\linewidth]{figures/logit_lens/baseline/Qwen_Qwen3_4B_Instruct_2507_quadrant_comparison.png}
        \caption{Aggregate Comparison of Logit Differences}
        \label{fig:quadrant_comp}
    \end{subfigure}

    \caption{Layer-wise activation steering analysis for Qwen/Qwen3-4B-Instruct-2507. 
    (a) and (b) show trajectories for logit difference and probability respectively. 
    (c) provides a direct comparison of the logit difference means across all four quadrants.}
    \label{fig:qwen_analysis}
\end{figure*}