Vision Transformers (ViTs) have demonstrated remarkable success in computer vision. However, the lack of inductive biases makes it particularly challenging to train ViTs on small datasets. Two methods are shown to mitigate this: 1) Self-Supervised Learning and Fine-Tuning sequentially (SSL + FT); 2) joint optimization of the Self-Supervised Auxiliary Task (SSAT) and downstream task. In this work, we systematically investigate the effectiveness of SSAT in both training from scratch and fine-tuning settings, further revealing that its success is closely tied to the design of the self-supervised objective within the joint optimization. Specifically, grounded in Information Bottleneck (IB) theory, we show that appropriately constructed self-supervised tasks can provide dual inductive biases that encourage local attention regularization (via compression) and rich semantic encoding (via reconstruction). Motivated by this, we propose a novel multi-view Feature-Guided Consistent Reconstruction (FGCR) method that accelerates learning minimal sufficient representations in IB, allowing ViTs to converge from scratch in small datasets quickly. Our evaluations across diverse datasets demonstrate that FGCR outperforms previous SSAT methods and yields additional gains over SSL + FT methods. Code is available at https://github.com/JAMESKirkQI/FGCR .

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Information Bottleneck Driven Masked Autoencoders for Data-Efficient Auxiliary Learning

  • Qi Cao,
  • Wenzheng Liu,
  • Chen Sun,
  • Jianhao Wei,
  • Jin Zhang

摘要

Vision Transformers (ViTs) have demonstrated remarkable success in computer vision. However, the lack of inductive biases makes it particularly challenging to train ViTs on small datasets. Two methods are shown to mitigate this: 1) Self-Supervised Learning and Fine-Tuning sequentially (SSL + FT); 2) joint optimization of the Self-Supervised Auxiliary Task (SSAT) and downstream task. In this work, we systematically investigate the effectiveness of SSAT in both training from scratch and fine-tuning settings, further revealing that its success is closely tied to the design of the self-supervised objective within the joint optimization. Specifically, grounded in Information Bottleneck (IB) theory, we show that appropriately constructed self-supervised tasks can provide dual inductive biases that encourage local attention regularization (via compression) and rich semantic encoding (via reconstruction). Motivated by this, we propose a novel multi-view Feature-Guided Consistent Reconstruction (FGCR) method that accelerates learning minimal sufficient representations in IB, allowing ViTs to converge from scratch in small datasets quickly. Our evaluations across diverse datasets demonstrate that FGCR outperforms previous SSAT methods and yields additional gains over SSL + FT methods. Code is available at https://github.com/JAMESKirkQI/FGCR .