无需标签即可监控自监督学习进度,用嵌入熵和聚类质量评估模型性能。
Label-free Monitoring of Self-Supervised Learning Progress
- 基于无标签数据的嵌入熵和聚类一致性设计评估指标。
- 聚类指标在SimCLR、MoCo-v2上与线性探测准确率相关,但SimSiam不成立。
- 嵌入熵随训练趋于稳定,可能适用于不同架构间的模型比较。
自监督学习(SSL)能有效利用无标签数据构建高层嵌入空间,用于下游任务。然而,现有模型质量监控方法仍依赖标注数据。当SSL应用于新数据域时,标注数据可能不可得。本文提出若干可在无标签数据嵌入上应用的评估指标,并通过与线性探测准确率对比验证其有效性。具体包括:使用k-means聚类并以轮廓系数和聚类一致性衡量聚类质量,以及测量嵌入分布熵。结果发现,随着训练进行,聚类更接近真实标签,但仅在SimCLR和MoCo-v2上,聚类指标与线性探测准确率显著相关,而SimSiam则无此关联。尽管熵在训练初期不稳定,但后期趋于稳定且与线性探测准确率呈现正相关;值得注意的是,SimSiam中熵随训练反而上升,这一异常现象需进一步研究。此外,聚类方法仅适用于同架构模型比较,而熵可能具有架构无关性。
原文摘要 · Abstract (English)
Self-supervised learning (SSL) is an effective method for exploiting unlabelled data to learn a high-level embedding space that can be used for various downstream tasks. However, existing methods to monitor the quality of the encoder -- either during training for one model or to compare several trained models -- still rely on access to annotated data. When SSL methodologies are applied to new data domains, a sufficiently large labelled dataset may not always be available. In this study, we propose several evaluation metrics which can be applied on the embeddings of unlabelled data and investigate their viability by comparing them to linear probe accuracy (a common metric which utilizes an annotated dataset). In particular, we apply $k$-means clustering and measure the clustering quality with the silhouette score and clustering agreement. We also measure the entropy of the embedding distribution. We find that while the clusters did correspond better to the ground truth annotations as training of the network progressed, label-free clustering metrics correlated with the linear probe accuracy only when training with SSL methods SimCLR and MoCo-v2, but not with SimSiam. Additionally, although entropy did not always have strong correlations with LP accuracy, this appears to be due to instability arising from early training, with the metric stabilizing and becoming more reliable at later stages of learning. Furthermore, while entropy generally decreases as learning progresses, this trend reverses for SimSiam. More research is required to establish the cause for this unexpected behaviour. Lastly, we find that while clustering based approaches are likely only viable for same-architecture comparisons, entropy may be architecture-independent.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。