Learning from Random Subspace Exploration: Generalized Test-Time Augmentation with Self-supervised Distillation

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Jelea, Andrei, Belbachir, Ahmed Nabil, Leordeanu, Marius
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866914226245730304
author Jelea, Andrei
Belbachir, Ahmed Nabil
Leordeanu, Marius
author_facet Jelea, Andrei
Belbachir, Ahmed Nabil
Leordeanu, Marius
contents We introduce Generalized Test-Time Augmentation (GTTA), a highly effective method for improving the performance of a trained model, which unlike other existing Test-Time Augmentation approaches from the literature is general enough to be used off-the-shelf for many vision and non-vision tasks, such as classification, regression, image segmentation and object detection. By applying a new general data transformation, that randomly perturbs multiple times the PCA subspace projection of a test input, GTTA creates valid augmented samples from the data distribution with high diversity, properties we theoretically show that are essential for a Test-Time Augmentation method to be effective. Different from other existing methods, we also propose a final self-supervised learning stage in which the ensemble output, acting as an unsupervised teacher, is used to train the initial single student model, thus reducing significantly the test time computational cost. Our comparisons to strong TTA approaches and SoTA models on various vision and non-vision well-known datasets and tasks, such as image classification and segmentation, pneumonia detection, speech recognition and house price prediction, validate the generality of the proposed GTTA. Furthermore, we also prove its effectiveness on the more specific real-world task of salmon segmentation and detection in low-visibility underwater videos, for which we introduce DeepSalmon, the largest dataset of its kind in the literature.
format Preprint
id arxiv_https___arxiv_org_abs_2507_01347
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Learning from Random Subspace Exploration: Generalized Test-Time Augmentation with Self-supervised Distillation
Jelea, Andrei
Belbachir, Ahmed Nabil
Leordeanu, Marius
Computer Vision and Pattern Recognition
We introduce Generalized Test-Time Augmentation (GTTA), a highly effective method for improving the performance of a trained model, which unlike other existing Test-Time Augmentation approaches from the literature is general enough to be used off-the-shelf for many vision and non-vision tasks, such as classification, regression, image segmentation and object detection. By applying a new general data transformation, that randomly perturbs multiple times the PCA subspace projection of a test input, GTTA creates valid augmented samples from the data distribution with high diversity, properties we theoretically show that are essential for a Test-Time Augmentation method to be effective. Different from other existing methods, we also propose a final self-supervised learning stage in which the ensemble output, acting as an unsupervised teacher, is used to train the initial single student model, thus reducing significantly the test time computational cost. Our comparisons to strong TTA approaches and SoTA models on various vision and non-vision well-known datasets and tasks, such as image classification and segmentation, pneumonia detection, speech recognition and house price prediction, validate the generality of the proposed GTTA. Furthermore, we also prove its effectiveness on the more specific real-world task of salmon segmentation and detection in low-visibility underwater videos, for which we introduce DeepSalmon, the largest dataset of its kind in the literature.
title Learning from Random Subspace Exploration: Generalized Test-Time Augmentation with Self-supervised Distillation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2507.01347