Updated results, hebrew, abstract, intro, etc.

This commit is contained in:
2026-08-30 06:57:52 +03:00
parent 24d6fac79c
commit 275e29bde0
13 changed files with 353 additions and 268 deletions
@@ -3,7 +3,7 @@
\label{sec:discussion}
\addcontentsline{tocheb}{section}{\protect\numberline{\secnumforhebrewtoc}{דיון}}
\subsection{Noise Robustness of \textsc{Roll}-AoC}
\subsection{Noise Robustness of \textsc{Roll}-AOC}
\label{sec:discussion:noise}
\addcontentsline{tocheb}{subsection}{\protect\numberline{\subsecnumforhebrewtoc}{עמידות לרעש של ROLL-AOC}}
@@ -23,7 +23,7 @@
% strengthening the hypothesis that this is a property of the objective rather than an artifact
% of a specific dataset.
The data shows that \textsc{Roll}-AoC is a loss robust to label noise. In KEEL, we see that under the heaviest noise rate, \textsc{Roll}-AoC degrades least. In CiFAR10n, \textsc{Roll}-AoC suffers the least as label noise gets worse. Across other large datasets, \textsc{Roll}-AoC shows to be either competitive or leading. The experiments used a wide variety of data, both tabular and image-based, with varying levels of noise.
The data shows that \textsc{Roll}-AOC is a loss robust to label noise. In KEEL, we see that under the heaviest noise rate, \textsc{Roll}-AOC degrades least. In CIFAR-10N, \textsc{Roll}-AOC suffers the least as label noise gets worse. Across other large datasets, \textsc{Roll}-AOC shows to be either competitive or leading. The experiments used a wide variety of data, both tabular and image-based, with varying levels of noise.
Having the ability to optimize the desired metric (AUC) directly, rather than in a round about way, allows
our method to overcome challenging data. When using a per-sample loss, optimizing for a mislabeled sample
@@ -39,17 +39,17 @@ from NLP-extracted labels). The same advantage appearing across tabular data, na
images strengthens the case that noise robustness is a structural property of the distributional objective,
rather than an artifact of any particular dataset or noise type.
\subsection{\textsc{ROLL}-AoC Versus \textsc{GCE} and \textsc{MAE}}
\subsection{\textsc{ROLL}-AOC Versus \textsc{GCE} and \textsc{MAE}}
\label{sec:discussion:gce-mae}
\addcontentsline{tocheb}{subsection}{\protect\numberline{\subsecnumforhebrewtoc}{ROLL-AOC מול GCE ו-MAE}}
\textsc{GCE} and \textsc{MAE} are used specifically for their robustness to noise, yet under 20\%
asymmetric noise on KEEL both degrade more than \textsc{Roll}-AoC (\textsc{GCE}: $-0.056$,
\textsc{MAE}: $-0.071$, \textsc{Roll}-AoC: $-0.033$). Whether this reflects a need for domain-specific
re-calibration or a fundamental limitation of per-sample objectives under label noise, \textsc{Roll}-AoC
asymmetric noise on KEEL both degrade more than \textsc{Roll}-AOC (\textsc{GCE}: $-0.056$,
\textsc{MAE}: $-0.071$, \textsc{Roll}-AOC: $-0.033$). Whether this reflects a need for domain-specific
re-calibration or a fundamental limitation of per-sample objectives under label noise, \textsc{Roll}-AOC
overcomes these challenges without any noise-specific adjustment.
\subsection{\textsc{Roll}-AoC Versus \textsc{LibAUC}: Stability Under Noise}
\subsection{\textsc{Roll}-AOC Versus \textsc{LibAUC}: Stability Under Noise}
\label{sec:discussion:libauc}
\addcontentsline{tocheb}{subsection}{\protect\numberline{\subsecnumforhebrewtoc}{ROLL-AOC מול LibAUC: יציבות תחת רעש}}
@@ -66,26 +66,26 @@ overcomes these challenges without any noise-specific adjustment.
% Frame as a practical stability advantage: in noisy or heterogeneous settings,
% PESG-based optimization may be brittle in ways that a standard Adam + KDE loss is not.
While \textsc{Roll}-AoC shows improvement over per-sample losses on noisy datasets, it also shows
better performance than our LibAUC baseline, which also optimizes for the same metric directly.
While \textsc{Roll}-AOC shows improvement over per-sample losses on noisy datasets, it also outperforms
our LibAUC baseline, which also optimizes for AUC directly.
This improvement could be explained by \textsc{Roll}-AoC better tracking the real AUC value.
LibAUC's loss uses an imperfect surrogate loss. The AUC metric itself can only be acheived
exactly when using a stepwise function instead of the pairwise surrogate loss, which is impossible to use.
The surrogate losses used is the squared pairwise surrogate loss given:
This advantage can be explained by how closely each method's loss value tracks the actual AUC.
LibAUC replaces the intractable pairwise indicator $\mathbf{1}[f(x^+) > f(x^-)]$ with a squared surrogate:
\[
\ell(f(x^+), f(x^-)) = \bigl(1 - f(x^+) + f(x^-)\bigr)^2
\]
The squared surrogate and the indicator it replaces follow fundamentally different patterns: the squared
loss is sensitive to the magnitude of score differences, not only to their sign. As a result, the aggregate
loss value gives little information about the AUC actually achieved --- a model with large score margins
on easy pairs can have a low surrogate loss yet mediocre ranking quality on hard pairs.
$$\ell(f(x^+), f(x^-)) = \bigl(1 - f(x^+) + f(x^-)\bigr)^2$$
While this helps the math simplify in order to reduce the computation, the connection to the
actual AUC value is lost, and so the metric isn't directly optimized for.
\text{Roll}-AoC does not exactly estimate AUC either, yet the error is limited. As long as the bandwidth
is estimated properly, and the AUC is sampled at enough rates, one can expect a value very close to the real
AUC. These two pre-conditions are non-trivial to acheive, yet hold well enough for the experiments done as
part of this work.
This better tracking presents two benefits. Firstly, the gradients themselves are weighted in a more-exact way,
helping our model converge towards better performance. Secondly, it presents itself in best-epoch selection. Our training process takes the model with the lowest loss. Considering our loss tracks the real AUC better (to be exact, it tracks the area over the curve which complements AUC), this allows our best model selection to be more exact.
\textsc{Roll}-AOC does not track AUC exactly either, but its error is bounded. The loss is the numerical
integral of $1 - \operatorname{TPR}(\alpha)$ over 99 FPR grid points, directly approximating the area
over the ROC curve ($\mathrm{AOC} = 1 - \mathrm{AUC}$). As long as the KDE bandwidth is well-estimated
and 99 thresholds provide sufficient coverage, the loss value stays close to the true AOC.
The practical consequence is twofold: gradient weighting is more accurately aligned with ranking quality,
and best-epoch model selection --- which picks the epoch with the lowest loss --- selects the epoch with
the highest estimated AUC rather than the lowest surrogate value.
\subsection{Failure of the Single-Threshold Variant (\textsc{Roll}-TPR)}
\label{sec:discussion:roll-tpr}
@@ -118,7 +118,7 @@ of its score space, which exacerbated already existing overfitting issues and ma
to random initialization. On datasets such as Wisconsin and New-Thyroid1, this manifested as extreme
run-to-run variance ($\pm 0.441$ and $\pm 0.241$ AUC respectively under clean labels), where the
outcome was determined more by initialization than by the optimization itself.
This stands in direct contrast to \textsc{Roll}-AoC: as noted in \Cref{rem:aoc-coverage}, the 99
This stands in direct contrast to \textsc{Roll}-AOC: as noted in \Cref{rem:aoc-coverage}, the 99
simultaneously active thresholds spread gradient coverage across the full score range, making convergence
far more robust to initialization.
@@ -175,7 +175,7 @@ can perform poorly wherever the score distribution exhibits multiple scales of v
The number of FPR thresholds evaluated per forward pass is itself a trade-off. More thresholds yield a
finer approximation of the ROC curve and, as noted in \Cref{rem:aoc-coverage}, denser gradient coverage
across the score space --- both of which contribute to the robustness of \textsc{Roll}-AoC. Fewer
across the score space --- both of which contribute to the robustness of \textsc{Roll}-AOC. Fewer
thresholds reduce training cost proportionally but produce a coarser AUC estimate and sparser gradient
signal. Each threshold evaluation requires a KDE CDF query and a bisection inversion, each $O(n)$ in
the batch size, making the total per-batch cost $O(kn)$ where $k$ is the number of thresholds. In the
@@ -208,13 +208,13 @@ straightforward mitigation that has not yet been explored.
In this work, we present \textsc{ROLL} as a general framework for differentiable, ROC-based optimization
via parametric score distribution fitting. As the primary instantiation, we develop the KDE-based variant
and provide its full forward pass and gradient derivations. \textsc{ROLL}-AoC directly integrates the
and provide its full forward pass and gradient derivations. \textsc{ROLL}-AOC directly integrates the
estimated ROC curve over 99 evenly-spaced FPR thresholds --- a fundamentally different approach from
\textsc{LibAUC}'s pairwise formulation, which optimizes a squared surrogate over score pairs rather than
the curve itself.
We evaluated \textsc{ROLL}-AoC against losses designed for noise robustness and class imbalance. Across
datasets spanning tabular and image data with varying noise types and rates, we show that \textsc{ROLL}-AoC
We evaluated \textsc{ROLL}-AOC against losses designed for noise robustness and class imbalance. Across
datasets spanning tabular and image data with varying noise types and rates, we show that \textsc{ROLL}-AOC
handles these challenges competitively, often leading, despite noise robustness not being an explicit
design objective.
@@ -222,9 +222,9 @@ design objective.
\label{sec:conclusions:limitations}
\addcontentsline{tocheb}{subsection}{\protect\numberline{\subsecnumforhebrewtoc}{מגבלות}}
\textsc{Roll}-AoC has several practical limitations worth noting.
\textsc{Roll}-AOC has several practical limitations worth noting.
The KDE instantiation uses a single bandwidth per class, which cannot accurately represent multimodal or heavy-tailed score distributions.
Computational cost is significant: evaluating 99 FPR thresholds per forward pass makes \textsc{Roll}-AoC considerably more expensive per training step than per-sample losses or \textsc{LibAUC}'s pairwise formulation.
Computational cost is significant: evaluating 99 FPR thresholds per forward pass makes \textsc{Roll}-AOC considerably more expensive per training step than per-sample losses or \textsc{LibAUC}'s pairwise formulation.
The single-threshold variant (\textsc{Roll}-TPR) is brittle in practice; gradient locality concentrates updates near one operating point and the benefit rarely transfers to held-out data.
Finally, the framework currently lacks formal convergence guarantees; the empirical results are encouraging but not theoretically grounded.
@@ -242,4 +242,4 @@ Variable-bandwidth (adaptive) KDE methods address the single-bandwidth limitatio
Formal convergence guarantees for KDE-ROLL under standard assumptions (batch size, bandwidth decay schedule, smoothness) remain an open problem, and would also clarify how the bandwidth scheduler interacts with gradient variance across training.
\paragraph{Partial and weighted AUC.}
Restricting or reweighting the FPR grid to emphasize specific operating regions is a straightforward modification to \textsc{Roll}-AoC, and a natural fit for settings such as medical imaging where only a narrow FPR range is clinically actionable.
Restricting or reweighting the FPR grid to emphasize specific operating regions is a straightforward modification to \textsc{Roll}-AOC, and a natural fit for settings such as medical imaging where only a narrow FPR range is clinically actionable.