纠错数据标注,只需一行代码:开源项目Cleanlab发布了2.0版本

简介: 纠错数据标注,只需一行代码:开源项目Cleanlab发布了2.0版本
只需要几行代码,Cleanlab 就能帮你纠正数据集中的错误。


从事 AI 研究工作的人都知道,数据准备几乎占据了数据科学和机器学习研究工作的 80%。它被认为是最耗时和最不愉快的数据科学任务。

你以为的研究工作是「花时间从数据、训练模型、高级建模技术中探索出很棒的见解」,实际上经常是「把大量时间花在清理数据上」,因为现实世界的数据是杂乱无章的,而且充满错误……


数据错误(例如训练集中的错误标记示例)会降低模型性能,数据集级别的问题(如重叠类)也会降低模型性能。即使在 gold-standard 基准数据集中,测试集错误也很常见,这可能会误导数据科学家选择劣质模型进行部署。虽然探索复杂的建模技术听起来比手动检查和清理单个数据点更有吸引力,但往往是后者提供了更大的收益。

为了帮助数据集纠错效率的提升,来自 MIT、亚马逊的研究者创造了数据标注纠错工具 Cleanlab。Cleanlab 通过仅自动标记真正需要注意的一小部分数据来减少此过程中的痛苦。


这个工具在三人合著的 NeurIPS 2021 论文《Pervasive Label Errors in Test Sets Destabilize Machine Learning Benchmarks》(ImageNet 验证集 6% 的标签都是错的,MIT:十大常用数据集没那么靠谱)也有提到。


论文链接:https://arxiv.org/abs/2103.14749

在这篇文章中,三人对 10 个主流机器学习数据集的测试集展开了研究,发现它们的平均错误率竟高达 3.4%。其中,最有名的 ImageNet 数据集的验证集中至少存在 2916 个错误,错误率为 6%;QuickDraw 数据集中至少存在 500 万个错误,错误率为 10%。

既然数据标注错误如此普遍,但又十分重要,那 cleanlab 是怎么解决这个问题的呢?

实现 80% 的自动化

Cleanlab 通过提供一个框架来简化以数据为中心的 AI ,帮助数据科学家和 ML 工程师完成 80% 的工作。Cleanlab 通过查找和修复示例级、类级和数据集级问题,支持机器学习和分析工作流,处理混乱的现实世界数据;测量和跟踪整体数据集质量;并为机器学习管道提供清洁数据。

「Cleanlab 背后的算法理论受到了量子信息理论的启发,当时我们的 CEO 正在麻省理工学院进行博士研究。我们的一些用户认为 Cleanlab 是黑魔法,但它大部分是发表在顶级 ML/AI 会议和期刊上的数学和科学研究。」

2021 年底,Cleanlab 公司成立。在过去的一年里,数十家科技、医疗保健、金融和数据相关的公司(例如特斯拉、摩根大通、Chase、富国银行、微软等)已经开始使用 cleanlab。迄今,这个项目已经累积了 3k Star:


近日,Cleanlab 正式发布了 2.0 版本,以适用于所有数据科学家、ML 数据集和模型。


cleanlab 2.0

cleanlab 2.0 是一个开源框架,用于机器学习和分析杂乱的真实数据。基于 MIT 的研究,cleanlab 可以识别数据集中的错误,测量数据集质量,用噪声数据训练可靠模型,并帮助管理高质量的数据集,每一个都只需要几行代码。

项目地址:https://github.com/cleanlab/cleanlab


Cleanlab 2.0 版本中开源的新特性示例(大部分是一行代码)包括:

  • 在数据集中查找问题并按质量对数据点进行排名
  • 在有标签问题的任何数据集上训练任何分类器
  • 在数据集级别查找要合并和 / 或删除的重叠类
  • 衡量数据集的整体标签健康状况


基本只需要一行代码,即可找出数据集中的哪些示例存在问题:



from cleanlab.classification import CleanLearningissues = CleanLearning(yourFavoriteModel).find_label_issues(data, labels)


一行代码,就能衡量和跟踪数据集的整体健康状况:



from cleanlab.dataset import overall_label_health_scorehealth = overall_label_health_score(labels, pred_probs)


此外,cleanlab 的所有功能都适用于任何数据集和任何模型,包括 scikit-learn、PyTorch、Tensorflow、Keras、JAX、HuggingFace、MXNet、XGBoost 等。如果你使用与 sklearn 兼容的分类器,cleanlab 可以开箱即用。

更多细节可参考项目文档。

相关文章
|
机器学习/深度学习 计算机视觉 网络架构
改进YOLOv8:添加CBAM注意力机制(涨点明显)
改进YOLOv8:添加CBAM注意力机制(涨点明显)
9440 1
|
SQL 资源调度 数据库
数仓学习---14、大数据技术之DolphinScheduler
数仓学习---14、大数据技术之DolphinScheduler
|
3月前
|
人工智能 监控 网络安全
2026年开工不想工作?3分钟部署OpenClaw(原Clawdbot)AI助手 24小时为我工作
春节假期结束,不少人刚回到工位就陷入“开工倦怠”:不想回邮件、不想整理报表、不想写代码、不想处理重复琐碎的事务,只想发呆摸鱼,但工作任务堆成山, deadlines 步步紧逼。有没有一种方式,能让AI替我们完成大部分机械性、重复性工作,实现“人在摸鱼,工作已完成”的理想状态?答案就是**阿里云OpenClaw(原Clawdbot)**——一款能7×24小时不间断工作的开源AI自动化助手,无需人工盯守,一句话下达指令,就能自动执行办公、开发、搜索、整理、定时任务等全场景操作。
482 9
|
5月前
|
人工智能 自然语言处理 安全
双第一,阿里云领跑安全运营智能体
全球权威咨询机构IDC发布了《中国安全运营智能体实测,2025》(Doc#CHC52346025,2025年11月)报告,报告针对国内20余家云厂商和安全厂商,从安全风险评估智能体、告警分诊智能体、事件调查与响应智能体、策略与规则管理智能体、威胁情报的收集与分析智能体、漏洞管理智能体、安全报告智能体、智能体管理等八大实测维度进行测评,整个测评流程依据IPDRR安全运营框架进行了严格测试。
|
JSON 人工智能 数据格式
AI计算机视觉笔记二十六:YOLOV8自训练关键点检测
本文档详细记录了使用YOLOv8训练关键点检测模型的过程。首先通过清华源安装YOLOv8,并验证安装。接着通过示例权重文件与测试图片`bus.jpg`演示预测流程。为准备训练数据,文档介绍了如何使用`labelme`标注工具进行关键点标注,并提供了一个Python脚本`labelme2yolo.py`将标注结果从JSON格式转换为YOLO所需的TXT格式。随后,通过Jupyter Notebook可视化标注结果确保准确性。最后,文档展示了如何组织数据集目录结构,并提供了训练与测试代码示例,包括配置文件`smoke.yaml`及训练脚本`train.py`,帮助读者完成自定义模型的训练与评估。
6041 2
|
12月前
|
机器学习/深度学习 PyTorch 算法框架/工具
高效处理多维数组:einsum()函数从入门到精通
本文深入解析了NumPy中的`einsum()`函数,从基础语法到高级应用全面展开。文章首先介绍了爱因斯坦求和约定的数学基础,解释了`einsum()`如何通过简洁的索引符号实现复杂的多维数组运算。
468 5
高效处理多维数组:einsum()函数从入门到精通
|
机器学习/深度学习 编解码 监控
目标检测实战(六): 使用YOLOv8完成对图像的目标检测任务(从数据准备到训练测试部署的完整流程)
这篇文章详细介绍了如何使用YOLOv8进行目标检测任务,包括环境搭建、数据准备、模型训练、验证测试以及模型转换等完整流程。
26814 59
目标检测实战(六): 使用YOLOv8完成对图像的目标检测任务(从数据准备到训练测试部署的完整流程)
|
消息中间件 监控 Java
RabbitMQ精讲7:与SpringBoot、Spring Cloud Stream整合实战
RabbitMQ精讲7:与SpringBoot、Spring Cloud Stream整合实战
RabbitMQ精讲7:与SpringBoot、Spring Cloud Stream整合实战