🚀 vit-base-patch16-224-in21k_lung_and_colon_cancer
本模型是一个图像分类模型,基于 google/vit-base-patch16-224-in21k 进行微调,可用于肺癌和结肠癌的多分类图像识别,在评估集上表现出色。
🚀 快速开始
本模型是 google/vit-base-patch16-224-in21k 的微调版本。
它在评估集上取得了以下结果:
- 损失率:0.0016
- 准确率:0.9994
- F1值
- 加权:0.9994
- 微平均:0.9994
- 宏平均:0.9994
- 召回率
- 加权:0.9994
- 微平均:0.9994
- 宏平均:0.9994
- 精确率
- 加权:0.9994
- 微平均:0.9994
- 宏平均:0.9994
✨ 主要特性
- 高精度:在评估集上准确率达到 0.9994,各项指标表现优异。
- 多分类能力:能够对肺癌和结肠癌进行多分类图像识别。
📚 详细文档
模型描述
这是一个用于肺癌和结肠癌的多分类图像分类模型。
有关其创建方式的更多信息,请查看以下链接:https://github.com/DunnBC22/Vision_Audio_and_Multimodal_Projects/blob/main/Computer%20Vision/Image%20Classification/Multiclass%20Classification/Lung%20%26%20Colon%20Cancer/Lung_and_colon_cancer_ViT.ipynb
预期用途与限制
本模型旨在展示使用技术解决复杂问题的能力。
训练和评估数据
数据集来源:https://www.kaggle.com/datasets/andrewmvd/lung-and-colon-cancer-histopathological-images
数据集中的示例图像:

训练过程
训练超参数
训练期间使用了以下超参数:
- 学习率:0.0002
- 训练批次大小:16
- 评估批次大小:8
- 随机种子:42
- 优化器:Adam(β1=0.9,β2=0.999,ε=1e-08)
- 学习率调度器类型:线性
- 训练轮数:3
训练结果
训练损失 |
轮数 |
步数 |
验证损失 |
准确率 |
加权 F1 值 |
微平均 F1 值 |
宏平均 F1 值 |
加权召回率 |
微平均召回率 |
宏平均召回率 |
加权精确率 |
微平均精确率 |
宏平均精确率 |
0.0574 |
1.0 |
1250 |
0.0410 |
0.9864 |
0.9864 |
0.9864 |
0.9865 |
0.9864 |
0.9864 |
0.9864 |
0.9872 |
0.9864 |
0.9875 |
0.0031 |
2.0 |
2500 |
0.0105 |
0.9972 |
0.9972 |
0.9972 |
0.9972 |
0.9972 |
0.9972 |
0.9973 |
0.9972 |
0.9972 |
0.9972 |
0.0007 |
3.0 |
3750 |
0.0016 |
0.9994 |
0.9994 |
0.9994 |
0.9994 |
0.9994 |
0.9994 |
0.9994 |
0.9994 |
0.9994 |
0.9994 |
框架版本
- Transformers 4.22.2
- Pytorch 1.12.1
- Datasets 2.5.2
- Tokenizers 0.12.1
📄 许可证
本模型采用 Apache-2.0 许可证。
📋 信息表格
属性 |
详情 |
模型类型 |
图像分类模型 |
训练数据 |
来自 imagefolder 数据集 |
评估指标 |
损失率、准确率、F1值(加权、微平均、宏平均)、召回率(加权、微平均、宏平均)、精确率(加权、微平均、宏平均) |
训练框架版本 |
Transformers 4.22.2、Pytorch 1.12.1、Datasets 2.5.2、Tokenizers 0.12.1 |