arrow
返回

Augmentation invariant manifold learning

delete2025-02-07
delete0
PRE
AI
S
Shulei Wang *
DOI:10.1093/jrsssb/qkaf003delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Data augmentation is a widely used technique and an essential ingredient in the recent advance in self-supervised representation learning. By preserving the similarity between augmented data, the resulting data representation can improve various downstream analyses and achieve state-of-the-art performance in many applications. Despite the empirical effectiveness, most existing methods lack theoretical understanding under a general nonlinear setting. To fill this gap, we develop a statistical framework on a low-dimensional product manifold to model the data augmentation transformation. Under this framework, we introduce a new representation learning method called augmentation invariant manifold learning and design a computationally efficient algorithm by reformulating it as a stochastic optimization problem. Compared with existing self-supervised methods, the new method simultaneously exploits the manifold's geometric structure and invariant property of augmented data and has an explicit theoretical guarantee. Our theoretical investigation characterizes the role of data augmentation in the proposed method and reveals why and how the data representation learned from augmented data can improve the k-nearest neighbour classifier in the downstream analysis, showing that a more complex data augmentation leads to more improvement in downstream analysis. Finally, numerical experiments on simulated and real data sets are presented to demonstrate the merit of the proposed method.
Keyword:
data augmentation
deep representation learning
manifold learning
k-nearest neighbour
self-supervised learning

期刊

J
Journal of the Royal Statistical Society Series B-Statistical Methodology
IF:
3.6
论文数:
1.5K
被引数:
3.2W

机构

University of Illinois System 封面图
University of Illinois System
学者数:
6.9W
论文数: 6.2W
被引数: 644
引用论文

引用论文

err
IF0
err
err0
PREAI
err
err分享
err收藏
YOUNG CHILDREN, NEW MEDIA
err2007-02-01
err0
PREAI
errEllen Wartella; Michael Robb
err分享
err收藏
Electron spin resonance study of elementary reactions of fluorine atoms
err2002-05-01
err0
PREAI
errEdward L. Cochran; Frank J. Adrian; Vernon A. Bowers
err分享
err收藏
学者 查看更多内容