9行代码提高少样本学习泛化能力,代码已开源 | ICLR2021 Oral

简介: 本文介绍一篇最新发表在ICLR2021 Oral上的少样本学习工作,他们尝试从数据分布估计的角度去缓解少样本学习中的过拟合现象,并提出通过分布矫正(估计)的方式弥合这种差距。

微信图片_20220112125830.jpg


链接:https://openreview.net/forum?id=JWOiYxMG92s


代码:https://github.com/ShuoYang-1998/ICLR2021-Oral_Distribution_Calibration


简介


从极少量样本中学习到泛化性能良好的模型是很困难的,因为极少的样本形成的数据分布往往与真实数据分布相差较大,在偏斜的数据分布上训练模型会导致严重的过拟合现象并严重破坏模型的泛化能力(见图1)。


在本文中我们尝试从数据分布估计的角度去缓解少样本学习中的过拟合现象,利用一个样本去估计该类别的整体数据分布,如果该分布估计足够准确,也许可以弥合少样本学习和传统多样本学习的差距。


微信图片_20220112125833.jpg


方法


直接从一个样本中估计整体数据分布是非常困难的,需要很强的先验去约束分布估计的过程。我们观察到如果假设每一个类别的特征都服从高斯分布,那么相似类别的分布统计量相似度非常高,如表1。


微信图片_20220112125834.jpg


从直观的角度理解,一个类别的mean代表该类别的general appearance,variance代表该类别某属性的变化范围(颜色、形状、姿势等)。


而相似的类别(如猫和老虎)具有相似的整体外观和相似的属性变化范围。受此启发,我们提出了通过迁移基类(base class)的分布统计量的方式对少样本类别的数据分布做‘矫正’(calibration)。


具体来说,我们首先为每一个base class i计算一个mean和covariance:


计算好的 和 储存起来当作base class分布先验。然后在进行少样本分类时我 们利用base class的分布先验去修正少样本类别的数据分布:得到修正后的少样本类别的分布 和 后,我们便可以从修正后的分布中直接采样:然后利用采样得到的数据和support set共同训练分类器:


至此,该算法结束。


流程如图:


微信图片_20220112125836.jpg


实验


我们的算法无需任何可训练参数,可以建立在任何已有的特征提取器和分类器之上,并极大的提高模型的泛化能力。


代码已开源,核心代码只有9行(evaluate_DC.py中的第10-19行)。


我们的方法搭配最简单的线性分类器便可以达到非常高的1-shot分类性能。


实验结果如图:


微信图片_20220112125838.jpg微信图片_20220112125839.jpg微信图片_20220112125841.jpg


我们分布估计的可视化如图:


微信图片_20220112125843.jpg


总结

在本工作中我们思考了少样本学习和多样本学习的核心差距,并提出通过分布矫正(估计)的方式弥合这种差距。在该工作的后续期刊拓展版本中我们从generalization error bound的角度为‘基于数据分布估计的少样本学习’这一类方法建立了理论框架,并证明了当数据分布足够准确时,少样本学习和多样本学习的泛化误差等价。


知乎链接:https://zhuanlan.zhihu.com/p/344531704

相关文章
|
人工智能 资源调度 算法
AI 绘画Stable Diffusion 研究(八)sd采样方法详解
AI 绘画Stable Diffusion 研究(八)sd采样方法详解
3782 0
毛概期末考试要点总结
毛概期末考试要点总结
1830 0
|
Python
解决pyinstaller不兼容python-docx的方法
python-docx是一个python的读写word的库,可以用来读写word文档,向word文档里插入表格,但是与pyinstaller不是原生兼容,需要小改一下
1412 0
解决pyinstaller不兼容python-docx的方法
|
11月前
南京观海微电子---PCIe协议接口
本文介绍了PCIe总线的引脚定义、协议架构及TLP数据包传输机制。涵盖物理层、数据链路层与事务层的功能,详解TLP封装与解包流程,以及各层在数据传输中的作用。
南京观海微电子---PCIe协议接口
|
12月前
|
数据采集 机器学习/深度学习 运维
2025年设备管理系统盘点:如何借助智能运维减少非计划停机与维护成本?
在制造业数字化转型中,设备管理面临非计划停机与高维护成本挑战。本文基于2025年调研,解析树根科技根云平台在IIoT接入、预测性维护、知识管理等方面的领先实践,助力企业实现高效智能运维。
560 0
|
机器学习/深度学习 人工智能 自然语言处理
基于星海智算云平台部署 DeepSeek-R1系列 70b 模型全攻略(附平台福利)
本文介绍了如何在星海智算云平台上部署DeepSeek-R1系列70B模型,解决官网访问不畅的问题。通过云端部署,用户可以按需付费,避免本地部署高昂成本(高达两百多万)。文章详细讲解了从实例创建到开始使用DeepSeek的八个步骤,并提供了成本优化技巧和新手注意事项。推荐使用双A100显卡,每小时费用仅13.32元。新用户还可领取福利,享受高性价比服务。立即注册体验:[星海智算云平台](https://gpu.spacehpc.com/user/register?inviteCode=52872508)。
1523 1
基于星海智算云平台部署 DeepSeek-R1系列 70b 模型全攻略(附平台福利)
|
人工智能 测试技术 程序员
3天功能开发→3小时:通义灵码2.0+DEEPSEEK实测报告,单元测试生成准确率92%的秘密
通义灵码2.0是阿里巴巴推出的一款智能编程辅助工具,借助AI的强大能力,为开发者提供高效的代码生成、单元测试自动生成及跨语言编程支持。其核心亮点包括: 通义灵码2.0不仅提升了开发效率,还降低了编程门槛,帮助初学者快速上手。强烈推荐给所有开发者! [了解更多](https://hyk52syx.blog.csdn.net/article/details/145642464?spm=1001.2014.3001.5502)
1907 23
|
存储 人工智能 数据处理
阿里云CTO周靖人:全面投入升级AI大基建
9月19日,在2024杭州云栖大会上,阿里云CTO周靖人表示,阿里云正在围绕AI时代,树立一个AI基础设施的新标准,全面升级从服务器到计算、存储、网络、数据处理、模型训练和推理平台的技术架构体系,让数据中心成为一台超级计算机,为每个AI和应用提供高性能、高效的算力服务。
24210 15
|
机器学习/深度学习 安全 固态存储
【YOLOv8改进 - 注意力机制】LS-YOLO MSFE:新颖的多尺度特征提取模块 | 小目标/遥感
YOLO系列目标检测模型的新发展,LS-YOLO专为滑坡检测设计。它使用多尺度滑坡数据集(MSLD)和多尺度特征提取(MSFE)模块,结合ECA注意力,提升定位精度。通过改进的解耦头,利用膨胀卷积增强上下文信息。在滑坡检测任务中,LS-YOLO相对于YOLOv5s的AP提高了2.18%,达到97.06%。论文和代码已开源。
|
关系型数据库 MySQL 数据库
龙蜥操作系统上安装MySQL:步骤详解与常见问题解决
龙蜥操作系统上安装MySQL:步骤详解与常见问题解决
2065 0

热门文章

最新文章