\begin{table*}[htbp]
\captionsetup{font=small,labelfont=small}
\begin{minipage}{0.48\textwidth}
\centering
\caption{Comparison of magnitude of the regularization term $\mathcal{R}(h, \mathcal{S}, \mathcal{T})$ and the magnitude of ${\sf BagCSI}$ loss on IPUMS dataset.}\label{tab:reg_term_mag_ipums}
\scriptsize
\begin{tabular}{l|rr|rr}
Dataset &\multicolumn{2}{c}{PLWFA} &\multicolumn{2}{c}{BLWFA} \\\cmidrule{1-5}
\diagbox{\textbfne{Bag Size}}{\textbfne{Method}} &$\mathcal{R}(h, \mathcal{S}, \mathcal{T})$ &${\sf BagCSI}$ &$\mathcal{R}(h, \mathcal{S}, \mathcal{T})$ &${\sf BagCSI}$ \\\midrule
8 &0.01 &1.59 &0.02 &1.59 \\
32 &0.03 &1.62 &0.03 &1.62 \\
128 &0.04 &1.62 &0.05 &1.62 \\
256 &0.06 &1.62 &0.06 &1.63\\
\end{tabular}
\end{minipage}
\hfill
\begin{minipage}{0.48\textwidth}
\centering
\caption{Comparison of magnitude of the regularization term $\mathcal{R}(h, \mathcal{S}, \mathcal{T})$ and the magnitude of ${\sf BagCSI}$ loss on Criteo dataset.}\label{tab:reg_term_mag_criteo}
\scriptsize
\begin{tabular}{l|rr|rr}
Dataset &\multicolumn{2}{c}{PLWFA} &\multicolumn{2}{c}{BLWFA} \\\cmidrule{1-5}
\diagbox{\textbfne{Bag Size}}{\textbfne{Method}} &$\mathcal{R}(h, \mathcal{S}, \mathcal{T})$ &${\sf BagCSI}$ &$\mathcal{R}(h, \mathcal{S}, \mathcal{T})$ &${\sf BagCSI}$ \\\midrule
64 &17.52 &577.50 &17.44 &569.73 \\
128 &17.46 &611.63 &17.40 &602.30 \\
256 &17.97 &654.50 &17.85 &644.50 \\
512 &18.19 &673.34 &18.22 &662.61 \\
\end{tabular}
\end{minipage}
\end{table*}

