English

Uncertain but Useful: Leveraging CNN Variability into Data Augmentation

Numerical Analysis 2025-09-08 v1 Artificial Intelligence Numerical Analysis

Abstract

Deep learning (DL) is rapidly advancing neuroimaging by achieving state-of-the-art performance with reduced computation times. Yet the numerical stability of DL models -- particularly during training -- remains underexplored. While inference with DL is relatively stable, training introduces additional variability primarily through iterative stochastic optimization. We investigate this training-time variability using FastSurfer, a CNN-based whole-brain segmentation pipeline. Controlled perturbations are introduced via floating point perturbations and random seeds. We find that: (i) FastSurfer exhibits higher variability compared to that of a traditional neuroimaging pipeline, suggesting that DL inherits and is particularly susceptible to sources of instability present in its predecessors; (ii) ensembles generated with perturbations achieve performance similar to an unperturbed baseline; and (iii) variability effectively produces ensembles of numerical model families that can be repurposed for downstream applications. As a proof of concept, we demonstrate that numerical ensembles can be used as a data augmentation strategy for brain age regression. These findings position training-time variability not only as a reproducibility concern but also as a resource that can be harnessed to improve robustness and enable new applications in neuroimaging.

Keywords

Cite

@article{arxiv.2509.05238,
  title  = {Uncertain but Useful: Leveraging CNN Variability into Data Augmentation},
  author = {Inés Gonzalez-Pepe and Vinuyan Sivakolunthu and Yohan Chatelain and Tristan Glatard},
  journal= {arXiv preprint arXiv:2509.05238},
  year   = {2025}
}
R2 v1 2026-07-01T05:23:26.141Z