PyCIL

GitHub
1.1k 159 中等 1 次阅读 昨天NOASSERTION其他开发框架
AI 解读 由 AI 自动生成,仅供参考

PyCIL 是一个专为“类增量学习”打造的 Python 工具箱,基于 PyTorch 构建。在人工智能领域,模型往往面临一个难题:当需要学习新类别的数据时,很容易遗忘旧知识,这种现象被称为“灾难性遗忘”。PyCIL 正是为了解决这一核心痛点而生,它提供了一套标准化的框架,帮助开发者在不重新训练所有历史数据的前提下,让模型持续、稳定地吸收新知识。

这款工具非常适合人工智能研究人员、算法工程师以及高校学生使用。无论是想要复现前沿论文结果,还是希望快速验证自己的增量学习算法,PyCIL 都能极大降低入门门槛和开发成本。其最显著的技术亮点在于“大而全”的算法库:目前它已复现并集成了超过 20 种主流及最新的类增量学习方法,可能是当前开源社区中实现算法数量最多的工具箱之一。此外,项目维护活跃,不仅涵盖了传统方法,还紧跟技术潮流,及时纳入了基于预训练大模型(如 CLIP)的最新研究成果。通过 PyCIL,用户可以轻松进行公平的算法对比与性能评估,是探索持续学习领域不可或缺的得力助手。

使用场景

某自动驾驶初创公司的算法团队需要让车辆识别系统在不断新增交通标志类别(如从仅识别限速牌扩展到识别施工警示、临时改道牌)时,无需重新训练整个模型即可持续学习。

没有 PyCIL 时

  • 复现成本极高:团队需手动查找并复现论文代码,不同增量学习算法(如 iCaRL、DER)的数据加载与训练逻辑差异巨大,耗费数周时间搭建基础框架。
  • 评估标准混乱:缺乏统一的评测脚本,每次对比新算法时,因数据划分或指标计算方式不一致,导致实验结果无法横向公平比较。
  • 灾难性遗忘难控:自研的简单微调策略导致模型在学会新标志后,迅速遗忘旧类别的特征,识别准确率断崖式下跌,且难以快速定位是哪种机制失效。
  • 扩展性差:每引入一种新 SOTA 方法都需要重构大量代码,难以快速验证多种策略组合以应对实际路测中的长尾分布问题。

使用 PyCIL 后

  • 开箱即用:直接调用 PyCIL 内置的 20+ 种复现算法(包括最新的 TagFex 等),通过统一接口即可在几分钟内切换不同策略进行训练,研发周期缩短 80%。
  • 标准化评测:利用工具自带的标准化评估模块,确保所有实验在相同的数据增量设置和指标体系下运行,快速产出可信的对比报告。
  • 有效抑制遗忘:借助 PyCIL 中成熟的回放机制与正则化方法,模型在学习新交通标志时显著保留了对旧类别的记忆,整体平均准确率大幅提升。
  • 灵活迭代:基于模块化设计,团队能轻松组合不同骨干网络与增量策略,快速针对特定路况数据定制最优方案,加速模型落地部署。

PyCIL 将原本繁琐的学术算法复现工作转化为标准化的工程流程,让团队能专注于解决自动驾驶场景下的真实数据挑战而非重复造轮子。

运行环境要求

操作系统
  • 未说明
GPU

未说明 (基于 PyTorch,通常建议配备 NVIDIA GPU 以加速训练)

内存

未说明

依赖
notesREADME 中明确列出的依赖版本较旧(PyTorch 1.8.1),在实际运行时可能需要根据硬件环境调整版本兼容性。运行实验前需编辑 JSON 配置文件和对应的模型 Python 文件来设置超参数。
python3.8+
torch==1.8.1
torchvision==0.6.0
tqdm
numpy
scipy
quadprog
POT
PyCIL hero image

快速开始

PyCIL:面向类增量学习的 Python 工具箱


简介已复现的方法复现结果使用方法许可证致谢联系方式


LICENSEPython PyTorch method CIL visitors

欢迎使用 PyCIL,这或许是目前实现方法最多的类增量学习工具箱。本仓库是论文《PyCIL:面向类增量学习的 Python 工具箱》[论文] 的 PyTorch 代码库。如果您在工作中使用了本仓库中的任何内容,请引用以下 BibTeX 条目:

@article{zhou2023pycil,
    author = {Da-Wei Zhou and Fu-Yun Wang and Han-Jia Ye and De-Chuan Zhan},
    title = {PyCIL: a Python toolbox for class-incremental learning},
    journal = {SCIENCE CHINA Information Sciences},
    year = {2023},
    volume = {66},
    number = {9},
    pages = {197101},
    doi = {https://doi.org/10.1007/s11432-022-3600-y}
  }

@article{zhou2024class,
    author = {Zhou, Da-Wei and Wang, Qi-Wei and Qi, Zhi-Hong and Ye, Han-Jia and Zhan, De-Chuan and Liu, Ziwei},
    title = {Class-Incremental Learning: A Survey},
    journal={IEEE Transactions on Pattern Analysis and Machine Intelligence},
    volume={46},
    number={12},
    pages={9851--9873},
    year = {2024}
}

@inproceedings{zhou2024continual,
    title={Continual learning with pre-trained models: A survey},
    author={Zhou, Da-Wei and Sun, Hai-Long and Ning, Jingyi and Ye, Han-Jia and Zhan, De-Chuan},
    booktitle={IJCAI},
    pages={8363-8371},
    year={2024}
}

最新动态

  • [2026-01]🌟 我们发布了 C3Box,一个基于 CLIP 的类增量学习工具箱。快来试试吧!
  • [2025-07]🌟 查看我们关于使用 CLIP 进行类增量学习的最新工作(ICCV 2025)!
  • [2025-07]🌟 查看我们关于基于预训练模型的类增量学习的最新工作(ICCV 2025)!
  • [2025-07]🌟 查看我们关于使用 PTM 进行领域增量学习的最新工作(ICML 2025)!
  • [2025-04]🌟 新增 TagFex。这是 2025 年的最先进方法!
  • [2025-03]🌟 查看我们关于类增量学习的最新工作(CVPR 2025)!
  • [2025-02]🌟 查看我们关于基于预训练模型的领域增量学习的最新工作(CVPR 2025)!
  • [2025-02]🌟 查看我们关于使用视觉语言模型进行类增量学习的最新工作(TPAMI 2025)!
  • [2024-12]🌟 查看我们关于基于预训练模型的类增量学习的最新工作(AAAI 2025)!
  • [2024-08]🌟 查看我们关于基于预训练模型的类增量学习的最新工作(IJCV 2024)!
  • [2024-07]🌟 查看我们关于类增量学习的严谨且统一的综述(TPAMI 2024),其中引入了一些与记忆无关的度量,并从多个角度进行了全面评估!
  • [2024-06]🌟 查看我们关于类增量学习中全层边距的工作(ICML 2024)!
  • [2024-03]🌟 查看我们关于基于预训练模型的类增量学习的最新工作(CVPR 2024)!
  • [2024-01]🌟 查看我们关于基于预训练模型的持续学习的最新综述(IJCAI 2024)!
  • [2023-09]🌟 我们发布了 PILOT 工具箱,用于基于预训练模型的类增量学习。快来试试吧!
  • [2023-07]🌟 新增 MEMOBEEFSimpleCIL。这些都是 2023 年的最先进方法!
  • [2022-12]🌟 新增 FrTrIL、PASS、IL2A 和 SSRE。
  • [2022-10]🌟 PyCIL 已发表在《SCIENCE CHINA Information Sciences》上(https://link.springer.com/article/10.1007/s11432-022-3600-y)。请查看[官方介绍](https://mp.weixin.qq.com/s/h1qu2LpdvjeHAPLOnG478A)!
  • [2022-08]🌟 新增 RMM。
  • [2022-07]🌟 新增 FOSTER。这是一种仅使用单一主干网络的最先进方法!
  • [2021-12]🌟 征集反馈:我们新增了一个板块,用于介绍使用 PyCIL 的优秀工作。如果您正在使用 PyCIL 在顶级会议或期刊上发表论文,请随时联系我们了解详情!
  • [2021-12]🌟 由于团队成员正致力于其他项目,且代码评审需求十分紧迫,我们将优先评审那些在其论文中明确引用并实现了我们工具箱论文中方法的算法。 请在提交代码之前阅读PR 政策

简介

传统的机器学习系统通常在封闭世界假设下部署,这意味着在离线训练过程中需要使用全部的训练数据。然而,在现实应用中,经常会遇到新类别的不断出现,模型需要能够持续地将这些新类别纳入其中。这种学习范式被称为类别增量学习(CIL)。我们提出了一款Python工具箱,实现了多种用于类别增量学习的关键算法,以减轻机器学习社区研究人员的工作负担。该工具箱不仅包含了EWC和iCaRL等类别增量学习领域的奠基性工作实现,还提供了当前最先进的算法,可用于开展新颖的基础研究。这款名为PyCIL(Python类别增量学习)的工具箱采用MIT许可证,完全开源。

如需了解更多关于增量学习的信息,可以参考以下资料:

  • 关于CIL的简要中文介绍可参见这里
  • 一个包含明确代码和详细解释的PyTorch类别增量学习教程可参见这里

复现的方法

  • FineTune: 基线方法,仅在新任务上更新参数。
  • EWC: 克服神经网络中的灾难性遗忘。PNAS2017 [论文]
  • LwF: 不忘旧知识的学习。ECCV2016 [论文]
  • Replay: 基于样本回放的基线方法。
  • GEM: 用于持续学习的梯度情节记忆。NIPS2017 [论文]
  • iCaRL: 增量分类器与表征学习。CVPR2017 [论文]
  • BiC: 大规模增量学习。CVPR2019 [论文]
  • WA: 在类别增量学习中保持判别能力和公平性。CVPR2020 [论文]
  • PODNet: PODNet:用于小任务增量学习的池化输出蒸馏。ECCV2020 [论文]
  • DER: 用于类别增量学习的动态可扩展表征。CVPR2021 [论文]
  • PASS: 用于增量学习的原型增强与自监督。CVPR2021 [论文]
  • RMM: 用于类别增量学习的强化内存管理。NeurIPS2021 [论文]
  • IL2A: 通过双重增强实现类别增量学习。NeurIPS2021 [论文]
  • ACIL: 基于绝对记忆和隐私保护的解析式类别增量学习。NeurIPS 2022 [论文]
  • SSRE: 用于无样本类别增量学习的自我维持表征扩展。CVPR2022 [论文]
  • FeTrIL: 用于无样本类别增量学习的特征迁移。WACV2023 [论文]
  • Coil: 用于类别增量学习的协同传输。ACM MM2021 [论文]
  • FOSTER: 用于类别增量学习的特征增强与压缩。ECCV 2022 [论文]
  • MEMO: 模型还是603个样本?迈向内存高效的类别增量学习。ICLR 2023 Spotlight [论文]
  • BEEF: BEEF:基于能量扩张与融合的双兼容类别增量学习。ICLR 2023 [论文]
  • DS-AL: 用于无样本类别增量学习的双流解析学习。AAAI 2024 [论文]
  • SimpleCIL: 重新审视预训练模型下的类别增量学习:泛化能力和适应性就是全部所需。IJCV 2024 [论文]
  • Aper: 重新审视预训练模型下的类别增量学习:泛化能力和适应性就是全部所需。IJCV 2024 [论文]
  • TagFex: 用于类别增量学习的任务无关引导特征扩展。CVPR 2025 [论文]

复现的结果

CIFAR-100

ImageNet-100

ImageNet-100(Top-5准确率)

更多实验细节和结果请参阅我们的综述

使用方法

克隆

克隆本GitHub仓库:

git clone https://github.com/G-U-N/PyCIL.git
cd PyCIL

依赖项

  1. torch 1.81
  2. torchvision 0.6.0
  3. tqdm
  4. numpy
  5. scipy
  6. quadprog
  7. POT

运行实验

  1. 编辑 [模型名称].json 文件以设置全局参数。
  2. 在对应的 [模型名称].py 文件中(例如 models/icarl.py)编辑超参数。
  3. 运行以下命令:
python main.py --config=./exps/[模型名称].json

其中 [模型名称] 应从 finetuneewclwfreplaygemicarlbicwapodnetder 等中选择。

  1. 超参数

使用 PyCIL 时,您可以在相应的 JSON 文件中编辑全局参数和特定算法的超参数。

这些参数包括:

  • memory-size:增量学习过程中的总示例数。假设当前阶段有 $K$ 个类别,则模型将为每个类别保留 $\left[\frac{memory-size}{K}\right]$ 个示例。
  • init-cls:第一个增量阶段的类别数量。由于 CIL 中第一阶段的类别数量设置不同,我们的框架支持多种选择来定义初始阶段。
  • increment:每个增量阶段 $i$ 的类别数量,$i > 1$。默认情况下,每个增量阶段的类别数量相等。
  • convnet-type:增量模型的主干网络。根据基准设置,CIFAR100 使用 ResNet32,而 ImageNet 使用 ResNet18
  • seed:用于打乱类别顺序的随机种子。根据基准设置,默认值为 1993。

其他与模型优化相关的参数,如批量大小、优化轮数、学习率、学习率衰减、权重衰减、里程碑和温度,可在对应的 Python 文件中进行修改。

数据集

我们已实现了 CIFAR100imagenet100imagenet1000 的预处理。在训练 CIFAR100 时,该框架会自动下载数据集。而在训练 imagenet100/1000 时,您需要在 utils/data.py 中指定您的数据集文件夹路径。

    def download_data(self):
        assert 0,"您应指定您的数据集文件夹"
        train_dir = '[DATA-PATH]/train/'
        test_dir = '[DATA-PATH]/val/'

此处 是 ImageNet100(或称 ImageNet-Sub)的数据文件列表。

使用 PyCIL 的优秀论文

我们的论文

  • 基于 CLIP 的类增量学习中的外部知识注入(ICCV 2025)[论文] [代码]

  • 针对预训练模型的类增量学习中任务特定适配器与通用适配器的融合(ICCV 2025)[论文] [代码]

  • 通过双平衡协作专家解决不平衡领域增量学习问题(ICML 2025)[论文] [代码]

  • 预训练模型驱动下的领域增量学习中的双重整合(CVPR 2025)[论文] [代码]

  • 面向类增量学习的任务无关引导特征扩展(CVPR 2025)[论文] [代码]

  • 视觉-语言模型的无遗忘学习(TPAMI 2025)[论文] [代码]

  • 重新审视基于预训练模型的类增量学习:泛化能力和适应性就是全部所需(IJCV 2025)[论文] [代码]

  • PILOT:基于预训练模型的持续学习工具箱(SCIS 2025)[论文] [代码]

  • 类增量学习综述(TPAMI 2024)[论文] [代码]

  • 预训练模型驱动下的可扩展子空间集成类增量学习(CVPR 2024)[论文] [代码]

  • 面向类增量学习的多层重演特征增强(ICML 2024)[论文] [代码]

  • 基于预训练模型的持续学习综述(IJCAI 2024)[论文] [代码]

  • 长尾类增量学习中的自适应适配器路由(机器学习 2024)[论文] [代码]

  • 无需训练的原型校准实现的小样本类增量学习(NeurIPS 2023)[论文] [代码]

  • BEEF:基于能量的扩展与融合实现的双向兼容类增量学习(ICLR 2023)[论文] [代码]

  • 一个模型还是 603 个示例?迈向内存高效的类增量学习(ICLR 2023)[论文] [代码]

  • 通过采样多阶段任务实现的小样本类增量学习(TPAMI 2022)[论文] [代码]

  • FOSTER:面向类增量学习的特征增强与压缩(ECCV 2022)[论文] [代码]

  • 向前兼容的小样本类增量学习(CVPR 2022)[论文] [代码]

  • 面向类增量学习的协同运输(ACM MM 2021)[论文] [代码]

其他优秀作品

  • 面向工业持续学习场景的现实评估:重点关注能耗与计算开销(ICCV 2023)[论文][代码]

  • 用于类别增量学习的动态残差分类器(ICCV 2023)[论文][代码]

  • 基于预训练Transformer的S-Prompts学习:面向领域增量学习的奥卡姆剃刀原则(NeurIPS 2022)[论文][代码]

  • ...

许可证

请查看本仓库中列出的 MIT 许可证

致谢

我们感谢以下项目在我们的工作中提供了有用的组件或功能。

我们的训练流程和数据配置基于 Continual-Learning-Reproduce。该仓库的基础分支中包含了原始信息。

联系方式

如有任何问题,欢迎通过提交 issue 提出新功能建议,或直接联系作者:Da-Wei Zhou(zhoudw@lamda.nju.edu.cn) 和 Fu-Yun Wang(wangfuyun@smail.nju.edu.cn)。祝您使用愉快。

星标历史 🚀

星标历史图表

版本历史

v0.2.12023/07/13
v0.12022/12/12

常见问题

相似工具推荐

openclaw

OpenClaw 是一款专为个人打造的本地化 AI 助手,旨在让你在自己的设备上拥有完全可控的智能伙伴。它打破了传统 AI 助手局限于特定网页或应用的束缚,能够直接接入你日常使用的各类通讯渠道,包括微信、WhatsApp、Telegram、Discord、iMessage 等数十种平台。无论你在哪个聊天软件中发送消息,OpenClaw 都能即时响应,甚至支持在 macOS、iOS 和 Android 设备上进行语音交互,并提供实时的画布渲染功能供你操控。 这款工具主要解决了用户对数据隐私、响应速度以及“始终在线”体验的需求。通过将 AI 部署在本地,用户无需依赖云端服务即可享受快速、私密的智能辅助,真正实现了“你的数据,你做主”。其独特的技术亮点在于强大的网关架构,将控制平面与核心助手分离,确保跨平台通信的流畅性与扩展性。 OpenClaw 非常适合希望构建个性化工作流的技术爱好者、开发者,以及注重隐私保护且不愿被单一生态绑定的普通用户。只要具备基础的终端操作能力(支持 macOS、Linux 及 Windows WSL2),即可通过简单的命令行引导完成部署。如果你渴望拥有一个懂你

349.3k|★★★☆☆|1周前
Agent开发框架图像

stable-diffusion-webui

stable-diffusion-webui 是一个基于 Gradio 构建的网页版操作界面,旨在让用户能够轻松地在本地运行和使用强大的 Stable Diffusion 图像生成模型。它解决了原始模型依赖命令行、操作门槛高且功能分散的痛点,将复杂的 AI 绘图流程整合进一个直观易用的图形化平台。 无论是希望快速上手的普通创作者、需要精细控制画面细节的设计师,还是想要深入探索模型潜力的开发者与研究人员,都能从中获益。其核心亮点在于极高的功能丰富度:不仅支持文生图、图生图、局部重绘(Inpainting)和外绘(Outpainting)等基础模式,还独创了注意力机制调整、提示词矩阵、负向提示词以及“高清修复”等高级功能。此外,它内置了 GFPGAN 和 CodeFormer 等人脸修复工具,支持多种神经网络放大算法,并允许用户通过插件系统无限扩展能力。即使是显存有限的设备,stable-diffusion-webui 也提供了相应的优化选项,让高质量的 AI 艺术创作变得触手可及。

162.1k|★★★☆☆|1周前
开发框架图像Agent

everything-claude-code

everything-claude-code 是一套专为 AI 编程助手(如 Claude Code、Codex、Cursor 等)打造的高性能优化系统。它不仅仅是一组配置文件,而是一个经过长期实战打磨的完整框架,旨在解决 AI 代理在实际开发中面临的效率低下、记忆丢失、安全隐患及缺乏持续学习能力等核心痛点。 通过引入技能模块化、直觉增强、记忆持久化机制以及内置的安全扫描功能,everything-claude-code 能显著提升 AI 在复杂任务中的表现,帮助开发者构建更稳定、更智能的生产级 AI 代理。其独特的“研究优先”开发理念和针对 Token 消耗的优化策略,使得模型响应更快、成本更低,同时有效防御潜在的攻击向量。 这套工具特别适合软件开发者、AI 研究人员以及希望深度定制 AI 工作流的技术团队使用。无论您是在构建大型代码库,还是需要 AI 协助进行安全审计与自动化测试,everything-claude-code 都能提供强大的底层支持。作为一个曾荣获 Anthropic 黑客大奖的开源项目,它融合了多语言支持与丰富的实战钩子(hooks),让 AI 真正成长为懂上

159.6k|★★☆☆☆|今天
开发框架Agent语言模型

ComfyUI

ComfyUI 是一款功能强大且高度模块化的视觉 AI 引擎,专为设计和执行复杂的 Stable Diffusion 图像生成流程而打造。它摒弃了传统的代码编写模式,采用直观的节点式流程图界面,让用户通过连接不同的功能模块即可构建个性化的生成管线。 这一设计巧妙解决了高级 AI 绘图工作流配置复杂、灵活性不足的痛点。用户无需具备编程背景,也能自由组合模型、调整参数并实时预览效果,轻松实现从基础文生图到多步骤高清修复等各类复杂任务。ComfyUI 拥有极佳的兼容性,不仅支持 Windows、macOS 和 Linux 全平台,还广泛适配 NVIDIA、AMD、Intel 及苹果 Silicon 等多种硬件架构,并率先支持 SDXL、Flux、SD3 等前沿模型。 无论是希望深入探索算法潜力的研究人员和开发者,还是追求极致创作自由度的设计师与资深 AI 绘画爱好者,ComfyUI 都能提供强大的支持。其独特的模块化架构允许社区不断扩展新功能,使其成为当前最灵活、生态最丰富的开源扩散模型工具之一,帮助用户将创意高效转化为现实。

108.3k|★★☆☆☆|1周前
开发框架图像Agent

gemini-cli

gemini-cli 是一款由谷歌推出的开源 AI 命令行工具,它将强大的 Gemini 大模型能力直接集成到用户的终端环境中。对于习惯在命令行工作的开发者而言,它提供了一条从输入提示词到获取模型响应的最短路径,无需切换窗口即可享受智能辅助。 这款工具主要解决了开发过程中频繁上下文切换的痛点,让用户能在熟悉的终端界面内直接完成代码理解、生成、调试以及自动化运维任务。无论是查询大型代码库、根据草图生成应用,还是执行复杂的 Git 操作,gemini-cli 都能通过自然语言指令高效处理。 它特别适合广大软件工程师、DevOps 人员及技术研究人员使用。其核心亮点包括支持高达 100 万 token 的超长上下文窗口,具备出色的逻辑推理能力;内置 Google 搜索、文件操作及 Shell 命令执行等实用工具;更独特的是,它支持 MCP(模型上下文协议),允许用户灵活扩展自定义集成,连接如图像生成等外部能力。此外,个人谷歌账号即可享受免费的额度支持,且项目基于 Apache 2.0 协议完全开源,是提升终端工作效率的理想助手。

100.8k|★★☆☆☆|1周前
插件Agent图像

markitdown

MarkItDown 是一款由微软 AutoGen 团队打造的轻量级 Python 工具,专为将各类文件高效转换为 Markdown 格式而设计。它支持 PDF、Word、Excel、PPT、图片(含 OCR)、音频(含语音转录)、HTML 乃至 YouTube 链接等多种格式的解析,能够精准提取文档中的标题、列表、表格和链接等关键结构信息。 在人工智能应用日益普及的今天,大语言模型(LLM)虽擅长处理文本,却难以直接读取复杂的二进制办公文档。MarkItDown 恰好解决了这一痛点,它将非结构化或半结构化的文件转化为模型“原生理解”且 Token 效率极高的 Markdown 格式,成为连接本地文件与 AI 分析 pipeline 的理想桥梁。此外,它还提供了 MCP(模型上下文协议)服务器,可无缝集成到 Claude Desktop 等 LLM 应用中。 这款工具特别适合开发者、数据科学家及 AI 研究人员使用,尤其是那些需要构建文档检索增强生成(RAG)系统、进行批量文本分析或希望让 AI 助手直接“阅读”本地文件的用户。虽然生成的内容也具备一定可读性,但其核心优势在于为机器

93.4k|★★☆☆☆|1周前
插件开发框架