arXiv:2412.02121cs.CV2024-12

用部分信息分解框架重看自监督学习,提升特征表示能力

Rethinking Self-Supervised Learning Within the Framework of Partial Information Decomposition

  • 引入部分信息分解新视角,用联合互信息替代传统互信息
  • 在4个数据集上改进4种基线模型,显著提升特征学习效果
  • 适合关注自监督学习机制、特征表示优化的研究者

自监督学习(SSL)在无标签数据上展现强大特征学习能力,但其内部互信息的作用仍存争议:有的主张增强增强视图间的互信息,有的则建议降低该互信息并增加任务相关信息。本文提出在部分信息分解(PID)框架下重新审视SSL核心思想,将传统互信息替换为更通用的联合互信息以调和矛盾。基于该框架的实例化研究揭示了现有流水线的改进路径,我们提出一个通用改进流程,聚焦提取PID中的独特信息成分,用于低层监督实现通用特征学习,并构建高层监督信号以支持任务相关特征学习。本质上,这相当于同时利用局部与全局聚类。在4个基线模型和4个数据集上的实验验证了方法的有效性与普适性。

原文摘要 · Abstract (English)

Self Supervised learning (SSL) has demonstrated its effectiveness in feature learning from unlabeled data. Regarding this success, there have been some arguments on the role that mutual information plays within the SSL framework. Some works argued for increasing mutual information between representation of augmented views. Others suggest decreasing mutual information between them, while increasing task-relevant information. We ponder upon this debate and propose to revisit the core idea of SSL within the framework of partial information decomposition (PID). Thus, with SSL under PID we propose to replace traditional mutual information with the more general concept of joint mutual information to resolve the argument. Our investigation on instantiation of SSL within the PID framework leads to upgrading the existing pipelines by considering the components of the PID in the SSL models for improved representation learning. Accordingly we propose a general pipeline that can be applied to improve existing baselines. Our pipeline focuses on extracting the unique information component under the PID to build upon lower level supervision for generic feature learning and on developing higher-level supervisory signals for task-related feature learning. In essence, this could be interpreted as a joint utilization of local and global clustering. Experiments on four baselines and four datasets show the effectiveness and generality of our approach in improving existing SSL frameworks.

自监督学习信息分解特征学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。