arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2501.08712cs.CVcs.AIcs.LG
收藏

面向等变表示的自监督变换学习

Self-supervised Transformation Learning for Equivariant Representations

Korea Advanced Institute of Science and Technology (KAIST)(韩国科学技术院(KAIST))

Jaemyung Yu, Jaehyun Choi, Dong-Jae Lee, HyeongGwon Hong, Junmo Kim

AI总结:

针对现有等变表示学习依赖变换标签、难处理复杂变换的问题,提出自监督变换学习STL,用图像对的变换表示替代标签,实现图像不变的变换表示与等变变换学习,在多类视觉任务上性能优于现有方法且适配性强。

摘要:

无监督表示学习已显著推动了各类机器学习任务的发展。在计算机视觉领域,当前最优方法利用随机裁剪、颜色抖动等变换来获取不变表示,即使输入经过变换,仍能将语义相同的输入嵌入到同一空间。然而,这会降低定位、花卉分类等需要精确特征的任务性能。为解决该问题,近年研究引入了等变表示学习,以捕捉对变换敏感的信息,但现有方法依赖变换标签,难以处理相互依赖的变换和复杂变换。我们提出自监督变换学习(Self-supervised Transformation Learning, STL),用从图像对中提取的变换表示替代变换标签。该方法确保变换表示具有图像不变性,并学习对应的等变变换,在不增加批次复杂度的前提下提升性能。我们在多种分类和检测任务上验证了该方法的有效性,在11个基准测试中有7个优于现有方法,且在检测任务中表现突出。通过整合AugMix等此前等变方法无法使用的复杂变换,该方法提升了各类任务的性能,凸显了其适应性与鲁棒性。此外,它与多种基础模型兼容,体现了灵活性与广泛适用性。代码可在https://github.com/jaemyung-u/stl获取。

英文摘要:

Unsupervised representation learning has significantly advanced various machine learning tasks. In the computer vision domain, state-of-the-art approaches utilize transformations like random crop and color jitter to achieve invariant representations, embedding semantically the same inputs despite transformations. However, this can degrade performance in tasks requiring precise features, such as localization or flower classification. To address this, recent research incorporates equivariant representation learning, which captures transformation-sensitive information. However, current methods depend on transformation labels and thus struggle with interdependency and complex transformations. We propose Self-supervised Transformation Learning (STL), replacing transformation labels with transformation representations derived from image pairs. The proposed method ensures transformation representation is image-invariant and learns corresponding equivariant transformations, enhancing performance without increased batch complexity. We demonstrate the approach's effectiveness across diverse classification and detection tasks, outperforming existing methods in 7 out of 11 benchmarks and excelling in detection. By integrating complex transformations like AugMix, unusable by prior equivariant methods, this approach enhances performance across tasks, underscoring its adaptability and resilience. Additionally, its compatibility with various base models highlights its flexibility and broad applicability. The code is available at https://github.com/jaemyung-u/stl.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑