重新想象线性探针:克莱尼-阿诺德网络在迁移学习中的应用
机器学习
2024-09-13 v1 人工智能
计算机视觉与模式识别
摘要
本文介绍了将克莱尼-阿诺德网络(Kolmogorov-Arnold Networks, KAN)引入迁移学习中的线性探针方法,以克服传统线性探针在建模数据中复杂关系方面的局限性。线性探针通常应用于预训练模型的最后一层,受限于其无法建模数据中的复杂关系。为此,我们提出用 KAN 替代线性探针层,KAN 利用样条基函数来逼近复杂函数。在本研究中,我们将 KAN 集成到在 ImageNet 上预训练的 ResNet-50 模型中,并在 CIFAR-10 数据集上进行评估。我们进行了系统的超参数搜索,重点关注网格大小和样条阶数(k),以优化 KAN 的灵活性和准确性。我们的结果表明,KAN 在各种配置下均一致优于传统线性探针,实现了准确率和泛化能力的显著提升。这些发现表明,KAN 提供了一种比传统线性探针技术更强大、更具适应性的替代方案。
引用
@article{arxiv.2409.07763,
title = {Reimagining Linear Probing: Kolmogorov-Arnold Networks in Transfer Learning},
author = {Sheng Shen and Rabih Younes},
journal= {arXiv preprint arXiv:2409.07763},
year = {2024}
}
备注
10 pages, 5 figure