\begin{table}[t]
\centering
\setlength\tabcolsep{3pt}

% For rotated text
\newcommand{\rottext}[1]{\rotatebox[origin=c]{90}{\textbf{#1}}}
\begin{tabular*}{\textwidth}{@{\extracolsep{\fill}} c l rrr r}
    & \multirow{2}{*}{\textbf{Method}} 
      & \multicolumn{3}{c}{Stage Classification} 
      & Survival Prediction \\
    \cmidrule(lr){3-5} \cmidrule(lr){6-6}
    & & AUC $\uparrow$ & F1$_m$ $\uparrow$ & Bal. Acc $\uparrow$ & C-Index $\uparrow$ \\
    \toprule
    
    % -------- TCGA-BRCA --------
    \multirow{8}{*}{\rottext{TCGA-BRCA}} 
    & \multicolumn{5}{l}{\textit{Comparable Baselines}} \\
    & DeepSets & 52.3$^\ast$(3.05) & 15.8$^\ast$ (5.50) & 25.4 (0.10) & 48.9$^\ast$(2.47) \\
    & ABMIL    & 61.8$^\ast$(5.64) & 20.2$^\ast$ (2.13) & 26.0 (1.11) & \underline{61.7} (1.81) \\
    & GraphTransformer & \underline{63.3} (2.33) & 18.6$^\ast$ (0.89) & 24.9 (0.14) & 52.7$^\ast$(5.61) \\
    & DM-GNN & 62.5$^\ast$(2.19) & \underline{23.2}$^\ast$ (5.40) & \textbf{28.6} (4.34) & 59.9 (5.22) \\
    & Ours     & \textbf{67.2} (3.08) & \textbf{28.0} (8.24) & \underline{27.0} (2.54) & \textbf{62.9} (3.67) \\
    \cmidrule(lr){2-6} 
    & \multicolumn{5}{l}{\textit{Foundation model}} \\
    & UNI2-h   & 69.8 (3.05) & 30.1 (4.45) & 31.0$^\ast$(2.78) & 62.1 (6.90) \\
    \midrule

    % -------- TCGA-UCEC --------
    \multirow{8}{*}{\rottext{TCGA-UCEC}} 
    & \multicolumn{5}{l}{\textit{Comparable Baselines}} \\
    & DeepSets & 49.4$^\ast$(1.48) & 9.46$^\ast$(6.47) & 24.4 (1.67) & 50.3$^\ast$(0.60) \\
    & ABMIL    & 49.0$^\ast$(2.61) & 18.8 (0.00) & 25.0 (0.00) & \underline{58.3} (5.31) \\
    & GraphTransformer & 54.1 (4.51) & 18.7 (1.18) & 25.2 (0.58) & 57.5 (7.84) \\
    & DM-GNN & \underline{55.3} (3.82) & \textbf{20.6} (2.61) & \underline{26.3} (3.19) & 55.9 (8.38) \\
    & Ours     & \textbf{56.4} (4.17) & \underline{19.5} (1.19) & \textbf{26.5} (2.37) & \textbf{60.0} (0.97) \\
    \cmidrule(lr){2-6} 
    & \multicolumn{5}{l}{\textit{Foundation Model}} \\
    & UNI2-h   & 62.7$^\ast$(5.03) & 28.0$^\ast$(4.67) & 30.4$^\ast$(2.92) & 62.8 (5.38) \\
    \bottomrule
\end{tabular*}

\caption{Performance on TCGA-BRCA and TCGA-UCEC. Best performance among comparable baselines is shown in \textbf{bold}, second best \underline{underlined}. UNI2-h is listed separately as it was pretrained on over 350,000 external WSIs. Asterisks (*) denote statistically significant differences ($p<0.05$) to our method.}
\label{tab:results-combined}
\end{table}
