4 个月前

自最优传输特征变换

自最优传输特征变换

摘要

自最优传输(Self-Optimal-Transport, SOT)特征变换旨在提升数据实例的特征集,以促进下游匹配或分组相关任务。变换后的特征集编码了实例特征之间丰富的高阶关系。变换特征之间的距离不仅捕捉了它们直接的原始相似性,还反映了它们对集合中其他特征相似性的第三方共识。一个特定的最小成本最大流分数匹配问题,其熵正则化版本可以通过最优传输(Optimal Transport, OT)优化进行近似求解,从而产生了我们的导出变换,该变换具有高效性、可微性、等变性、无参数性和概率解释性。实验证明,该变换在使用上非常有效且灵活,在多种任务和训练方案中插入时能持续改进网络性能。我们通过无监督聚类问题展示了其优点,并证明了它在小样本分类中的高效性和广泛适用性,取得了最先进的结果,并在大规模人员重识别任务中表现出色。

代码仓库

danielshalam/bpa
官方
pytorch

基准测试

基准方法指标
few-shot-image-classification-on-cifar-fs-5PT+MAP+SF+SOT (transductive)
Accuracy: 89.94
few-shot-image-classification-on-cifar-fs-5-1PT+MAP+SF+SOT (transductive)
Accuracy: 92.83
few-shot-image-classification-on-cub-200-5PT+MAP+SF+SOT (transductive)
Accuracy: 97.12
few-shot-image-classification-on-cub-200-5-1PT+MAP+SF+SOT (transductive)
Accuracy: 95.80
few-shot-image-classification-on-mini-2PT+MAP+SF+SOT (transductive)
Accuracy: 85.59
few-shot-image-classification-on-mini-3PT+MAP+SF+SOT (transductive)
Accuracy: 91.34

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
自最优传输特征变换 | 论文 | HyperAI超神经