CIFAR-10
收藏资源简介:
CIFAR-10数据集是由慕尼黑工业大学和牛津大学共同研究使用的图像数据集,主要用于训练和测试图像识别模型。该数据集包含多个类别的图像,每张图像都有明确的分类标签。数据集的创建过程涉及对原始数据进行精简和合成,以生成更小但信息丰富的合成数据集。CIFAR-10数据集主要应用于计算机视觉领域,特别是在深度学习模型的训练和评估中,旨在解决大规模数据集带来的计算成本问题。
The CIFAR-10 dataset is a collaborative research image dataset jointly used by the Technical University of Munich and the University of Oxford, primarily designed for training and testing image recognition models. This dataset encompasses images from multiple categories, each with a clear classification label. The creation process of the dataset involves simplification and synthesis of the original data to generate a smaller yet informative synthetic dataset. The CIFAR-10 dataset is predominantly applied in the field of computer vision, particularly for the training and evaluation of deep learning models, aiming to address the computational cost issues associated with large-scale datasets.
数据集概述:Dataset Distillation by Automatic Training Trajectories
数据集简介
- 数据集蒸馏方法:通过自动训练轨迹生成合成数据集
- 核心功能:从原始数据集(如CIFAR10、CIFAR100等)中提取专家轨迹并蒸馏合成数据
支持的数据集
- CIFAR10
- CIFAR100
- Tiny ImageNet
- ImageNet子集:
- imagenette
- imagewoof
- imagefruit
- imagemeow
主要功能
1. 专家轨迹生成
- 命令示例: bash python buffer_ATT.py --dataset=CIFAR10 --model=ConvNetD3 --train_epochs=50 --num_experts=100 --zca --data_path=data --buffer_path=buffer
2. 数据集蒸馏
- 支持不同IPC(Images Per Class)设置:
- IPC = 1
- IPC = 10
- IPC = 50
- 命令示例(CIFAR10 IPC=10): bash python3 distill_ATT.py --dataset=CIFAR10 --model=ConvNetD3 --ipc=10 --zca --syn_steps=30 --expert_epochs=2 --max_start_epoch=20 --lr_img=1e04 --lr_lr=1e-04 --lr_teacher=1e-02 --ema_decay=0.9999 --eval_it=100 --Iteration=60000 --buffer_path=buffer --data_path=data
性能评估
- 跨架构性能:
- 支持在不同模型架构上评估合成数据集性能
- 测试准确率:
- 提供在ConvNet上的测试准确率结果
依赖环境
- Python包依赖:
- 通过
requirements.txt安装 - 详细版本信息见
requirements_detail.txt
- 通过
监控工具
- 推荐使用wandb监控蒸馏过程
- 可选项,可通过设置
mode=disabled禁用
参考代码
- 引用了多个开源项目代码:
- FTD-distillation
- mtt-distillation
- DatasetCondensation
- pytorch-cifar
- CIFAR-ZOO
- rwightman的代码片段




