AI分类任务种类

AI分类任务种类分为3大类:二分类任务,多标签多分类任务,单标签多分类任务。

1 二分类任务

二分类任务很简单,就是"是"与"不是"。输入一个图片,判断是不是有猫,是为1,不是为0。

输出层神经元个数有两种情况:

(1)输出层只有一个神经元,各个输出神经元的处理方法(这里只有一个神经元):

logits+sigmoid激活+0或1标签+BCE损失计算。

(2)输出层的神经元有2个,对于这种特殊情况,各个输出神经元的处理方法:

logits+softmax激活+onehot编码标签+CCE损失计算。

二分类任务主流多是第一种情况,第二种情况少见。

2 多标签多分类任务

多标签多分类任务就输入有多个标签(即多个类型),输出类型也是多个(至少2个类别)。

什么意思?比如一张图片里有猫和狗两个类别,输出类别是人/猫/狗/车/房等5个类别,即输出层有5个神经元------每个神经元代表一类------就像信号灯一样,我亮起来给你留灯,你属于我这一类------你训练的目标就是让输出层中代表猫和狗的2个神经元同时输出1。

这种和二分类任务的第一种情况一样,所以我们可以把多标签多分类分解为多个二分类任务,于是它的解法和二分类任务第一种情况一样,各个输出神经元的处理方法:

logits+sigmoid激活+0或1标签+BCE损失计算。

正因为它和二分类任务主流一样,所以把它紧挨着二分类讲。要不然我为啥不先说单标签多分类任务呢?

3 单标签多分类任务

单标签多分类任务就是在多个分类中选一个类别作为你的类别。

比如你定的模型输出的类别有人/猫/狗/车/房等5大类别,你输入的图片只有一个标签(属于5类中的一类)。也就是说输出层神经元有5个,模型每次推理选择5类中最大的值作为输入的标签,在和真实标签比对进行训练。

也就是说真实标签是onehot编码。

于是,我们可以这样处理:

logits+softmax激活+onehot编码标签+CCE损失计算。

可以看到和二分类任务的第二种情况一样!

相关推荐
东坡肘子11 分钟前
参与学生移动应用大赛有感 -- 肘子的 Swift 周报 #161
人工智能·swiftui·swift
天天被压力13 分钟前
【A股分钟级高频数据完全手册 #11】券商实时行情数据接口:实时交易(券商源)接入与字段
java·人工智能·python
唐老板17 分钟前
agent 撑爆 Git 之后,两家改的不是同一层
人工智能·ai编程
天天被压力26 分钟前
【A股分钟级高频数据完全手册 #12】股票五档盘口数据接口:买卖五档实时盘口解析
java·人工智能·python
爱敲代码的小黄9 小时前
神经网络入门指南:从 CNN、RNN 到 Attention
人工智能·面试·程序员
一只爱撸猫的程序猿9 小时前
我看着一个 AI 引擎在 14 天里,从"记忆"走进了"搜索"
人工智能·后端·架构
我登哥MVP10 小时前
DeepSeek-Harness-Super-Agent-学习笔记
人工智能·深度学习·神经网络·机器学习·自然语言处理·agent
LucianaiB10 小时前
【元宝搜索开发者征文大赛】AI时代,学得慢就不用学?用元宝搜索打造一个教程保质期,确保每一个教程符合你的现状
前端·人工智能·ai编程
冬奇Lab12 小时前
一天一个开源项目(第256篇):Anatomy —— 让 Claude 为技术概念「发明一台机器」,画成可交互的等距插画
人工智能·开源